mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-06 05:46:05 +02:00
60 lines
2.1 KiB
Go
60 lines
2.1 KiB
Go
package webauthnutil
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/pomerium/webauthn"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
)
|
|
|
|
type mockDataBrokerServiceClient struct {
|
|
databroker.DataBrokerServiceClient
|
|
|
|
get func(ctx context.Context, in *databroker.GetRequest, opts ...grpc.CallOption) (*databroker.GetResponse, error)
|
|
put func(ctx context.Context, in *databroker.PutRequest, opts ...grpc.CallOption) (*databroker.PutResponse, error)
|
|
}
|
|
|
|
func (m mockDataBrokerServiceClient) Get(ctx context.Context, in *databroker.GetRequest, opts ...grpc.CallOption) (*databroker.GetResponse, error) {
|
|
return m.get(ctx, in, opts...)
|
|
}
|
|
|
|
func (m mockDataBrokerServiceClient) Put(ctx context.Context, in *databroker.PutRequest, opts ...grpc.CallOption) (*databroker.PutResponse, error) {
|
|
return m.put(ctx, in, opts...)
|
|
}
|
|
|
|
func TestCredentialStorage(t *testing.T) {
|
|
m := map[string]*databroker.Record{}
|
|
client := &mockDataBrokerServiceClient{
|
|
get: func(ctx context.Context, in *databroker.GetRequest, opts ...grpc.CallOption) (*databroker.GetResponse, error) {
|
|
record, ok := m[in.GetType()+"/"+in.GetId()]
|
|
if !ok {
|
|
return nil, status.Error(codes.NotFound, "record not found")
|
|
}
|
|
return &databroker.GetResponse{
|
|
Record: record,
|
|
}, nil
|
|
},
|
|
put: func(ctx context.Context, in *databroker.PutRequest, opts ...grpc.CallOption) (*databroker.PutResponse, error) {
|
|
m[in.GetRecord().GetType()+"/"+in.GetRecord().GetId()] = in.GetRecord()
|
|
return &databroker.PutResponse{
|
|
Record: in.GetRecord(),
|
|
}, nil
|
|
},
|
|
}
|
|
storage := NewCredentialStorage(client)
|
|
_, err := storage.GetCredential(context.Background(), []byte{0, 1, 2, 3, 4})
|
|
assert.ErrorIs(t, err, webauthn.ErrCredentialNotFound)
|
|
err = storage.SetCredential(context.Background(), &webauthn.Credential{
|
|
ID: []byte{0, 1, 2, 3, 4},
|
|
})
|
|
assert.NoError(t, err)
|
|
c, err := storage.GetCredential(context.Background(), []byte{0, 1, 2, 3, 4})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, []byte{0, 1, 2, 3, 4}, c.ID)
|
|
}
|