mirror of
https://github.com/drone-plugins/drone-docker.git
synced 2026-06-04 18:24:24 +08:00
Aishwarya Lad
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package docker
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
RegistryV1 string = "https://index.docker.io/v1/"
|
|
RegistryV2 string = "https://index.docker.io/v2/"
|
|
RegistryECRPublic string = "public.ecr.aws"
|
|
)
|
|
|
|
func TestConfig(t *testing.T) {
|
|
c := NewConfig()
|
|
assert.NotNil(t, c.Auths)
|
|
assert.NotNil(t, c.CredHelpers)
|
|
|
|
c.SetAuth(RegistryV1, "test", "password")
|
|
expectedAuth := Auth{Auth: "dGVzdDpwYXNzd29yZA=="}
|
|
assert.Equal(t, expectedAuth, c.Auths[RegistryV1])
|
|
|
|
c.SetCredHelper(RegistryECRPublic, "ecr-login")
|
|
assert.Equal(t, "ecr-login", c.CredHelpers[RegistryECRPublic])
|
|
|
|
tempDir, err := ioutil.TempDir("", "docker-config-test")
|
|
assert.NoError(t, err)
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
credentials := []RegistryCredentials{
|
|
{
|
|
Registry: "https://index.docker.io/v1/",
|
|
Username: "user1",
|
|
Password: "pass1",
|
|
},
|
|
{
|
|
Registry: "gcr.io",
|
|
Username: "user2",
|
|
Password: "pass2",
|
|
},
|
|
}
|
|
|
|
jsonBytes, err := c.CreateDockerConfigJson(credentials)
|
|
assert.NoError(t, err)
|
|
|
|
configPath := filepath.Join(tempDir, "config.json")
|
|
err = ioutil.WriteFile(configPath, jsonBytes, 0644)
|
|
assert.NoError(t, err)
|
|
|
|
data, err := ioutil.ReadFile(configPath)
|
|
assert.NoError(t, err)
|
|
|
|
var configFromFile Config
|
|
err = json.Unmarshal(data, &configFromFile)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, c.Auths, configFromFile.Auths)
|
|
assert.Equal(t, c.CredHelpers, configFromFile.CredHelpers)
|
|
}
|