소스 검색

Use of checkers on docker_cli_config_test.go.

Signed-off-by: liaoqingwei <liaoqingwei@huawei.com>
liaoqingwei 9 년 전
부모
커밋
d1ed5f62ad
1개의 변경된 파일28개의 추가작업 그리고 42개의 파일을 삭제
  1. 28 42
      integration-cli/docker_cli_config_test.go

+ 28 - 42
integration-cli/docker_cli_config_test.go

@@ -11,6 +11,7 @@ import (
 
 
 	"github.com/docker/docker/autogen/dockerversion"
 	"github.com/docker/docker/autogen/dockerversion"
 	"github.com/docker/docker/pkg/homedir"
 	"github.com/docker/docker/pkg/homedir"
+	"github.com/docker/docker/pkg/integration/checker"
 	"github.com/go-check/check"
 	"github.com/go-check/check"
 )
 )
 
 
@@ -31,7 +32,7 @@ func (s *DockerSuite) TestConfigHttpHeader(c *check.C) {
 	homeKey := homedir.Key()
 	homeKey := homedir.Key()
 	homeVal := homedir.Get()
 	homeVal := homedir.Get()
 	tmpDir, err := ioutil.TempDir("", "fake-home")
 	tmpDir, err := ioutil.TempDir("", "fake-home")
-	c.Assert(err, check.IsNil)
+	c.Assert(err, checker.IsNil)
 	defer os.RemoveAll(tmpDir)
 	defer os.RemoveAll(tmpDir)
 
 
 	dotDocker := filepath.Join(tmpDir, ".docker")
 	dotDocker := filepath.Join(tmpDir, ".docker")
@@ -46,46 +47,34 @@ func (s *DockerSuite) TestConfigHttpHeader(c *check.C) {
 	}`
 	}`
 
 
 	err = ioutil.WriteFile(tmpCfg, []byte(data), 0600)
 	err = ioutil.WriteFile(tmpCfg, []byte(data), 0600)
-	if err != nil {
-		c.Fatalf("Err creating file(%s): %v", tmpCfg, err)
-	}
+	c.Assert(err, checker.IsNil)
 
 
 	cmd := exec.Command(dockerBinary, "-H="+server.URL[7:], "ps")
 	cmd := exec.Command(dockerBinary, "-H="+server.URL[7:], "ps")
 	out, _, _ := runCommandWithOutput(cmd)
 	out, _, _ := runCommandWithOutput(cmd)
 
 
-	if headers["User-Agent"] == nil {
-		c.Fatalf("Missing User-Agent: %q\nout:%v", headers, out)
-	}
+	c.Assert(headers["User-Agent"], checker.NotNil, check.Commentf("Missing User-Agent"))
 
 
-	if headers["User-Agent"][0] != "Docker-Client/"+dockerversion.VERSION+" ("+runtime.GOOS+")" {
-		c.Fatalf("Badly formatted User-Agent: %q\nout:%v", headers, out)
-	}
+	c.Assert(headers["User-Agent"][0], checker.Equals, "Docker-Client/"+dockerversion.VERSION+" ("+runtime.GOOS+")", check.Commentf("Badly formatted User-Agent,out:%v", out))
+
+	c.Assert(headers["Myheader"], checker.NotNil)
+	c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("Missing/bad header,out:%v", out))
 
 
-	if headers["Myheader"] == nil || headers["Myheader"][0] != "MyValue" {
-		c.Fatalf("Missing/bad header: %q\nout:%v", headers, out)
-	}
 }
 }
 
 
 func (s *DockerSuite) TestConfigDir(c *check.C) {
 func (s *DockerSuite) TestConfigDir(c *check.C) {
 	cDir, err := ioutil.TempDir("", "fake-home")
 	cDir, err := ioutil.TempDir("", "fake-home")
-	c.Assert(err, check.IsNil)
+	c.Assert(err, checker.IsNil)
 	defer os.RemoveAll(cDir)
 	defer os.RemoveAll(cDir)
 
 
 	// First make sure pointing to empty dir doesn't generate an error
 	// First make sure pointing to empty dir doesn't generate an error
-	out, rc := dockerCmd(c, "--config", cDir, "ps")
-
-	if rc != 0 {
-		c.Fatalf("ps1 didn't work:\nrc:%d\nout%s", rc, out)
-	}
+	dockerCmd(c, "--config", cDir, "ps")
 
 
 	// Test with env var too
 	// Test with env var too
 	cmd := exec.Command(dockerBinary, "ps")
 	cmd := exec.Command(dockerBinary, "ps")
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
-	out, rc, err = runCommandWithOutput(cmd)
+	out, _, err := runCommandWithOutput(cmd)
 
 
-	if rc != 0 || err != nil {
-		c.Fatalf("ps2 didn't work:\nrc:%d\nout%s\nerr:%v", rc, out, err)
-	}
+	c.Assert(err, checker.IsNil, check.Commentf("ps2 didn't work,out:%v", out))
 
 
 	// Start a server so we can check to see if the config file was
 	// Start a server so we can check to see if the config file was
 	// loaded properly
 	// loaded properly
@@ -104,36 +93,34 @@ func (s *DockerSuite) TestConfigDir(c *check.C) {
 
 
 	tmpCfg := filepath.Join(cDir, "config.json")
 	tmpCfg := filepath.Join(cDir, "config.json")
 	err = ioutil.WriteFile(tmpCfg, []byte(data), 0600)
 	err = ioutil.WriteFile(tmpCfg, []byte(data), 0600)
-	if err != nil {
-		c.Fatalf("Err creating file(%s): %v", tmpCfg, err)
-	}
+	c.Assert(err, checker.IsNil, check.Commentf("Err creating file"))
 
 
 	cmd = exec.Command(dockerBinary, "--config", cDir, "-H="+server.URL[7:], "ps")
 	cmd = exec.Command(dockerBinary, "--config", cDir, "-H="+server.URL[7:], "ps")
-	out, _, _ = runCommandWithOutput(cmd)
+	out, _, err = runCommandWithOutput(cmd)
 
 
-	if headers["Myheader"] == nil || headers["Myheader"][0] != "MyValue" {
-		c.Fatalf("ps3 - Missing header: %q\nout:%v", headers, out)
-	}
+	c.Assert(err, checker.NotNil, check.Commentf("out:%v", out))
+	c.Assert(headers["Myheader"], checker.NotNil)
+	c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps3 - Missing header,out:%v", out))
 
 
 	// Reset headers and try again using env var this time
 	// Reset headers and try again using env var this time
 	headers = map[string][]string{}
 	headers = map[string][]string{}
 	cmd = exec.Command(dockerBinary, "-H="+server.URL[7:], "ps")
 	cmd = exec.Command(dockerBinary, "-H="+server.URL[7:], "ps")
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
-	out, _, _ = runCommandWithOutput(cmd)
+	out, _, err = runCommandWithOutput(cmd)
 
 
-	if headers["Myheader"] == nil || headers["Myheader"][0] != "MyValue" {
-		c.Fatalf("ps4 - Missing header: %q\nout:%v", headers, out)
-	}
+	c.Assert(err, checker.NotNil, check.Commentf("%v", out))
+	c.Assert(headers["Myheader"], checker.NotNil)
+	c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps4 - Missing header,out:%v", out))
 
 
 	// Reset headers and make sure flag overrides the env var
 	// Reset headers and make sure flag overrides the env var
 	headers = map[string][]string{}
 	headers = map[string][]string{}
 	cmd = exec.Command(dockerBinary, "--config", cDir, "-H="+server.URL[7:], "ps")
 	cmd = exec.Command(dockerBinary, "--config", cDir, "-H="+server.URL[7:], "ps")
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG=MissingDir")
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG=MissingDir")
-	out, _, _ = runCommandWithOutput(cmd)
+	out, _, err = runCommandWithOutput(cmd)
 
 
-	if headers["Myheader"] == nil || headers["Myheader"][0] != "MyValue" {
-		c.Fatalf("ps5 - Missing header: %q\nout:%v", headers, out)
-	}
+	c.Assert(err, checker.NotNil, check.Commentf("out:%v", out))
+	c.Assert(headers["Myheader"], checker.NotNil)
+	c.Assert(headers["Myheader"][0], checker.Equals, "MyValue", check.Commentf("ps5 - Missing header,out:%v", out))
 
 
 	// Reset headers and make sure flag overrides the env var.
 	// Reset headers and make sure flag overrides the env var.
 	// Almost same as previous but make sure the "MissingDir" isn't
 	// Almost same as previous but make sure the "MissingDir" isn't
@@ -141,10 +128,9 @@ func (s *DockerSuite) TestConfigDir(c *check.C) {
 	headers = map[string][]string{}
 	headers = map[string][]string{}
 	cmd = exec.Command(dockerBinary, "--config", "MissingDir", "-H="+server.URL[7:], "ps")
 	cmd = exec.Command(dockerBinary, "--config", "MissingDir", "-H="+server.URL[7:], "ps")
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
 	cmd.Env = append(os.Environ(), "DOCKER_CONFIG="+cDir)
-	out, _, _ = runCommandWithOutput(cmd)
+	out, _, err = runCommandWithOutput(cmd)
 
 
-	if headers["Myheader"] != nil {
-		c.Fatalf("ps6 - Headers are there but shouldn't be: %q\nout:%v", headers, out)
-	}
+	c.Assert(err, checker.NotNil, check.Commentf("out:%v", out))
+	c.Assert(headers["Myheader"], checker.IsNil, check.Commentf("ps6 - Headers shouldn't be the expected value,out:%v", out))
 
 
 }
 }