test: add withExtraHosts tests

orlangure · Dec 21, 2022 · 488e545 · 488e545
1 parent c2e37d8
commit 488e545
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/gnomock_test.go b/gnomock_test.go
@@ -268,6 +268,44 @@ func TestGnomock_witUseLocalImagesFirst(t *testing.T) {
 	require.NoError(t, gnomock.Stop(container))
 }
 
+func TestGnomock_withExtraHosts(t *testing.T) {
+	t.Parallel()
+
+	const (
+		busyboxImage = "docker.io/library/busybox:1.35.0"
+		retries      = "5"
+	)
+
+	container, err := gnomock.StartCustom(
+		busyboxImage,
+		gnomock.DefaultTCP(testutil.GoodPort80),
+		gnomock.WithExtraHosts([]string{"test:127.0.0.1"}),
+		gnomock.WithCommand("ping", "-c", retries, "test"),
+	)
+	require.NoError(t, err)
+	require.NotNil(t, container)
+	require.NoError(t, gnomock.Stop(container))
+
+	container, err = gnomock.StartCustom(
+		busyboxImage,
+		gnomock.DefaultTCP(testutil.GoodPort80),
+		gnomock.WithCommand("ping", "-c", retries, "test"),
+	)
+	require.Error(t, err)
+	require.Nil(t, container)
+	require.NoError(t, gnomock.Stop(container))
+
+	container, err = gnomock.StartCustom(
+		busyboxImage,
+		gnomock.DefaultTCP(testutil.GoodPort80),
+		gnomock.WithExtraHosts([]string{"test:127.0.0.1"}),
+		gnomock.WithCommand("ping", "-c", retries, "google.com"),
+	)
+	require.NoError(t, err)
+	require.NotNil(t, container)
+	require.NoError(t, gnomock.Stop(container))
+}
+
 func initf(context.Context, *gnomock.Container) error {
 	return nil
 }