add DriverFromClient constructor

Signed-off-by: lwsanty <[email protected]>
bblfsh · Nov 14, 2019 · 37abd69 · 37abd69
1 parent 36bfb4c
commit 37abd69
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/protocol/driver.go b/protocol/driver.go
@@ -79,9 +79,13 @@ func RegisterDriver(srv *grpc.Server, d driver.Driver) {
 
 // AsDriver creates a v2 driver client for a given gRPC client.
 func AsDriver(cc *grpc.ClientConn) driver.Driver {
+	return DriverFromClient(NewDriverClient(cc), NewDriverHostClient(cc))
+}
+
+func DriverFromClient(c DriverClient, h DriverHostClient) driver.Driver {
 	return &client{
-		c: NewDriverClient(cc),
-		h: NewDriverHostClient(cc),
+		c: c,
+		h: h,
 	}
 }