generate os-config.yml

cmd/control/config_test.go

@@ -0,0 +1,56 @@
+package control
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"os"
+)
+
+func TestGenTpl(t *testing.T) {
+	assert := require.New(t)
+	tpl := `
+  services:
+    {{if eq "amd64" .ARCH -}}
+    acpid:
+      image: rancher/os-acpid:0.x.x
+      labels:
+        io.rancher.os.scope: system
+      net: host
+      uts: host
+      privileged: true
+      volumes_from:
+      - command-volumes
+      - system-volumes
+    {{end -}}
+    all-volumes:`
+
+	for _, tc := range []struct {
+		arch     string
+		expected string
+	}{
+		{"amd64", `
+  services:
+    acpid:
+      image: rancher/os-acpid:0.x.x
+      labels:
+        io.rancher.os.scope: system
+      net: host
+      uts: host
+      privileged: true
+      volumes_from:
+      - command-volumes
+      - system-volumes
+    all-volumes:`},
+		{"arm", `
+  services:
+    all-volumes:`},
+	} {
+		out := &bytes.Buffer{}
+		os.Setenv("ARCH", tc.arch)
+		genTpl(strings.NewReader(tpl), out)
+		assert.Equal(tc.expected, out.String(), tc.arch)
+	}
+}