58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|