integration-cli/docker_hub_pull_suite_test.go
9e4addde
 package main
 
 import (
 	"os/exec"
0b18b1b3
 	"runtime"
9e4addde
 	"strings"
 
ee7fdbe8
 	"github.com/docker/docker/pkg/integration/checker"
9e4addde
 	"github.com/go-check/check"
 )
 
 func init() {
0b18b1b3
 	// FIXME. Temporarily turning this off for Windows as GH16039 was breaking
 	// Windows to Linux CI @icecrime
 	if runtime.GOOS != "windows" {
 		check.Suite(newDockerHubPullSuite())
 	}
9e4addde
 }
 
 // DockerHubPullSuite provides a isolated daemon that doesn't have all the
 // images that are baked into our 'global' test environment daemon (e.g.,
 // busybox, httpserver, ...).
 //
 // We use it for push/pull tests where we want to start fresh, and measure the
 // relative impact of each individual operation. As part of this suite, all
 // images are removed after each test.
 type DockerHubPullSuite struct {
 	d  *Daemon
 	ds *DockerSuite
 }
 
 // newDockerHubPullSuite returns a new instance of a DockerHubPullSuite.
 func newDockerHubPullSuite() *DockerHubPullSuite {
 	return &DockerHubPullSuite{
 		ds: &DockerSuite{},
 	}
 }
 
 // SetUpSuite starts the suite daemon.
 func (s *DockerHubPullSuite) SetUpSuite(c *check.C) {
f9a3558a
 	testRequires(c, DaemonIsLinux)
9e4addde
 	s.d = NewDaemon(c)
ee7fdbe8
 	err := s.d.Start()
 	c.Assert(err, checker.IsNil, check.Commentf("starting push/pull test daemon: %v", err))
9e4addde
 }
 
 // TearDownSuite stops the suite daemon.
 func (s *DockerHubPullSuite) TearDownSuite(c *check.C) {
f9a3558a
 	if s.d != nil {
ee7fdbe8
 		err := s.d.Stop()
 		c.Assert(err, checker.IsNil, check.Commentf("stopping push/pull test daemon: %v", err))
9e4addde
 	}
 }
 
 // SetUpTest declares that all tests of this suite require network.
 func (s *DockerHubPullSuite) SetUpTest(c *check.C) {
 	testRequires(c, Network)
 }
 
 // TearDownTest removes all images from the suite daemon.
 func (s *DockerHubPullSuite) TearDownTest(c *check.C) {
 	out := s.Cmd(c, "images", "-aq")
 	images := strings.Split(out, "\n")
 	images = append([]string{"-f"}, images...)
 	s.d.Cmd("rmi", images...)
 	s.ds.TearDownTest(c)
 }
 
 // Cmd executes a command against the suite daemon and returns the combined
 // output. The function fails the test when the command returns an error.
 func (s *DockerHubPullSuite) Cmd(c *check.C, name string, arg ...string) string {
 	out, err := s.CmdWithError(name, arg...)
ee7fdbe8
 	c.Assert(err, checker.IsNil, check.Commentf("%q failed with errors: %s, %v", strings.Join(arg, " "), out, err))
9e4addde
 	return out
 }
 
 // CmdWithError executes a command against the suite daemon and returns the
 // combined output as well as any error.
 func (s *DockerHubPullSuite) CmdWithError(name string, arg ...string) (string, error) {
 	c := s.MakeCmd(name, arg...)
 	b, err := c.CombinedOutput()
 	return string(b), err
 }
 
 // MakeCmd returns a exec.Cmd command to run against the suite daemon.
 func (s *DockerHubPullSuite) MakeCmd(name string, arg ...string) *exec.Cmd {
 	args := []string{"--host", s.d.sock(), name}
 	args = append(args, arg...)
 	return exec.Command(dockerBinary, args...)
 }