Merge pull request #252 from authzed/flesh-out-tests

Flesh out client tests

.github/workflows/test.yaml

@@ -6,6 +6,11 @@ on:
       - "main"
   pull_request:
     branches: ["*"]
+  # NOTE: this is required for checks to be run in the
+  # merge queue.
+  merge_group:
+    types:
+      - "checks_requested"
 jobs:
   tests:
     name: "Unit and Integration Tests"

v1/client.go

@@ -11,11 +11,18 @@ import (
 //
 // Clients are backed by a gRPC client and as such are thread-safe.
 type Client struct {
+	// Provide a handle on the underlying connection to enable cleanup
+	// behaviors (among others)
+	conn *grpc.ClientConn
 	v1.SchemaServiceClient
 	v1.PermissionsServiceClient
 	v1.WatchServiceClient
 }
 
+func (c *Client) Close() error {
+	return c.conn.Close()
+}
+
 // ClientWithExperimental represents and open connection to Authzed with
 // experimental services available.
 //
@@ -34,6 +41,7 @@ func NewClient(endpoint string, opts ...grpc.DialOption) (*Client, error) {
 	}
 
 	return &Client{
+		conn,
 		v1.NewSchemaServiceClient(conn),
 		v1.NewPermissionsServiceClient(conn),
 		v1.NewWatchServiceClient(conn),
@@ -50,6 +58,7 @@ func NewClientWithExperimentalAPIs(endpoint string, opts ...grpc.DialOption) (*C
 
 	return &ClientWithExperimental{
 		Client{
+			conn,
 			v1.NewSchemaServiceClient(conn),
 			v1.NewPermissionsServiceClient(conn),
 			v1.NewWatchServiceClient(conn),