first commit

unitard_test.go

@@ -0,0 +1,57 @@
+package unitard
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+)
+
+func TestTemplate(t *testing.T) {
+	u := Unit{
+		name:          "test_unit",
+		binary:        "/fullpath/to/foobar",
+		systemCtlPath: "/who/cares",
+		unitFilePath:  "/doesnt/matter",
+	}
+
+	buff := bytes.NewBuffer(nil) // create empty buffer
+
+	err := u.writeTemplate(buff)
+	if err != nil {
+		t.Errorf("failed to write template: %s", err)
+	}
+	t.Logf("template:\n%s", buff.String())
+
+	if !strings.Contains(buff.String(), "Description=test_unit") {
+		t.Error("template does not contain description?")
+	}
+	if !strings.Contains(buff.String(), "ExecStart=/fullpath/to/foobar") {
+		t.Error("template does not contain exec start?")
+	}
+}
+
+func TestCheckName(t *testing.T) {
+	validNames := []string{
+		"test_unit",
+		"leotard123",
+		"winger_01",
+	}
+	invalidNames := []string{
+		"no way",
+		"doesn't_work",
+		"C:/dev/null",
+		"/no/slashes",
+	}
+
+	for _, v := range validNames {
+		if !checkName(v) {
+			t.Errorf("%s not valid but should be", v)
+		}
+	}
+	for _, i := range invalidNames {
+		if checkName(i) {
+			t.Errorf("%s  valid but should be", i)
+		}
+	}
+
+}