Function TestPullImageAuthenticated

internal/docker/docker_impl_test.go:18–67

(t *testing.T)

)

func TestPullImageAuthenticated(t *testing.T) {
    testRegistry := test.Registry(t, true)
    metricsCollector := metrics.New(dummy.New())
    clientFactory := &dockerV20ClientFactory{
        backendFailuresMetric: metricsCollector.MustCreateCounter("backend-failures", "requests", ""),
        backendRequestsMetric: metricsCollector.MustCreateCounter("backend-failures", "requests", ""),
    }
    ctx := context.Background()

    t.Run("unauthenticated", func(t *testing.T) {
        cfg := config.DockerConfig{}
        structutils.Defaults(&cfg)
        cfg.Execution.ContainerConfig.Image = fmt.Sprintf("localhost:%d/containerssh/guest-image", testRegistry.Port())

        logger := log.NewTestLogger(t)
        client, err := clientFactory.get(ctx, cfg, logger)
        if err != nil {
            t.Fatal(err)
        }

        pullCtx, cancel := context.WithTimeout(context.Background(), time.Minute)
        defer cancel()
        if err := client.pullImage(pullCtx); err == nil {
            t.Fatalf("Pulling without credentials didn't fail.")
        }
    })
    t.Run("authenticated", func(t *testing.T) {
        cfg := config.DockerConfig{}
        structutils.Defaults(&cfg)
        cfg.Execution.ContainerConfig.Image = fmt.Sprintf("localhost:%d/containerssh/agent", testRegistry.Port())
        cfg.Execution.Auth = &registry.AuthConfig{
            Username:      *testRegistry.Username(),
            Password:      *testRegistry.Password(),
            Email:         "noreply@containerssh.io",
            ServerAddress: fmt.Sprintf("localhost:%d", testRegistry.Port()),
        }

        logger := log.NewTestLogger(t)
        client, err := clientFactory.get(ctx, cfg, logger)
        if err != nil {
            t.Fatal(err)
        }

        pullCtx, cancel := context.WithTimeout(context.Background(), time.Minute)
        defer cancel()
        if err := client.pullImage(pullCtx); err != nil {
            t.Fatalf("Pulling with credentials failed (%v).", err)
        }
    })
}