21 lines
554 B
Go
21 lines
554 B
Go
package main
|
|
|
|
import "testing"
|
|
|
|
func TestTrimHostname(t *testing.T) {
|
|
registry := "000000000000.dkr.ecr.us-east-1.amazonaws.com"
|
|
// map full repo path to expected repo name
|
|
repos := map[string]string{
|
|
registry + "/repo": "repo",
|
|
registry + "/namespace/repo": "namespace/repo",
|
|
registry + "/namespace/namespace/repo": "namespace/namespace/repo",
|
|
}
|
|
|
|
for repo, name := range repos {
|
|
splitName := trimHostname(repo, registry)
|
|
if splitName != name {
|
|
t.Errorf("%s is not equal to %s.", splitName, name)
|
|
}
|
|
}
|
|
}
|