Skip to content

Commit 451ef2c

Browse files
committed
Use assert.Equal to assert fields in struct
1 parent 6fa6f8b commit 451ef2c

File tree

2 files changed

+48
-17
lines changed

2 files changed

+48
-17
lines changed

restapi/admin_remote_buckets_test.go

Lines changed: 47 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import (
2424
"os"
2525
"testing"
2626

27+
"github.com/go-openapi/swag"
2728
"github.com/minio/console/models"
2829
"github.com/minio/console/restapi/operations"
2930
bucketApi "github.com/minio/console/restapi/operations/bucket"
@@ -34,18 +35,49 @@ import (
3435

3536
type RemoteBucketsTestSuite struct {
3637
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
4347
}
4448

4549
func (suite *RemoteBucketsTestSuite) SetupSuite() {
4650
suite.assert = assert.New(suite.T())
4751
suite.adminClient = AdminClientMock{}
4852
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+
}
4981
}
5082

5183
func (suite *RemoteBucketsTestSuite) SetupTest() {
@@ -146,16 +178,12 @@ func (suite *RemoteBucketsTestSuite) initRemoteBucketDetailsRequest() (params bu
146178
func (suite *RemoteBucketsTestSuite) TestGetRemoteBucketWithoutError() {
147179
ctx := context.Background()
148180
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
155182
}
156183
res, err := getRemoteBucket(ctx, &suite.adminClient, "bucketName")
157-
suite.assert.NotNil(res)
158184
suite.assert.Nil(err)
185+
suite.assert.NotNil(res)
186+
suite.assert.Equal(suite.mockRemoteBucket, res)
159187
}
160188

161189
func (suite *RemoteBucketsTestSuite) TestDeleteRemoteBucketHandlerWithError() {
@@ -273,18 +301,21 @@ func (suite *RemoteBucketsTestSuite) TestListExternalBucketsWithError() {
273301
return madmin.AccountInfo{}, errors.New("error")
274302
}
275303
res, err := listExternalBuckets(ctx, &suite.adminClient)
276-
suite.assert.Nil(res)
277304
suite.assert.NotNil(err)
305+
suite.assert.Nil(res)
278306
}
279307

280308
func (suite *RemoteBucketsTestSuite) TestListExternalBucketsWithoutError() {
281309
ctx := context.Background()
282310
minioAccountInfoMock = func(ctx context.Context) (madmin.AccountInfo, error) {
283-
return madmin.AccountInfo{}, nil
311+
return madmin.AccountInfo{
312+
Buckets: []madmin.BucketAccessInfo{},
313+
}, nil
284314
}
285315
res, err := listExternalBuckets(ctx, &suite.adminClient)
286-
suite.assert.NotNil(res)
287316
suite.assert.Nil(err)
317+
suite.assert.NotNil(res)
318+
suite.assert.Equal(suite.mockListBuckets, res)
288319
}
289320

290321
func (suite *RemoteBucketsTestSuite) TestDeleteBucketReplicationRuleHandlerWithError() {

restapi/user_buckets.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ func getAccountBuckets(ctx context.Context, client MinioAdmin) ([]*models.Bucket
381381
if err != nil {
382382
return []*models.Bucket{}, err
383383
}
384-
var bucketInfos []*models.Bucket
384+
bucketInfos := []*models.Bucket{}
385385
for _, bucket := range info.Buckets {
386386
bucketElem := &models.Bucket{
387387
CreationDate: bucket.Created.Format(time.RFC3339),

0 commit comments

Comments
 (0)