feat: Sites feature backend

test/sqlite_schema_unchanged_test.go

@@ -163,7 +163,7 @@ func TestSQLiteSchemaIsUntouchedByTheModelChanges(t *testing.T) {
 		seedLegacyRows(t, db)
 		before := takeSqliteSnapshot(t, db)
 
-		if err := database.Migrate(db); err != nil {
+		if err := database.Migrate(db, migrationPublicURL); err != nil {
 			t.Fatalf("migrate: %v", err)
 		}
 		after := takeSqliteSnapshot(t, db)
@@ -173,17 +173,38 @@ func TestSQLiteSchemaIsUntouchedByTheModelChanges(t *testing.T) {
 				delete(after.schema, name)
 			}
 		}
+		for _, name := range []string{"articles", "idx_articles_site_id"} {
+			delete(before.schema, name)
+			delete(after.schema, name)
+		}
 		if !reflect.DeepEqual(before.schema, after.schema) {
 			t.Errorf("schema changed:\nbefore %v\nafter  %v", before.schema, after.schema)
 		}
+		assertSiteColumnAdded(t, db)
+		delete(after.rows["articles"][0], "site_id")
 		if !reflect.DeepEqual(before.rows, after.rows) {
 			t.Errorf("rows changed:\nbefore %v\nafter  %v", before.rows, after.rows)
 		}
 	})
 }
 
+func assertSiteColumnAdded(t *testing.T, db *gorm.DB) {
+	t.Helper()
+	var columns []string
+	if err := db.Raw("SELECT name FROM pragma_table_info('articles') ORDER BY cid").Scan(&columns).Error; err != nil {
+		t.Fatalf("read article columns: %v", err)
+	}
+	if columns[len(columns)-1] != "site_id" {
+		t.Errorf("expected site_id to be the only added article column, got %v", columns)
+	}
+	var sitesTable int
+	if err := db.Raw("SELECT COUNT(*) FROM sqlite_master WHERE type = 'table' AND name = 'sites'").Scan(&sitesTable).Error; err != nil || sitesTable != 1 {
+		t.Errorf("expected the sites table to exist: %d %v", sitesTable, err)
+	}
+}
+
 func isNewTableObject(name string) bool {
-	for _, prefix := range []string{"audit_", "idx_audit_", "image_variant", "idx_image_variant", "idx_images_path"} {
+	for _, prefix := range []string{"audit_", "idx_audit_", "image_variant", "idx_image_variant", "idx_images_path", "idx_sites_key", "sites"} {
 		if strings.HasPrefix(name, prefix) {
 			return true
 		}