diff --git a/cli/main.go b/cli/main.go index 6a717ed5..13ea7911 100644 --- a/cli/main.go +++ b/cli/main.go @@ -24,6 +24,7 @@ import ( "os/signal" "path/filepath" "regexp" + "strings" "syscall" "time" @@ -73,13 +74,21 @@ func init() { if err != nil { fatal(errors.Wrap(err, "unable to get absolute bin path")) } - if err := os.Setenv("PATH", fmt.Sprintf("%s:%s", os.Getenv("PATH"), path)); err != nil { + + if err := os.Setenv("PATH", appendPaths(os.Getenv("PATH"), path)); err != nil { panic(err) } // Seed random rand.Seed(time.Now().UnixNano()) } +func appendPaths(envPath string, path string) string { + if envPath == "" { + return path + } + return strings.Join([]string{envPath, path}, string(os.PathListSeparator)) +} + func isContextAgnosticCommand(cmd *cobra.Command) bool { if cmd == nil { return false diff --git a/cli/main_test.go b/cli/main_test.go index c151ce0d..20426de3 100644 --- a/cli/main_test.go +++ b/cli/main_test.go @@ -69,3 +69,8 @@ func TestCheckOwnCommand(t *testing.T) { assert.Assert(t, !isContextAgnosticCommand(cmd.LogsCommand())) assert.Assert(t, !isContextAgnosticCommand(cmd.PsCommand())) } + +func TestAppendPaths(t *testing.T) { + assert.Equal(t, appendPaths("", "/bin/path"), "/bin/path") + assert.Equal(t, appendPaths("path1", "binaryPath"), "path1"+string(os.PathListSeparator)+"binaryPath") +} diff --git a/tests/aci-e2e/e2e-aci_test.go b/tests/aci-e2e/e2e-aci_test.go index 9e77e257..759afb19 100644 --- a/tests/aci-e2e/e2e-aci_test.go +++ b/tests/aci-e2e/e2e-aci_test.go @@ -434,14 +434,7 @@ func TestContainerRunAttached(t *testing.T) { endpoint = fmt.Sprintf("http://%s:%d", fqdn, port.HostPort) assert.Assert(t, !strings.Contains(followLogsProcess.Stdout(), "/test")) - checkRequest := func(t poll.LogT) poll.Result { - r, _ := http.Get(endpoint + "/test") - if r != nil && r.StatusCode == http.StatusNotFound { - return poll.Success() - } - return poll.Continue("waiting for container to serve request") - } - poll.WaitOn(t, checkRequest, poll.WithDelay(1*time.Second), poll.WithTimeout(60*time.Second)) + HTTPGetWithRetry(t, endpoint+"/test", http.StatusNotFound, 2*time.Second, 60*time.Second) checkLog := func(t poll.LogT) poll.Result { if strings.Contains(followLogsProcess.Stdout(), "/test") {