feat(transport/bytestream): Add Close method for shutdown

transport/bytestream/client.go

@@ -33,13 +33,15 @@ const (
 type Client struct {
 	client  pb.ByteStreamClient
 	options []grpc.CallOption
+	conn    *grpc.ClientConn
 }
 
 // NewClient creates a new bytestream.Client.
 func NewClient(cc *grpc.ClientConn, options ...grpc.CallOption) *Client {
 	return &Client{
 		client:  pb.NewByteStreamClient(cc),
 		options: options,
+		conn:    cc,
 	}
 }
 
@@ -228,3 +230,9 @@ func (c *Client) NewWriter(ctx context.Context, resourceName string) (*Writer, e
 		resourceName: resourceName,
 	}, nil
 }
+
+// Close closes the connection to the API service. The user should invoke this when
+// the client is no longer required.
+func (c *Client) Close() {
+	c.conn.Close()
+}

transport/bytestream/client_test.go

@@ -307,7 +307,6 @@ func TestClientWrite_WriteFails(t *testing.T) {
 }
 
 func TestClientWrite_CloseAndRecvFails(t *testing.T) {
-	t.Skip("https://github.com/googleapis/google-api-go-client/issues/775")
 	setup := newTestSetup("")
 	w, err := setup.client.NewWriter(setup.ctx, "CloseAndRecvFails")
 	if err != nil {
@@ -425,5 +424,6 @@ func newTestSetup(input string) *TestSetup {
 }
 
 func (testSetup *TestSetup) Close() {
+	testSetup.client.Close()
 	testSetup.rpcTest.Close()
 }