mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-11 17:04:22 -05:00
b461c40c3a
Signed-off-by: Daniel Weiße <dw@edgeless.systems>
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package kms
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/kms/server/kmsapi/kmsproto"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/test/bufconn"
|
|
)
|
|
|
|
type stubClient struct {
|
|
getDataKeyErr error
|
|
dataKey []byte
|
|
}
|
|
|
|
func (c *stubClient) GetDataKey(context.Context, *kmsproto.GetDataKeyRequest, *grpc.ClientConn) (*kmsproto.GetDataKeyResponse, error) {
|
|
return &kmsproto.GetDataKeyResponse{DataKey: c.dataKey}, c.getDataKeyErr
|
|
}
|
|
|
|
func TestGetDataKey(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
client *stubClient
|
|
wantErr bool
|
|
}{
|
|
"GetDataKey success": {
|
|
client: &stubClient{dataKey: []byte{0x1, 0x2, 0x3}},
|
|
},
|
|
"GetDataKey error": {
|
|
client: &stubClient{getDataKeyErr: errors.New("error")},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
listener := bufconn.Listen(1)
|
|
defer listener.Close()
|
|
|
|
client := New(listener.Addr().String())
|
|
|
|
client.grpc = tc.client
|
|
|
|
res, err := client.GetDataKey(context.Background(), "disk-uuid", 32)
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
} else {
|
|
assert.NoError(err)
|
|
assert.Equal(tc.client.dataKey, res)
|
|
}
|
|
})
|
|
}
|
|
}
|