JakeHillion/storj
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

satellite/metainfo: tests for new GetBucketLocation method

Browse Source 
Change-Id: I809ebab51606aa9e55dff3c40ef2e865caf06924
This commit is contained in:
Michal Niewrzal 2023-08-25 11:50:02 +02:00 committed by Artur M. Wolff
parent 37d6df23fa
commit b26df035f9
2 changed files with 70 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
satellite/metainfo/endpoint_bucket_test.go
View File

@ -12,6 +12,7 @@ import (
"github.com/stretchr/testify/require"
"github.com/zeebo/errs"
"go.uber.org/zap"
"storj.io/common/errs2"
"storj.io/common/memory"
@ -21,7 +22,10 @@ import (
"storj.io/common/testcontext"
"storj.io/common/testrand"
"storj.io/storj/private/testplanet"
"storj.io/storj/satellite"
"storj.io/storj/satellite/buckets"
"storj.io/storj/satellite/nodeselection"
"storj.io/storj/satellite/overlay"
"storj.io/uplink"
"storj.io/uplink/private/metaclient"
)
@ -297,3 +301,62 @@ func TestBucketCreationWithDefaultPlacement(t *testing.T) {
})
}
func TestGetBucketLocation(t *testing.T) {
placementRules := overlay.NewPlacementRules()
err := placementRules.AddPlacementFromString(fmt.Sprintf(`40:annotated(annotated(country("PL"),annotation("%s","Poland")),annotation("%s","%s"))`,
nodeselection.Location, nodeselection.AutoExcludeSubnet, nodeselection.AutoExcludeSubnetOFF))
require.NoError(t, err)
testplanet.Run(t, testplanet.Config{
SatelliteCount: 1, StorageNodeCount: 0, UplinkCount: 1,
Reconfigure: testplanet.Reconfigure{
Satellite: func(log *zap.Logger, index int, config *satellite.Config) {
config.Placement = *placementRules
},
},
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
apiKey := planet.Uplinks[0].APIKey[planet.Satellites[0].ID()]
satellite := planet.Satellites[0]
// not existing bucket
_, err := satellite.API.Metainfo.Endpoint.GetBucketLocation(ctx, &pb.GetBucketLocationRequest{
Header: &pb.RequestHeader{
ApiKey: apiKey.SerializeRaw(),
},
Name: []byte("test-bucket"),
})
require.True(t, errs2.IsRPC(err, rpcstatus.NotFound))
err = planet.Uplinks[0].CreateBucket(ctx, planet.Satellites[0], "test-bucket")
require.NoError(t, err)
// bucket without location
response, err := satellite.API.Metainfo.Endpoint.GetBucketLocation(ctx, &pb.GetBucketLocationRequest{
Header: &pb.RequestHeader{
ApiKey: apiKey.SerializeRaw(),
},
Name: []byte("test-bucket"),
})
require.NoError(t, err)
require.Empty(t, response.Location)
_, err = satellite.DB.Buckets().UpdateBucket(ctx, buckets.Bucket{
ProjectID: planet.Uplinks[0].Projects[0].ID,
Name: "test-bucket",
Placement: storj.PlacementConstraint(40),
})
require.NoError(t, err)
// bucket with location
response, err = satellite.API.Metainfo.Endpoint.GetBucketLocation(ctx, &pb.GetBucketLocationRequest{
Header: &pb.RequestHeader{
ApiKey: apiKey.SerializeRaw(),
},
Name: []byte("test-bucket"),
})
require.NoError(t, err)
require.Equal(t, []byte("Poland"), response.Location)
})
}

7
satellite/metainfo/endpoint_test.go
View File

@ -217,6 +217,13 @@ func TestEndpoint_NoStorageNodes(t *testing.T) {
_, err = client.GetBucket(ctx, metaclient.GetBucketParams{})
assertInvalidArgument(t, err, false)
_, err = planet.Satellites[0].Metainfo.Endpoint.GetBucketLocation(ctx, &pb.GetBucketLocationRequest{
Header: &pb.RequestHeader{
ApiKey: []byte(invalidAPIKey),
},
})
assertInvalidArgument(t, err, false)
_, err = client.GetObject(ctx, metaclient.GetObjectParams{})
assertInvalidArgument(t, err, false)