package main
 
 import (
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"os"
 	"path/filepath"

 	"runtime"

 	"github.com/docker/docker/api"

 	"github.com/docker/docker/dockerversion"

 	"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"))

 	c.Assert(headers["User-Agent"][0], checker.Equals, "Docker-Client/"+dockerversion.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"))

 }