84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package frostfs
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/layer"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/middleware"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/authmate"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/creds/accessbox"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-s3-gw/creds/tokens"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/bearer"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
)
|
|
|
|
func TestGetCredsObject(t *testing.T) {
|
|
ctx, bktName, payload, newPayload := context.Background(), "bucket", []byte("payload"), []byte("new-payload")
|
|
|
|
key, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
|
|
var userID user.ID
|
|
userID.SetScriptHash(key.PublicKey().GetScriptHash())
|
|
|
|
var token bearer.Token
|
|
err = token.Sign(key.PrivateKey)
|
|
require.NoError(t, err)
|
|
|
|
ctx = middleware.SetBox(ctx, &middleware.Box{AccessBox: &accessbox.Box{
|
|
Gate: &accessbox.GateData{
|
|
BearerToken: &token,
|
|
},
|
|
}})
|
|
|
|
frostfs := NewAuthmateFrostFS(layer.NewTestFrostFS(key), zaptest.NewLogger(t))
|
|
|
|
cnrID, err := frostfs.CreateContainer(ctx, authmate.PrmContainerCreate{
|
|
FriendlyName: bktName,
|
|
Owner: userID,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
objID, err := frostfs.CreateObject(ctx, tokens.PrmObjectCreate{
|
|
Container: cnrID,
|
|
Payload: payload,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
var addr oid.Address
|
|
addr.SetContainer(cnrID)
|
|
addr.SetObject(objID)
|
|
|
|
accessKeyID := getAccessKeyID(addr)
|
|
|
|
obj, err := frostfs.GetCredsObject(ctx, tokens.PrmGetCredsObject{
|
|
Container: cnrID,
|
|
AccessKeyID: accessKeyID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, payload, obj.Payload())
|
|
|
|
_, err = frostfs.CreateObject(ctx, tokens.PrmObjectCreate{
|
|
Container: cnrID,
|
|
Payload: newPayload,
|
|
NewVersionForAccessKeyID: accessKeyID,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
obj, err = frostfs.GetCredsObject(ctx, tokens.PrmGetCredsObject{
|
|
Container: cnrID,
|
|
AccessKeyID: getAccessKeyID(addr),
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, newPayload, obj.Payload())
|
|
}
|
|
|
|
func getAccessKeyID(addr oid.Address) string {
|
|
return strings.ReplaceAll(addr.EncodeToString(), "/", "0")
|
|
}
|