generator/katenaryfile/main_test.go

@@ -0,0 +1,66 @@
+package katenaryfile
+
+import (
+	"katenary/generator/labels"
+	"log"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/compose-spec/compose-go/cli"
+)
+
+func TestBuildSchema(t *testing.T) {
+	sh := GenerateSchema()
+	if len(sh) == 0 {
+		t.Errorf("Expected schema to be defined")
+	}
+}
+
+func TestOverrideProjectWithKatenaryFile(t *testing.T) {
+	composeContent := `
+services:
+  webapp:
+    image: ngnx:latest
+`
+
+	katenaryfileContent := `
+webapp:
+  ports:
+    - 80
+`
+
+	// create /tmp/katenary-test-override directory, save the compose.yaml file
+	tmpDir, err := os.MkdirTemp("", "katenary-test-override")
+	if err != nil {
+		t.Fatalf(err.Error())
+	}
+	composeFile := filepath.Join(tmpDir, "compose.yaml")
+	katenaryFile := filepath.Join(tmpDir, "katenary.yaml")
+
+	os.MkdirAll(tmpDir, 0755)
+	if err := os.WriteFile(composeFile, []byte(composeContent), 0644); err != nil {
+		t.Log(err)
+	}
+	if err := os.WriteFile(katenaryFile, []byte(katenaryfileContent), 0644); err != nil {
+		t.Log(err)
+	}
+	defer os.RemoveAll(tmpDir)
+
+	c, _ := os.ReadFile(composeFile)
+	log.Println(string(c))
+
+	// chand dir to this directory
+	os.Chdir(tmpDir)
+	options, _ := cli.NewProjectOptions(nil,
+		cli.WithWorkingDirectory(tmpDir),
+		cli.WithDefaultConfigPath,
+	)
+	project, err := cli.ProjectFromOptions(options)
+
+	OverrideWithConfig(project)
+	w := project.Services[0].Labels
+	if v, ok := w[labels.LabelPorts]; !ok {
+		t.Fatal("Expected ports to be defined", v)
+	}
+}