Add tests

Author: mrueg

go.mod

@@ -4,6 +4,7 @@ go 1.23.4
 
 require (
 	github.com/go-sprout/sprout v1.0.0
+	github.com/google/go-cmp v0.6.0
 	github.com/open-policy-agent/frameworks/constraint v0.0.0-20220218180203-c2a0d8cdf85a
 	github.com/open-policy-agent/opa v1.1.0
 	github.com/sirupsen/logrus v1.9.3

internal/commands/create_test.go

@@ -0,0 +1,118 @@
+package commands
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	log "github.com/sirupsen/logrus/hooks/test"
+
+	"github.com/plexsystems/konstraint/internal/rego"
+)
+
+func TestRenderConstraint(t *testing.T) {
+	_, entry := log.NewNullLogger()
+
+	violations, err := GetViolations()
+	if err != nil {
+		t.Errorf("Error getting violations: %v", err)
+	}
+
+	expected, err := os.ReadFile("../../test/constraint_Test.yaml")
+	if err != nil {
+		t.Errorf("Error reading expected file: %v", err)
+	}
+
+	// Need to remove carriage return for testing on Windows
+	expected = bytes.Replace(expected, []byte("\r"), []byte(""), -1)
+
+	actual, err := renderConstraint(violations[0], "", entry.LastEntry())
+	if err != nil {
+		t.Errorf("Error rendering constraint: %v", err)
+	}
+
+	if !bytes.Equal(actual, expected) {
+		t.Errorf("Unexpected rendered template:\n %v", cmp.Diff(string(expected), string(actual)))
+	}
+}
+
+func TestRenderConstraintWithCustomTemplate(t *testing.T) {
+	_, entry := log.NewNullLogger()
+
+	violations, err := GetViolations()
+	if err != nil {
+		t.Errorf("Error getting violations: %v", err)
+	}
+
+	expected, err := os.ReadFile("../../test/custom/constraint_Test.yaml")
+	if err != nil {
+		t.Errorf("Error reading expected file: %v", err)
+	}
+
+	actual, err := renderConstraint(violations[0], "constraint_template.tpl", entry.LastEntry())
+	if err != nil {
+		t.Errorf("Error rendering constraint: %v", err)
+	}
+
+	if !bytes.Equal(actual, expected) {
+		t.Errorf("Unexpected rendered template:\n %v", cmp.Diff(string(expected), string(actual)))
+	}
+}
+
+func TestRenderConstraintTemplate(t *testing.T) {
+	_, entry := log.NewNullLogger()
+
+	violations, err := GetViolations()
+	if err != nil {
+		t.Errorf("Error getting violations: %v", err)
+	}
+
+	expected, err := os.ReadFile("../../test/template_Test.yaml")
+	if err != nil {
+		t.Errorf("Error reading expected file: %v", err)
+	}
+
+	// Need to remove carriage return for testing on windows
+	expected = bytes.Replace(expected, []byte("\r"), []byte(""), -1)
+
+	actual, err := renderConstraintTemplate(violations[0], "v1beta1", "", entry.LastEntry())
+	if err != nil {
+		t.Errorf("Error rendering constrainttemplate: %v", err)
+	}
+
+	if !bytes.Equal(actual, expected) {
+		t.Errorf("Unexpected rendered template:\n %v", cmp.Diff(string(expected), string(actual)))
+	}
+}
+
+func TestRenderConstraintTemplateWithCustomTemplate(t *testing.T) {
+	_, entry := log.NewNullLogger()
+
+	violations, err := GetViolations()
+	if err != nil {
+		t.Errorf("Error getting violations: %v", err)
+	}
+
+	expected, err := os.ReadFile("../../test/custom/template_Test.yaml")
+	if err != nil {
+		t.Errorf("Error reading expected file: %v", err)
+	}
+
+	actual, err := renderConstraintTemplate(violations[0], "v1", "constrainttemplate_template.tpl", entry.LastEntry())
+	if err != nil {
+		t.Errorf("Error rendering constrainttemplate: %v", err)
+	}
+
+	if !bytes.Equal(actual, expected) {
+		t.Errorf("Unexpected rendered template:\n %v", cmp.Diff(string(expected), string(actual)))
+	}
+}
+
+func GetViolations() ([]rego.Rego, error) {
+	violations, err := rego.GetViolations("../../test")
+	if err != nil {
+		return nil, err
+	}
+	return violations, nil
+}