From 8dd9d1cadcedb9e441fe5dc4c3249eaf274caec6 Mon Sep 17 00:00:00 2001 From: Jaime Martinez Date: Mon, 7 Jun 2021 12:02:57 +1000 Subject: Use gitlab stub in config_test.go --- test/acceptance/config_test.go | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) (limited to 'test') diff --git a/test/acceptance/config_test.go b/test/acceptance/config_test.go index 93e9aa22..5db00696 100644 --- a/test/acceptance/config_test.go +++ b/test/acceptance/config_test.go @@ -4,7 +4,6 @@ import ( "fmt" "net" "net/http" - "os" "testing" "github.com/stretchr/testify/require" @@ -12,11 +11,12 @@ import ( func TestEnvironmentVariablesConfig(t *testing.T) { skipUnlessEnabled(t) - os.Setenv("LISTEN_HTTP", net.JoinHostPort(httpListener.Host, httpListener.Port)) - defer func() { os.Unsetenv("LISTEN_HTTP") }() - teardown := RunPagesProcessWithoutWait(t, *pagesBinary, []ListenSpec{}, "") + envVarValue := "LISTEN_HTTP=" + net.JoinHostPort(httpListener.Host, httpListener.Port) + + _, teardown := RunPagesProcessWithStubGitLabServer(t, false, *pagesBinary, []ListenSpec{}, []string{envVarValue}) defer teardown() + require.NoError(t, httpListener.WaitUntilRequestSucceeds(nil)) rsp, err := GetPageFromListener(t, httpListener, "group.gitlab-example.com:", "project/") @@ -28,10 +28,9 @@ func TestEnvironmentVariablesConfig(t *testing.T) { func TestMixedConfigSources(t *testing.T) { skipUnlessEnabled(t) - os.Setenv("LISTEN_HTTP", net.JoinHostPort(httpListener.Host, httpListener.Port)) - defer func() { os.Unsetenv("LISTEN_HTTP") }() + envVarValue := "LISTEN_HTTP=" + net.JoinHostPort(httpListener.Host, httpListener.Port) - teardown := RunPagesProcessWithoutWait(t, *pagesBinary, []ListenSpec{httpsListener}, "") + _, teardown := RunPagesProcessWithStubGitLabServer(t, false, *pagesBinary, []ListenSpec{httpsListener}, []string{envVarValue}) defer teardown() for _, listener := range []ListenSpec{httpListener, httpsListener} { @@ -44,15 +43,13 @@ func TestMixedConfigSources(t *testing.T) { } } -func TestMultiFlagEnvironmentVariables(t *testing.T) { +func TestMultipleListenersFromEnvironmentVariables(t *testing.T) { skipUnlessEnabled(t) - listenSpecs := []ListenSpec{{"http", "127.0.0.1", "37001"}, {"http", "127.0.0.1", "37002"}} - envVarValue := fmt.Sprintf("%s,%s", net.JoinHostPort("127.0.0.1", "37001"), net.JoinHostPort("127.0.0.1", "37002")) - os.Setenv("LISTEN_HTTP", envVarValue) - defer func() { os.Unsetenv("LISTEN_HTTP") }() + listenSpecs := []ListenSpec{{"http", "127.0.0.1", "37001"}, {"http", "127.0.0.1", "37002"}} + envVarValue := fmt.Sprintf("LISTEN_HTTP=%s,%s", net.JoinHostPort("127.0.0.1", "37001"), net.JoinHostPort("127.0.0.1", "37002")) - teardown := RunPagesProcess(t, *pagesBinary, []ListenSpec{}, "") + _, teardown := RunPagesProcessWithStubGitLabServer(t, false, *pagesBinary, []ListenSpec{}, []string{envVarValue}) defer teardown() for _, listener := range listenSpecs { -- cgit v1.2.3