Add sqlite-lint to 'pkg/schema/lint'

pkg/schema/lint/checks_test.go

@@ -0,0 +1,66 @@
+package lint_test
+
+import (
+	"slices"
+	"testing"
+
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/stretchr/testify/require"
+
+	"git.offline-twitter.com/offline-labs/gas-stack/pkg/schema"
+	"git.offline-twitter.com/offline-labs/gas-stack/pkg/schema/lint"
+)
+
+func TestFailureCases(t *testing.T) {
+	test_cases := []struct {
+		sqlFile          string
+		expectedFailures []string
+	}{
+		{"test_schemas/failure-has-foreign-key-no-index.sql", []string{"require_indexes_for_foreign_keys"}},
+		{"test_schemas/failure-has-ints.sql", []string{"forbid_int_type"}},
+		{"test_schemas/failure-has-nulls.sql", []string{"require_not_null"}},
+		{"test_schemas/failure-no-strict.sql", []string{"require_strict"}},
+		{"test_schemas/failure-total.sql", []string{
+			"require_not_null",
+			"require_explicit_primary_key",
+			"forbid_int_type",
+			"require_strict",
+			"require_indexes_for_foreign_keys",
+		}},
+	}
+
+	for _, test_case := range test_cases {
+		t.Run(test_case.sqlFile, func(t *testing.T) {
+			schema, err := schema.SchemaFromSQLFile(test_case.sqlFile)
+			require.NoError(t, err)
+
+			for _, check := range lint.Checks {
+				results := check.Execute(schema)
+
+				is_failure := len(results) > 0
+				is_failure_expected := slices.Contains(test_case.expectedFailures, check.Name)
+
+				if is_failure != is_failure_expected {
+					if is_failure_expected {
+						t.Errorf("Expected check '%s' to fail, but it passed: %s", check.Name, test_case.sqlFile)
+					} else {
+						t.Errorf("Expected check '%s' to pass, but it failed: %s (%q.%q)", check.Name, test_case.sqlFile, results[0].TableName, results[0].ColumnName)
+					}
+				}
+			}
+		})
+	}
+}
+
+func TestSuccessCase(t *testing.T) {
+	file := "test_schemas/success.sql"
+	schema, err := schema.SchemaFromSQLFile(file)
+	require.NoError(t, err)
+
+	for _, check := range lint.Checks {
+		results := check.Execute(schema)
+		for _, r := range results {
+			t.Errorf("Unexpected error in file %q: %#v", file, r)
+		}
+	}
+}