@@ -24,6 +24,7 @@ import (
24
24
"os"
25
25
"testing"
26
26
27
+ "github.com/go-openapi/swag"
27
28
"github.com/minio/console/models"
28
29
"github.com/minio/console/restapi/operations"
29
30
bucketApi "github.com/minio/console/restapi/operations/bucket"
@@ -34,18 +35,49 @@ import (
34
35
35
36
type RemoteBucketsTestSuite struct {
36
37
suite.Suite
37
- assert * assert.Assertions
38
- currentServer string
39
- isServerSet bool
40
- server * httptest.Server
41
- adminClient AdminClientMock
42
- minioClient minioClientMock
38
+ assert * assert.Assertions
39
+ currentServer string
40
+ isServerSet bool
41
+ server * httptest.Server
42
+ adminClient AdminClientMock
43
+ minioClient minioClientMock
44
+ mockRemoteBucket * models.RemoteBucket
45
+ mockBucketTarget * madmin.BucketTarget
46
+ mockListBuckets * models.ListBucketsResponse
43
47
}
44
48
45
49
func (suite * RemoteBucketsTestSuite ) SetupSuite () {
46
50
suite .assert = assert .New (suite .T ())
47
51
suite .adminClient = AdminClientMock {}
48
52
suite .minioClient = minioClientMock {}
53
+ suite .mockObjects ()
54
+ }
55
+
56
+ func (suite * RemoteBucketsTestSuite ) mockObjects () {
57
+ suite .mockListBuckets = & models.ListBucketsResponse {
58
+ Buckets : []* models.Bucket {},
59
+ Total : 0 ,
60
+ }
61
+ suite .mockRemoteBucket = & models.RemoteBucket {
62
+ AccessKey : swag .String ("accessKey" ),
63
+ SecretKey : "secretKey" ,
64
+ RemoteARN : swag .String ("remoteARN" ),
65
+ Service : "replication" ,
66
+ SourceBucket : swag .String ("sourceBucket" ),
67
+ TargetBucket : "targetBucket" ,
68
+ TargetURL : "targetURL" ,
69
+ Status : "" ,
70
+ }
71
+ suite .mockBucketTarget = & madmin.BucketTarget {
72
+ Credentials : & madmin.Credentials {
73
+ AccessKey : * suite .mockRemoteBucket .AccessKey ,
74
+ SecretKey : suite .mockRemoteBucket .SecretKey ,
75
+ },
76
+ Arn : * suite .mockRemoteBucket .RemoteARN ,
77
+ SourceBucket : * suite .mockRemoteBucket .SourceBucket ,
78
+ TargetBucket : suite .mockRemoteBucket .TargetBucket ,
79
+ Endpoint : suite .mockRemoteBucket .TargetURL ,
80
+ }
49
81
}
50
82
51
83
func (suite * RemoteBucketsTestSuite ) SetupTest () {
@@ -146,16 +178,12 @@ func (suite *RemoteBucketsTestSuite) initRemoteBucketDetailsRequest() (params bu
146
178
func (suite * RemoteBucketsTestSuite ) TestGetRemoteBucketWithoutError () {
147
179
ctx := context .Background ()
148
180
minioGetRemoteBucketMock = func (_ context.Context , _ , _ string ) (targets * madmin.BucketTarget , err error ) {
149
- return & madmin.BucketTarget {
150
- Credentials : & madmin.Credentials {
151
- AccessKey : "accessKey" ,
152
- SecretKey : "secretKey" ,
153
- },
154
- }, nil
181
+ return suite .mockBucketTarget , nil
155
182
}
156
183
res , err := getRemoteBucket (ctx , & suite .adminClient , "bucketName" )
157
- suite .assert .NotNil (res )
158
184
suite .assert .Nil (err )
185
+ suite .assert .NotNil (res )
186
+ suite .assert .Equal (suite .mockRemoteBucket , res )
159
187
}
160
188
161
189
func (suite * RemoteBucketsTestSuite ) TestDeleteRemoteBucketHandlerWithError () {
@@ -273,18 +301,21 @@ func (suite *RemoteBucketsTestSuite) TestListExternalBucketsWithError() {
273
301
return madmin.AccountInfo {}, errors .New ("error" )
274
302
}
275
303
res , err := listExternalBuckets (ctx , & suite .adminClient )
276
- suite .assert .Nil (res )
277
304
suite .assert .NotNil (err )
305
+ suite .assert .Nil (res )
278
306
}
279
307
280
308
func (suite * RemoteBucketsTestSuite ) TestListExternalBucketsWithoutError () {
281
309
ctx := context .Background ()
282
310
minioAccountInfoMock = func (ctx context.Context ) (madmin.AccountInfo , error ) {
283
- return madmin.AccountInfo {}, nil
311
+ return madmin.AccountInfo {
312
+ Buckets : []madmin.BucketAccessInfo {},
313
+ }, nil
284
314
}
285
315
res , err := listExternalBuckets (ctx , & suite .adminClient )
286
- suite .assert .NotNil (res )
287
316
suite .assert .Nil (err )
317
+ suite .assert .NotNil (res )
318
+ suite .assert .Equal (suite .mockListBuckets , res )
288
319
}
289
320
290
321
func (suite * RemoteBucketsTestSuite ) TestDeleteBucketReplicationRuleHandlerWithError () {
0 commit comments