Always assert RunQemu is successful

rancher · Oct 23, 2016 · 2ae22a6 · 2ae22a6
1 parent 4057275
commit 2ae22a6
Show file tree

Hide file tree

Showing 28 changed files with 36 additions and 72 deletions.
diff --git a/tests/bad_state_test.go b/tests/bad_state_test.go
@@ -3,13 +3,11 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestBadState(c *C) {
-	err := s.RunQemu("--no-format", "--append", "rancher.state.dev=LABEL=BAD_STATE")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--no-format", "--append", "rancher.state.dev=LABEL=BAD_STATE")
 	s.CheckCall(c, "mount | grep /var/lib/docker | grep rootfs")
 }
 
 func (s *QemuSuite) TestBadStateWithWait(c *C) {
-	err := s.RunQemu("--no-format", "--append", "rancher.state.dev=LABEL=BAD_STATE", "--append", "rancher.state.wait")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--no-format", "--append", "rancher.state.dev=LABEL=BAD_STATE", "--append", "rancher.state.wait")
 	s.CheckCall(c, "mount | grep /var/lib/docker | grep rootfs")
 }
diff --git a/tests/cloud_config_hostname_test.go b/tests/cloud_config_hostname_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestCloudConfigHostname(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_13/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_13/cloud-config.yml")
 
 	s.CheckCall(c, "hostname | grep rancher-test")
 	s.CheckCall(c, "cat /etc/hosts | grep rancher-test")

diff --git a/tests/common_test.go b/tests/common_test.go
@@ -31,9 +31,8 @@ var (
 		"arm":   "armhfbuild/nginx",
 		"arm64": "armhfbuild/nginx",
 	}[runtime.GOARCH]
-	DockerUrl = "https://experimental.docker.com/builds/Linux/x86_64/docker-1.10.0-dev"
-	Version   = os.Getenv("VERSION")
-	Suffix    = os.Getenv("SUFFIX")
+	Version = os.Getenv("VERSION")
+	Suffix  = os.Getenv("SUFFIX")
 )
 
 type QemuSuite struct {
@@ -47,7 +46,7 @@ func (s *QemuSuite) TearDownTest(c *C) {
 	time.Sleep(time.Millisecond * 1000)
 }
 
-func (s *QemuSuite) RunQemu(additionalArgs ...string) error {
+func (s *QemuSuite) RunQemu(c *C, additionalArgs ...string) {
 	runArgs := []string{
 		"--qemu",
 		"--no-rebuild",
@@ -56,10 +55,10 @@ func (s *QemuSuite) RunQemu(additionalArgs ...string) error {
 	}
 	runArgs = append(runArgs, additionalArgs...)
 
-	return s.runQemu(runArgs...)
+	c.Assert(s.runQemu(runArgs...), IsNil)
 }
 
-func (s *QemuSuite) RunQemuInstalled(additionalArgs ...string) error {
+func (s *QemuSuite) RunQemuInstalled(c *C, additionalArgs ...string) {
 	runArgs := []string{
 		"--qemu",
 		"--no-rebuild",
@@ -68,7 +67,7 @@ func (s *QemuSuite) RunQemuInstalled(additionalArgs ...string) error {
 	}
 	runArgs = append(runArgs, additionalArgs...)
 
-	return s.runQemu(runArgs...)
+	c.Assert(s.runQemu(runArgs...), IsNil)
 }
 
 func (s *QemuSuite) runQemu(args ...string) error {

diff --git a/tests/consoles_test.go b/tests/consoles_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestCloudConfigConsole(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_03/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_03/cloud-config.yml")
 
 	s.CheckCall(c, "apt-get --version")
 	s.CheckCall(c, `
@@ -13,8 +12,7 @@ sudo ros console list | grep debian | grep current`)
 }
 
 func (s *QemuSuite) TestConsoleCommand(c *C) {
-	err := s.RunQemu()
-	c.Assert(err, IsNil)
+	s.RunQemu(c)
 
 	s.CheckCall(c, `
 sudo ros console list | grep default | grep current

diff --git a/tests/custom_docker_test.go b/tests/custom_docker_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestCustomDocker(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_05/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_05/cloud-config.yml")
 
 	s.CheckCall(c, `
 set -ex
@@ -35,8 +34,7 @@ docker ps | grep nginx`)
 }
 
 func (s *QemuSuite) TestCustomDockerInPersistentConsole(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_25/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_25/cloud-config.yml")
 
 	s.CheckCall(c, `
 set -ex

diff --git a/tests/dhcp_hostname_test.go b/tests/dhcp_hostname_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestDhcpHostname(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_12/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_12/cloud-config.yml")
 
 	s.CheckCall(c, "hostname | grep rancher-dev")
 	s.CheckCall(c, "cat /etc/hosts | grep rancher-dev")

diff --git a/tests/environment_test.go b/tests/environment_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestEnvironment(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_11/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_11/cloud-config.yml")
 
 	s.CheckCall(c, "sudo system-docker inspect env | grep A=A")
 	s.CheckCall(c, "sudo system-docker inspect env | grep BB=BB")

diff --git a/tests/http_proxy_test.go b/tests/http_proxy_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestHttpProxy(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_17/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_17/cloud-config.yml")
 
 	s.CheckCall(c, `
 set -x -e

diff --git a/tests/kernel_headers_test.go b/tests/kernel_headers_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestKernelHeaders(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_22/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_22/cloud-config.yml")
 
 	s.CheckCall(c, `
 sleep 15

diff --git a/tests/lenient_service_parsing_test.go b/tests/lenient_service_parsing_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestLenientServiceParsing(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_19/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_19/cloud-config.yml")
 
 	s.CheckCall(c, `
 sleep 5

diff --git a/tests/misc_test.go b/tests/misc_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestMisc(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_01/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_01/cloud-config.yml")
 
 	s.CheckCall(c, "sudo ros env printenv FLANNEL_NETWORK | grep '10.244.0.0/16'")
 

diff --git a/tests/mounts_test.go b/tests/mounts_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestMounts(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_23/cloud-config.yml", "--second-drive")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_23/cloud-config.yml", "--second-drive")
 
 	s.CheckCall(c, "cat /home/rancher/test | grep test")
 

diff --git a/tests/network_from_url_test.go b/tests/network_from_url_test.go
@@ -8,8 +8,7 @@ func (s *QemuSuite) TestNetworkFromUrl(c *C) {
 	for i := 0; i < 7; i++ {
 		args = append(args, netArgs...)
 	}
-	err := s.RunQemu(args...)
-	c.Assert(err, IsNil)
+	s.RunQemu(c, args...)
 
 	s.CheckCall(c, `
 cat > test-merge << "SCRIPT"

diff --git a/tests/network_on_boot_test.go b/tests/network_on_boot_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestNetworkOnBoot(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_18/cloud-config.yml", "-net", "nic,vlan=1,model=virtio")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_18/cloud-config.yml", "-net", "nic,vlan=1,model=virtio")
 
 	s.CheckCall(c, "apt-get --version")
 	s.CheckCall(c, "sudo system-docker images | grep tianon/true")

diff --git a/tests/network_test.go b/tests/network_test.go
@@ -8,8 +8,7 @@ func (s *QemuSuite) TestNetwork(c *C) {
 	for i := 0; i < 7; i++ {
 		args = append(args, netArgs...)
 	}
-	err := s.RunQemu(args...)
-	c.Assert(err, IsNil)
+	s.RunQemu(c, args...)
 
 	s.CheckCall(c, `
 cat > test-merge << "SCRIPT"

diff --git a/tests/nonexistent_state_test.go b/tests/nonexistent_state_test.go
@@ -3,8 +3,6 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestNonexistentState(c *C) {
-	err := s.RunQemu("--no-format", "--append", "rancher.state.dev=LABEL=NONEXISTENT")
-	c.Assert(err, IsNil)
-
+	s.RunQemu(c, "--no-format", "--append", "rancher.state.dev=LABEL=NONEXISTENT")
 	s.CheckCall(c, "sudo ros config get rancher.state.dev | grep LABEL=NONEXISTENT")
 }
diff --git a/tests/oem_test.go b/tests/oem_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestOem(c *C) {
-	err := s.RunQemu("--second-drive")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--second-drive")
 
 	s.CheckCall(c, `
 set -x

diff --git a/tests/preload_test.go b/tests/preload_test.go
@@ -5,8 +5,7 @@ import (
 )
 
 func (s *QemuSuite) TestPreload(c *C) {
-	err := s.RunQemu()
-	c.Assert(err, IsNil)
+	s.RunQemu(c)
 
 	s.CheckCall(c, `
 docker pull busybox

diff --git a/tests/ros_config_test.go b/tests/ros_config_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestRosConfig(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_14/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_14/cloud-config.yml")
 
 	s.CheckCall(c, `
 set -x -e

diff --git a/tests/shared_mount_test.go b/tests/shared_mount_test.go
@@ -7,9 +7,7 @@ import (
 )
 
 func (s *QemuSuite) TestSharedMount(c *C) {
-	err := s.RunQemu()
-	c.Assert(err, IsNil)
-
+	s.RunQemu(c)
 	s.CheckCall(c, fmt.Sprintf(`
 set -x -e
 

diff --git a/tests/ssh_key_merge_test.go b/tests/ssh_key_merge_test.go
@@ -3,9 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestSshKeyMerge(c *C) {
-	err := s.RunQemu()
-	c.Assert(err, IsNil)
-
+	s.RunQemu(c)
 	s.CheckCall(c, `
 cat > test-merge << "SCRIPT"
 set -x -e

diff --git a/tests/start_commands_test.go b/tests/start_commands_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func (s *QemuSuite) TestStartCommands(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_26/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_26/cloud-config.yml")
 
 	for i := 1; i < 5; i++ {
 		s.CheckCall(c, fmt.Sprintf("ls /home/rancher | grep test%d", i))

diff --git a/tests/subdir_test.go b/tests/subdir_test.go
@@ -3,9 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestSubdir(c *C) {
-	err := s.RunQemu("--append", "rancher.state.directory=ros_subdir")
-	c.Assert(err, IsNil)
-
+	s.RunQemu(c, "--append", "rancher.state.directory=ros_subdir")
 	s.CheckCall(c, `
 set -x -e
 mkdir x

diff --git a/tests/swap_test.go b/tests/swap_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestSwap(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_21/cloud-config.yml", "--second-drive")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_21/cloud-config.yml", "--second-drive")
 
 	s.CheckCall(c, "sudo mkswap /dev/vdb")
 	s.CheckCall(c, "sudo cloud-init-execute")

diff --git a/tests/sysctl_test.go b/tests/sysctl_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestSysctl(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_20/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_20/cloud-config.yml")
 
 	s.CheckCall(c, "sudo cat /proc/sys/kernel/domainname | grep test")
 	s.CheckCall(c, "sudo cat /proc/sys/dev/cdrom/debug | grep 1")

diff --git a/tests/tls_test.go b/tests/tls_test.go
@@ -3,9 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestTls(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_02/cloud-config.yml")
-	c.Assert(err, IsNil)
-
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_02/cloud-config.yml")
 	s.CheckCall(c, `
 set -e -x
 sudo ros tls gen

diff --git a/tests/upgrade_test.go b/tests/upgrade_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func (s *QemuSuite) TestUpgrade(c *C) {
-	err := s.RunQemuInstalled()
-	c.Assert(err, IsNil)
+	s.RunQemuInstalled(c)
 
 	s.CheckCall(c, `
 set -ex

diff --git a/tests/write_files_test.go b/tests/write_files_test.go
@@ -3,8 +3,7 @@ package integration
 import . "gopkg.in/check.v1"
 
 func (s *QemuSuite) TestWriteFiles(c *C) {
-	err := s.RunQemu("--cloud-config", "./tests/assets/test_24/cloud-config.yml")
-	c.Assert(err, IsNil)
+	s.RunQemu(c, "--cloud-config", "./tests/assets/test_24/cloud-config.yml")
 
 	s.CheckCall(c, "sudo cat /test | grep 'console content'")
 	s.CheckCall(c, "sudo cat /test2 | grep 'console content'")