package main import ( "io/ioutil" "net/http" "net/http/httptest" "os" "path/filepath" "runtime" "github.com/docker/docker/api" "github.com/docker/docker/integration-cli/checker" "github.com/docker/docker/pkg/homedir" icmd "github.com/docker/docker/pkg/testutil/cmd" "github.com/go-check/check" ) func (s *DockerSuite) TestConfigHTTPHeader(c *check.C) { testRequires(c, UnixCli) // Can't set/unset HOME on windows right now // We either need a level of Go that supports Unsetenv (for cases // when HOME/USERPROFILE isn't set), or we need to be able to use // os/user but user.Current() only works if we aren't statically compiling var headers map[string][]string server := httptest.NewServer(http.HandlerFunc( func(w http.ResponseWriter, r *http.Request) { w.Header().Set("API-Version", api.DefaultVersion) headers = r.Header })) defer server.Close() homeKey := homedir.Key() homeVal := homedir.Get() tmpDir, err := ioutil.TempDir("", "fake-home") c.Assert(err, checker.IsNil) defer os.RemoveAll(tmpDir) dotDocker := filepath.Join(tmpDir, ".docker") os.Mkdir(dotDocker, 0600) tmpCfg := filepath.Join(dotDocker, "config.json") defer func() { os.Setenv(homeKey, homeVal) }() os.Setenv(homeKey, tmpDir) data := `{ "HttpHeaders": { "MyHeader": "MyValue" } }` err = ioutil.WriteFile(tmpCfg, []byte(data), 0600) c.Assert(err, checker.IsNil) result := icmd.RunCommand(dockerBinary, "-H="+server.URL[7:], "ps") result.Assert(c, icmd.Expected{ ExitCode: 1, Error: "exit status 1", }) c.Assert(headers["User-Agent"], checker.NotNil, check.Commentf("Missing User-Agent")) //TODO(tiborvass): restore dockerversion.Version instead of library-import c.Assert(headers["User-Agent"][0], checker.Equals, "Docker-Client/unknown-version ("+runtime.GOOS+")", check.Commentf("Badly formatted User-Agent,out:%v", result.Combined())) c.Assert(headers["Myheader"], checker.NotNil) c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("Missing/bad header,out:%v", result.Combined())) } func (s *DockerSuite) TestConfigDir(c *check.C) { cDir, err := ioutil.TempDir("", "fake-home") c.Assert(err, checker.IsNil) defer os.RemoveAll(cDir) // First make sure pointing to empty dir doesn't generate an error dockerCmd(c, "--config", cDir, "ps") // Test with env var too icmd.RunCmd(icmd.Cmd{ Command: []string{dockerBinary, "ps"}, Env: appendBaseEnv(true, "DOCKER_CONFIG="+cDir), }).Assert(c, icmd.Success) // Start a server so we can check to see if the config file was // loaded properly var headers map[string][]string server := httptest.NewServer(http.HandlerFunc( func(w http.ResponseWriter, r *http.Request) { headers = r.Header })) defer server.Close() // Create a dummy config file in our new config dir data := `{ "HttpHeaders": { "MyHeader": "MyValue" } }` tmpCfg := filepath.Join(cDir, "config.json") err = ioutil.WriteFile(tmpCfg, []byte(data), 0600) c.Assert(err, checker.IsNil, check.Commentf("Err creating file")) env := appendBaseEnv(false) icmd.RunCmd(icmd.Cmd{ Command: []string{dockerBinary, "--config", cDir, "-H=" + server.URL[7:], "ps"}, Env: env, }).Assert(c, icmd.Expected{ ExitCode: 1, Error: "exit status 1", }) c.Assert(headers["Myheader"], checker.NotNil) c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps3 - Missing header")) // Reset headers and try again using env var this time headers = map[string][]string{} icmd.RunCmd(icmd.Cmd{ Command: []string{dockerBinary, "--config", cDir, "-H=" + server.URL[7:], "ps"}, Env: append(env, "DOCKER_CONFIG="+cDir), }).Assert(c, icmd.Expected{ ExitCode: 1, }) c.Assert(headers["Myheader"], checker.NotNil) c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps4 - Missing header")) // FIXME(vdemeester) should be a unit test // Reset headers and make sure flag overrides the env var headers = map[string][]string{} icmd.RunCmd(icmd.Cmd{ Command: []string{dockerBinary, "--config", cDir, "-H=" + server.URL[7:], "ps"}, Env: append(env, "DOCKER_CONFIG=MissingDir"), }).Assert(c, icmd.Expected{ ExitCode: 1, }) c.Assert(headers["Myheader"], checker.NotNil) c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps5 - Missing header")) // FIXME(vdemeester) should be a unit test // Reset headers and make sure flag overrides the env var. // Almost same as previous but make sure the "MissingDir" isn't // ignore - we don't want to default back to the env var. headers = map[string][]string{} icmd.RunCmd(icmd.Cmd{ Command: []string{dockerBinary, "--config", "MissingDir", "-H=" + server.URL[7:], "ps"}, Env: append(env, "DOCKER_CONFIG="+cDir), }).Assert(c, icmd.Expected{ ExitCode: 1, Error: "exit status 1", }) c.Assert(headers["Myheader"], checker.IsNil, check.Commentf("ps6 - Headers shouldn't be the expected value")) }