Skip to content

Commit

Permalink
Add metadata tests for import and export
Browse files Browse the repository at this point in the history
  • Loading branch information
tstirrat15 committed Sep 17, 2024
1 parent 3a52ae7 commit 0e7bb64
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions internal/services/v1/metadata_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -173,6 +173,36 @@ func TestAllMethodsReturnMetadata(t *testing.T) {
require.NoError(t, err)
}

return trailer
},
"ImportBulkRelationships": func(t *testing.T, client v1.PermissionsServiceClient) metadata.MD {
var trailer metadata.MD
writer, err := client.ImportBulkRelationships(ctx, grpc.Trailer(&trailer))
require.NoError(t, err)
_, err = writer.CloseAndRecv()
require.NoError(t, err)
return trailer
},
"ExportBulkRelationships": func(t *testing.T, client v1.PermissionsServiceClient) metadata.MD {
var trailer metadata.MD
stream, err := client.ExportBulkRelationships(ctx, &v1.ExportBulkRelationshipsRequest{
Consistency: &v1.Consistency{
Requirement: &v1.Consistency_AtLeastAsFresh{
AtLeastAsFresh: zedtoken.MustNewFromRevision(revision),
},
},
}, grpc.Trailer(&trailer))
require.NoError(t, err)

for {
_, err := stream.Recv()
if errors.Is(err, io.EOF) {
break
}

require.NoError(t, err)
}

return trailer
},
},
Expand Down Expand Up @@ -214,6 +244,9 @@ func checkServiceMethods[T any](
methodName := et.Method(i).Name
t.Run(methodName, func(t *testing.T) {
handler, ok := handlers[methodName]
if !ok {
return
}
require.True(t, ok, "missing handler for method %s under %T", methodName, new(T))

trailer := handler(t, client)
Expand Down

0 comments on commit 0e7bb64

Please sign in to comment.