backport of commit 4c9f476 (#12687)

Co-authored-by: Thomas Lefebvre <tlefebvre@cloudflare.com>
hashicorp · Apr 19, 2022 · 4a1656f · 4a1656f
1 parent 9d79397
commit 4a1656f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/drivers/mock/driver.go b/drivers/mock/driver.go
@@ -670,8 +670,10 @@ func (d *Driver) GetHandle(taskID string) *taskHandle {
 	return h
 }
 
-func (d *Driver) CreateNetwork(allocID string) (*drivers.NetworkIsolationSpec, error) {
-	return nil, nil
+var _ drivers.DriverNetworkManager = (*Driver)(nil)
+
+func (d *Driver) CreateNetwork(allocID string, request *drivers.NetworkCreateRequest) (*drivers.NetworkIsolationSpec, bool, error) {
+	return nil, true, nil
 }
 
 func (d *Driver) DestroyNetwork(allocID string, spec *drivers.NetworkIsolationSpec) error {

diff --git a/plugins/drivers/client.go b/plugins/drivers/client.go
@@ -468,7 +468,9 @@ func (d *driverPluginClient) ExecTaskStreamingRaw(ctx context.Context,
 	}
 }
 
-func (d *driverPluginClient) CreateNetwork(allocID string) (*NetworkIsolationSpec, bool, error) {
+var _ DriverNetworkManager = (*driverPluginClient)(nil)
+
+func (d *driverPluginClient) CreateNetwork(allocID string, _ *NetworkCreateRequest) (*NetworkIsolationSpec, bool, error) {
 	req := &proto.CreateNetworkRequest{
 		AllocId: allocID,
 	}