Add test harness

pkg/composableschemadsl/compiler/importer-test/README.md

@@ -0,0 +1,4 @@
+## Test Structure
+Every folder should have a `root.zed`. This is the target for compilation.
+
+Every folder will have an `expected.zed`, which is the output of the compilation process.

pkg/composableschemadsl/compiler/importer-test/nested-local/root.zed

@@ -0,0 +1,6 @@
+from .user.user import user
+
+definition resource {
+  relation viewer: user
+  permission view = viewer
+}

pkg/composableschemadsl/compiler/importer-test/nested-local/user/user.zed

@@ -0,0 +1 @@
+definition user {}

pkg/composableschemadsl/compiler/importer-test/nested-two-layer-local/definitions/user/user.zed

@@ -0,0 +1 @@
+definition user {}

pkg/composableschemadsl/compiler/importer-test/nested-two-layer-local/root.zed

@@ -0,0 +1,6 @@
+from .definitions.user.user import user
+
+definition resource {
+  relation viewer: user
+  permission view = viewer
+}

pkg/composableschemadsl/compiler/importer-test/simple-local-with-hop/indirection.zed

@@ -0,0 +1 @@
+from .user import user

pkg/composableschemadsl/compiler/importer-test/simple-local-with-hop/root.zed

@@ -0,0 +1,6 @@
+from .indirection import user
+
+definition resource {
+  relation viewer: user
+  permission view = viewer
+}

pkg/composableschemadsl/compiler/importer-test/simple-local-with-hop/user.zed

@@ -0,0 +1 @@
+definition user {}

pkg/composableschemadsl/compiler/importer-test/simple-local/root.zed

@@ -0,0 +1,6 @@
+from .user import user
+
+definition resource {
+  relation viewer: user
+  permission view = viewer
+}

pkg/composableschemadsl/compiler/importer-test/simple-local/user.zed

@@ -0,0 +1 @@
+definition user {}

pkg/composableschemadsl/compiler/importer_test.go

@@ -0,0 +1,80 @@
+package compiler_test
+
+import (
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/authzed/spicedb/pkg/composableschemadsl/input"
+	"github.com/authzed/spicedb/pkg/composableschemadsl/compiler"
+	"github.com/authzed/spicedb/pkg/composableschemadsl/generator"
+)
+
+type importerTest struct {
+	name string
+	folder string
+}
+
+func (it *importerTest) input() string {
+	b, err := os.ReadFile(fmt.Sprintf("importer-test/%s/root.zed", it.folder))
+	if err != nil {
+		panic(err)
+	}
+
+	return string(b)
+}
+
+func (it *importerTest) expected() string {
+	b, err := os.ReadFile(fmt.Sprintf("importer-test/%s/expected.zed", it.folder))
+	if err != nil {
+		panic(err)
+	}
+
+	return string(b)
+}
+
+func (it *importerTest) writeExpected(schema string) {
+	err := os.WriteFile(fmt.Sprintf("importer-test/%s/expected.zed", it.folder), []byte(schema), 0o_600)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func TestImporter(t *testing.T) {
+	importerTests := []importerTest{
+		{"simple local import", "simple-local"},
+		{"simple local import", "simple-local-with-hop"},
+		{"nested local import", "nested-local"},
+		{"nested local two layers deep import", "nested-two-layer-local"},
+	}
+
+	for _, test := range importerTests {
+		t.Run(test.name, func(t *testing.T) {
+			t.Parallel()
+
+			inputSchema := test.input()
+
+			compiled, err := compiler.Compile(compiler.InputSchema{
+				Source:       input.Source("schema"),
+				SchemaString: inputSchema,
+			}, compiler.AllowUnprefixedObjectType())
+			require.NoError(t, err)
+
+			generated, _, err := generator.GenerateSchema(compiled.OrderedDefinitions)
+			require.NoError(t, err)
+
+			if os.Getenv("REGEN") == "true" {
+				test.writeExpected(generated)
+			} else {
+				expected := test.expected()
+				if !assert.Equal(t, expected, generated, test.name) {
+					t.Log(generated)
+				}
+			}
+		})
+	}
+
+}