storj/satellite/admin/bucket_testplanet_test.go

117 lines
3.2 KiB
Go
Raw Normal View History

// Copyright (C) 2021 Storj Labs, Inc.
// See LICENSE for copying information.
package admin_test
import (
"encoding/json"
"fmt"
"net/http"
"testing"
"github.com/stretchr/testify/require"
"go.uber.org/zap"
"storj.io/common/storj"
"storj.io/common/testcontext"
"storj.io/common/uuid"
"storj.io/storj/private/testplanet"
"storj.io/storj/satellite"
"storj.io/storj/satellite/buckets"
)
func TestAdminBucketGeofenceAPI(t *testing.T) {
testplanet.Run(t, testplanet.Config{
SatelliteCount: 1,
StorageNodeCount: 0,
UplinkCount: 1,
Reconfigure: testplanet.Reconfigure{
Satellite: func(_ *zap.Logger, _ int, config *satellite.Config) {
config.Admin.Address = "127.0.0.1:0"
},
},
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
uplink := planet.Uplinks[0]
sat := planet.Satellites[0]
address := sat.Admin.Admin.Listener.Addr()
project, err := sat.DB.Console().Projects().Get(ctx, uplink.Projects[0].ID)
require.NoError(t, err)
err = uplink.CreateBucket(ctx, sat, "filled")
require.NoError(t, err)
_, err = sat.DB.Buckets().UpdateBucket(ctx, buckets.Bucket{
Name: "filled",
ProjectID: project.ID,
Placement: storj.EEA,
})
require.NoError(t, err)
err = uplink.Upload(ctx, sat, "filled", "README.md", []byte("hello world"))
require.NoError(t, err)
err = uplink.CreateBucket(ctx, sat, "empty")
require.NoError(t, err)
testCases := []struct {
name string
project uuid.UUID
bucket []byte
// expectations
status int
body string
}{
{
name: "bucket does not exist",
project: project.ID,
bucket: []byte("non-existent"),
status: http.StatusBadRequest,
body: `{"error":"bucket does not exist","detail":""}`,
},
{
name: "bucket is not empty",
project: project.ID,
bucket: []byte("filled"),
status: http.StatusBadRequest,
body: `{"error":"bucket must be empty","detail":""}`,
},
{
name: "validated",
project: project.ID,
bucket: []byte("empty"),
status: http.StatusOK,
body: "",
},
}
for _, testCase := range testCases {
baseURL := fmt.Sprintf("http://%s/api/projects/%s/buckets/%s", address, testCase.project, string(testCase.bucket))
t.Log(baseURL)
baseGeofenceURL := fmt.Sprintf("http://%s/api/projects/%s/buckets/%s/geofence", address, testCase.project, string(testCase.bucket))
t.Log(baseGeofenceURL)
t.Run(testCase.name, func(t *testing.T) {
assertReq(ctx, t, baseGeofenceURL+"?region=EU", "POST", "", testCase.status, testCase.body, sat.Config.Console.AuthToken)
if testCase.status == http.StatusOK {
b, err := sat.DB.Buckets().GetBucket(ctx, testCase.bucket, testCase.project)
require.NoError(t, err)
expected, err := json.Marshal(buckets.Bucket{
ID: b.ID,
Name: b.Name,
ProjectID: testCase.project,
Created: b.Created,
Placement: storj.EU,
})
require.NoError(t, err, "failed to json encode expected bucket")
assertGet(ctx, t, baseURL, string(expected), sat.Config.Console.AuthToken)
}
assertReq(ctx, t, baseGeofenceURL, "DELETE", "", testCase.status, testCase.body, sat.Config.Console.AuthToken)
})
}
})
}