dokku/plugins/common/common_test.go

package common

import (
	"io/ioutil"
	"os"
	"strings"
	"testing"

	. "github.com/onsi/gomega"
)

var (
	testAppName = "test-app-1"
	testAppDir  = strings.Join([]string{"/home/dokku/", testAppName}, "")
	testEnvFile = strings.Join([]string{testAppDir, "/ENV"}, "")
	testEnvLine = "export testKey=TESTING"
	testAppName2 = "01-test-app-1"
	testAppDir2  = strings.Join([]string{"/home/dokku/", testAppName}, "")
	testEnvFile2 = strings.Join([]string{testAppDir, "/ENV"}, "")
	testEnvLine2 = "export testKey=TESTING"
)

func setupTestApp() (err error) {
	Expect(os.MkdirAll(testAppDir, 0644)).To(Succeed())
	b := []byte(testEnvLine + "\n")
	if err = ioutil.WriteFile(testEnvFile, b, 0644); err != nil {
		return
	}
	return
}

func setupTestApp2() (err error) {
	Expect(os.MkdirAll(testAppDir2, 0644)).To(Succeed())
	b := []byte(testEnvLine2 + "\n")
	if err = ioutil.WriteFile(testEnvFile2, b, 0644); err != nil {
		return
	}
	return
}

func teardownTestApp() {
	os.RemoveAll(testAppDir)
}

func teardownTestApp2() {
	os.RemoveAll(testAppDir2)
}

func TestCommonGetEnv(t *testing.T) {
	RegisterTestingT(t)
	Expect(MustGetEnv("DOKKU_ROOT")).To(Equal("/home/dokku"))
}

func TestCommonGetAppImageRepo(t *testing.T) {
	RegisterTestingT(t)
	Expect(GetAppImageRepo("testapp")).To(Equal("dokku/testapp"))
}

func TestCommonVerifyImageInvalid(t *testing.T) {
	RegisterTestingT(t)
	Expect(VerifyImage("testapp")).To(Equal(false))
}

func TestCommonVerifyAppNameInvalid(t *testing.T) {
	RegisterTestingT(t)
	err := VerifyAppName("1994testApp")
	Expect(err).To(HaveOccurred())
}

func TestCommonVerifyAppName(t *testing.T) {
	RegisterTestingT(t)
	Expect(setupTestApp()).To(Succeed())
	Expect(VerifyAppName(testAppName)).To(Succeed())
	teardownTestApp()

	RegisterTestingT(t)
	Expect(setupTestApp2()).To(Succeed())
	Expect(VerifyAppName(testAppName2)).To(Succeed())
	teardownTestApp2()
}

func TestCommonDokkuAppsError(t *testing.T) {
	RegisterTestingT(t)
	_, err := DokkuApps()
	Expect(err).To(HaveOccurred())
}

func TestCommonDokkuApps(t *testing.T) {
	RegisterTestingT(t)
	Expect(setupTestApp()).To(Succeed())
	apps, err := DokkuApps()
	Expect(err).NotTo(HaveOccurred())
	Expect(apps).To(HaveLen(1))
	Expect(apps[0]).To(Equal(testAppName))
	teardownTestApp()
}

func TestCommonFileToSlice(t *testing.T) {
	RegisterTestingT(t)
	Expect(setupTestApp()).To(Succeed())
	lines, err := FileToSlice(testEnvFile)
	Expect(err).NotTo(HaveOccurred())
	Expect(lines).To(Equal([]string{testEnvLine}))
	teardownTestApp()
}

func TestCommonFileExists(t *testing.T) {
	RegisterTestingT(t)
	Expect(setupTestApp()).To(Succeed())
	Expect(FileExists(testEnvFile)).To(BeTrue())
	teardownTestApp()
}

func TestCommonReadFirstLine(t *testing.T) {
	RegisterTestingT(t)
	line := ReadFirstLine(testEnvFile)
	Expect(line).To(Equal(""))
	Expect(setupTestApp()).To(Succeed())
	line = ReadFirstLine(testEnvFile)
	Expect(line).To(Equal(testEnvLine))
	teardownTestApp()
}

func TestCommonStripInlineComments(t *testing.T) {
	RegisterTestingT(t)
	text := StripInlineComments(strings.Join([]string{testEnvLine, "# testing comment"}, " "))
	Expect(text).To(Equal(testEnvLine))
}