fn: tests for private repo auth and rename DOCKER_AUTH (#1134)

Renamed DOCKER_AUTH with FN_ prefix to clarify the purpose. Docker
does not use this variable.

New tests to clarify the repo/auth-config behavior.

api/agent/drivers/docker/registry.go

@@ -14,12 +14,9 @@ var (
 )
 
 func registryFromEnv() (map[string]driverAuthConfig, error) {
-	drvAuths := make(map[string]driverAuthConfig)
-
 	var auths *docker.AuthConfigurations
 	var err error
-	if reg := os.Getenv("DOCKER_AUTH"); reg != "" {
-		// TODO docker does not use this itself, we should get rid of env docker config (nor is this documented..)
+	if reg := os.Getenv("FN_DOCKER_AUTH"); reg != "" {
 		auths, err = docker.NewAuthConfigurations(strings.NewReader(reg))
 	} else {
 		auths, err = docker.NewAuthConfigurationsFromDockerCfg()
@@ -27,9 +24,15 @@ func registryFromEnv() (map[string]driverAuthConfig, error) {
 
 	if err != nil {
 		logrus.WithError(err).Info("no docker auths from config files found (this is fine)")
-		return drvAuths, nil
+		return map[string]driverAuthConfig{}, nil
 	}
 
+	return preprocessAuths(auths)
+}
+
+func preprocessAuths(auths *docker.AuthConfigurations) (map[string]driverAuthConfig, error) {
+	drvAuths := make(map[string]driverAuthConfig)
+
 	for key, v := range auths.Configs {
 
 		u, err := url.Parse(v.ServerAddress)
@@ -42,7 +45,6 @@ func registryFromEnv() (map[string]driverAuthConfig, error) {
 			subdomains: getSubdomains(u.Host),
 		}
 	}
-
 	return drvAuths, nil
 }
 

api/agent/drivers/docker/registry_test.go

@@ -1,7 +1,10 @@
 package docker
 
 import (
+	"strings"
 	"testing"
+
+	"github.com/fsouza/go-dockerclient"
 )
 
 func verify(expected []string, checks map[string]bool) bool {
@@ -57,3 +60,54 @@ func TestRegistrySubDomains(t *testing.T) {
 		t.Fatalf("subdomain results failed expected[%+v] != results[%+v]", exp, res)
 	}
 }
+
+func TestRegistryEnv(t *testing.T) {
+
+	testCfg := `{
+	"auths":{
+		"https://my.registry.com":{"auth":"Y29jbzpjaGVlc2UK"},
+		"https://my.registry.com:5000":{"auth":"Y29jbzpjaGVlc2UK"},
+		"https://index.docker.io/v2/":{"auth":"Y29jbzpjaGVlc2UK"}
+	}}`
+
+	auths, err := docker.NewAuthConfigurations(strings.NewReader(testCfg))
+	if err != nil {
+		t.Fatalf("parsing test cfg failed: %s", err)
+	}
+
+	drvAuths, err := preprocessAuths(auths)
+	if err != nil {
+		t.Fatalf("preprocess test cfg failed: %s", err)
+	}
+
+	res := findRegistryConfig("", drvAuths)
+	if res == nil || res.ServerAddress != "https://index.docker.io/v2/" {
+		t.Fatalf("empty registry should pickup docker %v", res)
+	}
+
+	res = findRegistryConfig("docker.io", drvAuths)
+	if res == nil || res.ServerAddress != "https://index.docker.io/v2/" {
+		t.Fatalf("docker.io registry should pickup docker %v", res)
+	}
+
+	res = findRegistryConfig("localhost", drvAuths)
+	if res == nil || res.ServerAddress != "" {
+		t.Fatalf("localhost registry should pickup a default (empty) cfg %v", res)
+	}
+
+	res = findRegistryConfig("registry.com", drvAuths)
+	if res == nil || res.ServerAddress != "https://my.registry.com" {
+		t.Fatalf("registry.com registry should pickup my.registry.com cfg %v", res)
+	}
+
+	res = findRegistryConfig("my.registry.com", drvAuths)
+	if res == nil || res.ServerAddress != "https://my.registry.com" {
+		t.Fatalf("my.registry.com registry should pickup my.registry.com cfg %v", res)
+	}
+
+	res = findRegistryConfig("registry.com:5000", drvAuths)
+	if res == nil || res.ServerAddress != "https://my.registry.com:5000" {
+		t.Fatalf("registry.com:5000 registry should pickup my.registry.com:5000 cfg %v", res)
+	}
+
+}