@@ -22,8 +22,12 @@ import (
22
22
"bytes"
23
23
"encoding/json"
24
24
"fmt"
25
+ "io"
25
26
"log"
27
+ "mime/multipart"
26
28
"net/http"
29
+ "os"
30
+ "path"
27
31
"testing"
28
32
"time"
29
33
@@ -155,33 +159,33 @@ func NotifyPostgres() (*http.Response, error) {
155
159
156
160
func TestNotifyPostgres (t * testing.T ) {
157
161
// Variables
158
- assert := assert .New (t )
162
+ asserter := assert .New (t )
159
163
160
164
// Test
161
165
response , err := NotifyPostgres ()
162
166
finalResponse := inspectHTTPResponse (response )
163
- assert .Nil (err )
167
+ asserter .Nil (err )
164
168
if err != nil {
165
169
log .Println (err )
166
- assert .Fail (finalResponse )
170
+ asserter .Fail (finalResponse )
167
171
return
168
172
}
169
173
if response != nil {
170
- assert .Equal (200 , response .StatusCode , finalResponse )
174
+ asserter .Equal (200 , response .StatusCode , finalResponse )
171
175
}
172
176
}
173
177
174
178
func TestRestartService (t * testing.T ) {
175
- assert := assert .New (t )
179
+ asserter := assert .New (t )
176
180
restartResponse , restartError := RestartService ()
177
- assert .Nil (restartError )
181
+ asserter .Nil (restartError )
178
182
if restartError != nil {
179
183
log .Println (restartError )
180
184
return
181
185
}
182
186
addObjRsp := inspectHTTPResponse (restartResponse )
183
187
if restartResponse != nil {
184
- assert .Equal (
188
+ asserter .Equal (
185
189
204 ,
186
190
restartResponse .StatusCode ,
187
191
addObjRsp ,
@@ -212,18 +216,18 @@ func ListPoliciesWithBucket(bucketName string) (*http.Response, error) {
212
216
func TestListPoliciesWithBucket (t * testing.T ) {
213
217
// Test Variables
214
218
bucketName := "testlistpolicieswithbucket"
215
- assert := assert .New (t )
219
+ asserter := assert .New (t )
216
220
217
221
// Test
218
222
response , err := ListPoliciesWithBucket (bucketName )
219
- assert .Nil (err )
223
+ asserter .Nil (err )
220
224
if err != nil {
221
225
log .Println (err )
222
226
return
223
227
}
224
228
parsedResponse := inspectHTTPResponse (response )
225
229
if response != nil {
226
- assert .Equal (
230
+ asserter .Equal (
227
231
200 ,
228
232
response .StatusCode ,
229
233
parsedResponse ,
@@ -254,18 +258,18 @@ func ListUsersWithAccessToBucket(bucketName string) (*http.Response, error) {
254
258
func TestListUsersWithAccessToBucket (t * testing.T ) {
255
259
// Test Variables
256
260
bucketName := "testlistuserswithaccesstobucket1"
257
- assert := assert .New (t )
261
+ asserter := assert .New (t )
258
262
259
263
// Test
260
264
response , err := ListUsersWithAccessToBucket (bucketName )
261
- assert .Nil (err )
265
+ asserter .Nil (err )
262
266
if err != nil {
263
267
log .Println (err )
264
268
return
265
269
}
266
270
parsedResponse := inspectHTTPResponse (response )
267
271
if response != nil {
268
- assert .Equal (
272
+ asserter .Equal (
269
273
200 ,
270
274
response .StatusCode ,
271
275
parsedResponse ,
@@ -274,16 +278,16 @@ func TestListUsersWithAccessToBucket(t *testing.T) {
274
278
}
275
279
276
280
func TestGetNodes (t * testing.T ) {
277
- assert := assert .New (t )
281
+ asserter := assert .New (t )
278
282
getNodesResponse , getNodesError := GetNodes ()
279
- assert .Nil (getNodesError )
283
+ asserter .Nil (getNodesError )
280
284
if getNodesError != nil {
281
285
log .Println (getNodesError )
282
286
return
283
287
}
284
288
addObjRsp := inspectHTTPResponse (getNodesResponse )
285
289
if getNodesResponse != nil {
286
- assert .Equal (
290
+ asserter .Equal (
287
291
200 ,
288
292
getNodesResponse .StatusCode ,
289
293
addObjRsp ,
@@ -312,16 +316,89 @@ func ArnList() (*http.Response, error) {
312
316
}
313
317
314
318
func TestArnList (t * testing.T ) {
315
- assert := assert .New (t )
319
+ asserter := assert .New (t )
316
320
resp , err := ArnList ()
317
- assert .Nil (err )
321
+ asserter .Nil (err )
318
322
if err != nil {
319
323
log .Println (err )
320
324
return
321
325
}
322
326
objRsp := inspectHTTPResponse (resp )
323
327
if resp != nil {
324
- assert .Equal (
328
+ asserter .Equal (
329
+ 200 ,
330
+ resp .StatusCode ,
331
+ objRsp ,
332
+ )
333
+ }
334
+ }
335
+
336
+ func ExportConfig () (* http.Response , error ) {
337
+ request , err := http .NewRequest (
338
+ "GET" , "http://localhost:9090/api/v1/configs/export" , nil )
339
+ if err != nil {
340
+ log .Println (err )
341
+ }
342
+ request .Header .Add ("Cookie" , fmt .Sprintf ("token=%s" , token ))
343
+ client := & http.Client {
344
+ Timeout : 2 * time .Second ,
345
+ }
346
+ response , err := client .Do (request )
347
+ return response , err
348
+ }
349
+
350
+ func ImportConfig () (* http.Response , error ) {
351
+ body := & bytes.Buffer {}
352
+ writer := multipart .NewWriter (body )
353
+ formFile , _ := writer .CreateFormFile ("file" , "sample-import-config.txt" )
354
+ fileDir , _ := os .Getwd ()
355
+ fileName := "sample-import-config.txt"
356
+ filePath := path .Join (fileDir , fileName )
357
+ file , _ := os .Open (filePath )
358
+ io .Copy (formFile , file )
359
+ writer .Close ()
360
+ request , err := http .NewRequest (
361
+ "POST" , "http://localhost:9090/api/v1/configs/import" ,
362
+ bytes .NewReader (body .Bytes ()),
363
+ )
364
+ if err != nil {
365
+ log .Println (err )
366
+ }
367
+ request .Header .Add ("Cookie" , fmt .Sprintf ("token=%s" , token ))
368
+ request .Header .Set ("Content-Type" , writer .FormDataContentType ())
369
+
370
+ client := & http.Client {
371
+ Timeout : 2 * time .Second ,
372
+ }
373
+
374
+ rsp , _ := client .Do (request )
375
+ if rsp .StatusCode != http .StatusOK {
376
+ log .Printf ("Request failed with response code: %d" , rsp .StatusCode )
377
+ }
378
+ return rsp , err
379
+ }
380
+
381
+ func TestExportConfig (t * testing.T ) {
382
+ asserter := assert .New (t )
383
+ resp , err := ExportConfig ()
384
+ asserter .Nil (err )
385
+ objRsp := inspectHTTPResponse (resp )
386
+ if resp != nil {
387
+ asserter .Equal (
388
+ 200 ,
389
+ resp .StatusCode ,
390
+ objRsp ,
391
+ )
392
+ }
393
+ }
394
+
395
+ func TestImportConfig (t * testing.T ) {
396
+ asserter := assert .New (t )
397
+ resp , err := ImportConfig ()
398
+ asserter .Nil (err )
399
+ objRsp := inspectHTTPResponse (resp )
400
+ if resp != nil {
401
+ asserter .Equal (
325
402
200 ,
326
403
resp .StatusCode ,
327
404
objRsp ,
0 commit comments