reg/repoutils/repoutils_test.go

package repoutils

import (
	"testing"

	"github.com/docker/distribution/reference"
	"github.com/docker/docker/api/types"
	"github.com/google/go-cmp/cmp"
)

func TestGetAuthConfig(t *testing.T) {
	configTestcases := []struct {
		name                         string
		username, password, registry string
		err                          error
		config                       types.AuthConfig
	}{
		{
			name:     "pass in all details",
			username: "jess",
			password: "password",
			registry: "r.j3ss.co",
			config: types.AuthConfig{
				Username:      "jess",
				Password:      "password",
				ServerAddress: "r.j3ss.co",
			},
		},
	}

	for _, testcase := range configTestcases {
		cfg, err := GetAuthConfig(testcase.username, testcase.password, testcase.registry)
		if err != nil {
			if err.Error() != testcase.err.Error() {
				t.Fatalf("%q: expected err (%v), got err (%v)", testcase.name, testcase.err, err)
			}
			continue
		}
		if diff := cmp.Diff(testcase.config, cfg); diff != "" {
			t.Errorf("%s: authconfig differs: (-got +want)\n%s", testcase.name, diff)
		}
	}
}

func TestGetRepoAndRef(t *testing.T) {
	imageTestcases := []struct {
		// input is the repository name or name component testcase
		input string
		// err is the error expected from Parse, or nil
		err error
		// repository is the string representation for the reference
		repository string
		// ref the reference
		ref string
	}{
		{
			input:      "alpine",
			repository: "alpine",
			ref:        "latest",
		},
		{
			input:      "docker:dind",
			repository: "docker",
			ref:        "dind",
		},
		{
			input: "",
			err:   reference.ErrNameEmpty,
		},
		{
			input:      "chrome@sha256:2a6c8ad38c41ae5122d76be59b34893d7fa1bdfaddd85bf0e57d0d16c0f7f91e",
			repository: "chrome",
			ref:        "sha256:2a6c8ad38c41ae5122d76be59b34893d7fa1bdfaddd85bf0e57d0d16c0f7f91e",
		},
	}

	for _, testcase := range imageTestcases {
		repo, ref, err := GetRepoAndRef(testcase.input)
		if err != nil {
			if err.Error() != testcase.err.Error() {
				t.Fatalf("%q: expected err (%v), got err (%v)", testcase.input, testcase.err, err)
			}
			continue
		}

		if testcase.repository != repo {
			t.Fatalf("%q: expected repo (%s), got repo (%s)", testcase.input, testcase.repository, repo)
		}

		if testcase.ref != ref {
			t.Fatalf("%q: expected ref (%s), got ref (%s)", testcase.input, testcase.ref, ref)
		}
	}
}