fix: schema import walk wasn't calling next() (#581)

block · Nov 9, 2023 · ab4d8c6 · ab4d8c6
1 parent 01ce2c3
commit ab4d8c6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/backend/schema/schema.go b/backend/schema/schema.go
@@ -248,7 +248,7 @@ func (m *Module) Imports() []string {
 
 		default:
 		}
-		return nil
+		return next()
 	})
 	return maps.Keys(imports)
 }

diff --git a/backend/schema/schema_test.go b/backend/schema/schema_test.go
@@ -89,6 +89,19 @@ module todo {
 	assert.Equal(t, strings.TrimSpace(expected), strings.TrimSpace(schema.String()))
 }
 
+func TestImports(t *testing.T) {
+	input := `
+	module test {
+		data Data {
+			ref other.Data
+		}
+	}
+	`
+	schema, err := ParseModuleString("", input)
+	assert.NoError(t, err)
+	assert.Equal(t, []string{"other"}, schema.Imports())
+}
+
 func TestVisit(t *testing.T) {
 	expected := `
 Schema