Add extra host to start command (#369)

* Add extra host to start command Signed-off-by: Kevin Su <[email protected]> * Fix tests Signed-off-by: Kevin Su <[email protected]> * Fix tests Signed-off-by: Kevin Su <[email protected]> * Make generate Signed-off-by: Kevin Su <[email protected]> * update Signed-off-by: Kevin Su <[email protected]> Signed-off-by: Kevin Su <[email protected]>
flyteorg · Nov 10, 2022 · 209fe93 · 209fe93
1 parent 77deca5
commit 209fe93
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/docker/docker_util.go b/pkg/docker/docker_util.go
@@ -47,6 +47,7 @@ var (
 	}
 	StdWriterPrefixLen = 8
 	StartingBufLen     = 32*1024 + StdWriterPrefixLen + 1
+	ExtraHosts         = []string{"host.docker.internal:127.0.0.1"}
 )
 
 // GetDockerClient will returns the docker client
@@ -237,6 +238,7 @@ func StartContainer(ctx context.Context, cli Docker, volumes []mount.Mount, expo
 		Mounts:       volumes,
 		PortBindings: portBindings,
 		Privileged:   true,
+		ExtraHosts:   ExtraHosts, // add it because linux machine doesn't have this host name by default
 	}, nil,
 		nil, name)
 

diff --git a/pkg/docker/docker_util_test.go b/pkg/docker/docker_util_test.go
@@ -156,6 +156,7 @@ func TestStartContainer(t *testing.T) {
 			Mounts:       Volumes,
 			PortBindings: p2,
 			Privileged:   true,
+			ExtraHosts:   ExtraHosts,
 		}, nil, nil, mock.Anything).Return(container.ContainerCreateCreatedBody{
 			ID: "Hello",
 		}, nil)
@@ -185,6 +186,7 @@ func TestStartContainer(t *testing.T) {
 			Mounts:       Volumes,
 			PortBindings: p2,
 			Privileged:   true,
+			ExtraHosts:   ExtraHosts,
 		}, nil, nil, mock.Anything).Return(container.ContainerCreateCreatedBody{
 			ID: "Hello",
 		}, nil)
@@ -211,6 +213,7 @@ func TestStartContainer(t *testing.T) {
 			Mounts:       Volumes,
 			PortBindings: p2,
 			Privileged:   true,
+			ExtraHosts:   ExtraHosts,
 		}, nil, nil, mock.Anything).Return(container.ContainerCreateCreatedBody{
 			ID: "",
 		}, fmt.Errorf("error"))
@@ -236,6 +239,7 @@ func TestStartContainer(t *testing.T) {
 			Mounts:       Volumes,
 			PortBindings: p2,
 			Privileged:   true,
+			ExtraHosts:   ExtraHosts,
 		}, nil, nil, mock.Anything).Return(container.ContainerCreateCreatedBody{
 			ID: "Hello",
 		}, nil)