123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- package main
- import (
- "os/exec"
- "strings"
- "testing"
- "github.com/docker/docker/integration-cli/daemon"
- testdaemon "github.com/docker/docker/testutil/daemon"
- "gotest.tools/v3/assert"
- )
- // DockerHubPullSuite provides an 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.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 *testing.T) {
- testRequires(c, DaemonIsLinux, testEnv.IsLocalDaemon)
- s.d = daemon.New(c, dockerBinary, dockerdBinary, testdaemon.WithEnvironment(testEnv.Execution))
- s.d.Start(c)
- }
- // TearDownSuite stops the suite daemon.
- func (s *DockerHubPullSuite) TearDownSuite(c *testing.T) {
- if s.d != nil {
- s.d.Stop(c)
- }
- }
- // SetUpTest declares that all tests of this suite require network.
- func (s *DockerHubPullSuite) SetUpTest(c *testing.T) {
- testRequires(c, Network)
- }
- // TearDownTest removes all images from the suite daemon.
- func (s *DockerHubPullSuite) TearDownTest(c *testing.T) {
- out := s.Cmd(c, "images", "-aq")
- images := strings.Split(out, "\n")
- images = append([]string{"rmi", "-f"}, images...)
- s.d.Cmd(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 *testing.T, name string, arg ...string) string {
- out, err := s.CmdWithError(name, arg...)
- assert.Assert(c, err == nil, "%q failed with errors: %s, %v", strings.Join(arg, " "), out, err)
- 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 an 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...)
- }
|