Merge pull request #8 from baude/clicontext

Clicontext

2 files changed, 73 insertions, 0 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
new file mode 100644
index 000000000..24e5fdfac
--- /dev/null
+++ b/libpod/util_test.go
@@ -0,0 +1,19 @@
+package libpod
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+var (
+	sliceData = []string{"one", "two", "three", "four"}
+)
+
+func TestStringInSlice(t *testing.T) {
+	// string is in the slice
+	assert.True(t, StringInSlice("one", sliceData))
+	// string is not in the slice
+	assert.False(t, StringInSlice("five", sliceData))
+	// string is not in empty slice
+	assert.False(t, StringInSlice("one", []string{}))
+}