-
Notifications
You must be signed in to change notification settings - Fork 168
/
get_store.go
78 lines (64 loc) · 2.43 KB
/
get_store.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package test
import (
"context"
"testing"
"github.com/google/go-cmp/cmp"
openfgav1 "github.com/openfga/api/proto/openfga/v1"
"github.com/openfga/openfga/pkg/server/commands"
serverErrors "github.com/openfga/openfga/pkg/server/errors"
"github.com/openfga/openfga/pkg/storage"
"github.com/openfga/openfga/pkg/testutils"
"github.com/stretchr/testify/require"
"google.golang.org/protobuf/protoadapt"
"google.golang.org/protobuf/testing/protocmp"
)
func TestGetStoreQuery(t *testing.T, datastore storage.OpenFGADatastore) {
type getStoreQueryTest struct {
_name string
request *openfgav1.GetStoreRequest
expectedResponse *openfgav1.GetStoreResponse
err error
}
var tests = []getStoreQueryTest{
{
_name: "ReturnsNotFound",
request: &openfgav1.GetStoreRequest{StoreId: "non-existent store"},
err: serverErrors.StoreIDNotFound,
},
}
ignoreStoreFields := protocmp.IgnoreFields(protoadapt.MessageV2Of(&openfgav1.GetStoreResponse{}), "created_at", "updated_at", "id")
ctx := context.Background()
for _, test := range tests {
t.Run(test._name, func(t *testing.T) {
query := commands.NewGetStoreQuery(datastore)
resp, err := query.Execute(ctx, test.request)
if test.err != nil {
require.ErrorIs(t, err, test.err)
require.Nil(t, resp)
} else {
require.NoError(t, err)
if diff := cmp.Diff(test.expectedResponse, resp, ignoreStoreFields, protocmp.Transform()); diff != "" {
t.Errorf("store mismatch (-want +got):\n%s", diff)
}
}
})
}
}
func TestGetStoreSucceeds(t *testing.T, datastore storage.OpenFGADatastore) {
ctx := context.Background()
store := testutils.CreateRandomString(10)
createStoreQuery := commands.NewCreateStoreCommand(datastore)
createStoreResponse, err := createStoreQuery.Execute(ctx, &openfgav1.CreateStoreRequest{Name: store})
require.NoError(t, err)
query := commands.NewGetStoreQuery(datastore)
actualResponse, actualError := query.Execute(ctx, &openfgav1.GetStoreRequest{StoreId: createStoreResponse.Id})
require.NoError(t, actualError)
expectedResponse := &openfgav1.GetStoreResponse{
Id: createStoreResponse.Id,
Name: store,
}
ignoreStoreFields := protocmp.IgnoreFields(protoadapt.MessageV2Of(&openfgav1.GetStoreResponse{}), "created_at", "updated_at", "id")
if diff := cmp.Diff(expectedResponse, actualResponse, ignoreStoreFields, protocmp.Transform()); diff != "" {
t.Errorf("mismatch (-want +got):\n%s", diff)
}
}