diff options
Diffstat (limited to 'libpod')
-rw-r--r-- | libpod/runtime_img_test.go | 54 | ||||
-rw-r--r-- | libpod/util_test.go | 6 |
2 files changed, 57 insertions, 3 deletions
diff --git a/libpod/runtime_img_test.go b/libpod/runtime_img_test.go new file mode 100644 index 000000000..f7f7128a4 --- /dev/null +++ b/libpod/runtime_img_test.go @@ -0,0 +1,54 @@ +package libpod + +import ( + "io/ioutil" + "os" + "reflect" + "testing" + + "github.com/stretchr/testify/assert" +) + +var ( + registry = `[registries.search] +registries = ['one'] + +[registries.insecure] +registries = ['two']` +) + +func createTmpFile(content []byte) (string, error) { + tmpfile, err := ioutil.TempFile(os.TempDir(), "unittest") + if err != nil { + return "", err + } + + if _, err := tmpfile.Write(content); err != nil { + return "", err + + } + if err := tmpfile.Close(); err != nil { + return "", err + } + return tmpfile.Name(), nil +} + +func TestGetRegistries(t *testing.T) { + registryPath, err := createTmpFile([]byte(registry)) + assert.NoError(t, err) + defer os.Remove(registryPath) + os.Setenv("REGISTRIES_CONFIG_PATH", registryPath) + registries, err := GetRegistries() + assert.NoError(t, err) + assert.True(t, reflect.DeepEqual(registries, []string{"one"})) +} + +func TestGetInsecureRegistries(t *testing.T) { + registryPath, err := createTmpFile([]byte(registry)) + assert.NoError(t, err) + os.Setenv("REGISTRIES_CONFIG_PATH", registryPath) + defer os.Remove(registryPath) + registries, err := GetInsecureRegistries() + assert.NoError(t, err) + assert.True(t, reflect.DeepEqual(registries, []string{"two"})) +} diff --git a/libpod/util_test.go b/libpod/util_test.go index b3d336d1f..24e5fdfac 100644 --- a/libpod/util_test.go +++ b/libpod/util_test.go @@ -1,10 +1,10 @@ package libpod - import ( - "testing" "github.com/stretchr/testify/assert" + "testing" ) + var ( sliceData = []string{"one", "two", "three", "four"} ) @@ -16,4 +16,4 @@ func TestStringInSlice(t *testing.T) { assert.False(t, StringInSlice("five", sliceData)) // string is not in empty slice assert.False(t, StringInSlice("one", []string{})) -}
\ No newline at end of file +} |