Implemented RegisterGatewayServiceClient

Implemented a new function to register a GatewayServiceClient. Previously, this could only be done by passing an actual connection. However, that is unnecessarily limiting. Introducing the RegisterGatewayServiceClient allows us to register mocked gateways for unit testing

Author: jessegeens

pkg/rgrpc/todo/pool/pool.go

@@ -97,6 +97,13 @@ func NewConn(options Options) (*grpc.ClientConn, error) {
 	return conn, nil
 }
 
+func RegisterGatewayServiceClient(client gateway.GatewayAPIClient, endpoint string) {
+	gatewayProviders.m.Lock()
+	defer gatewayProviders.m.Unlock()
+
+	gatewayProviders.conn[endpoint] = client
+}
+
 // GetGatewayServiceClient returns a GatewayServiceClient.
 func GetGatewayServiceClient(opts ...Option) (gateway.GatewayAPIClient, error) {
 	gatewayProviders.m.Lock()