Generated column UPDATE fixes (#112)

Author: stephencelis

Sources/StructuredQueriesCore/Statements/Update.swift

@@ -68,8 +68,8 @@ extension PrimaryKeyedTable {
     _ row: Self
   ) -> UpdateOf<Self> {
     update(or: conflictResolution) { updates in
-      for column in TableColumns.allColumns where column.name != columns.primaryKey.name {
-        func open<Root, Value>(_ column: some TableColumnExpression<Root, Value>) {
+      for column in TableColumns.writableColumns where column.name != columns.primaryKey.name {
+        func open<Root, Value>(_ column: some WritableTableColumnExpression<Root, Value>) {
           updates.set(
             column,
             Value(queryOutput: (row as! Root)[keyPath: column.keyPath]).queryFragment

Tests/StructuredQueriesTests/PrimaryKeyedTableTests.swift

@@ -252,9 +252,53 @@ extension SnapshotTests {
         └───────────────────────────────────────────────────┘
         """
       }
-    }
-
-    @Test func tableAliasGenerated() {
+      assertQuery(
+        Row
+          .update(Row(id: UUID(2), isDeleted: true, isNotDeleted: false))
+          .returning(\.self)
+      ) {
+        """
+        UPDATE "rows"
+        SET "isDeleted" = 1
+        WHERE ("rows"."id" = '00000000-0000-0000-0000-000000000002')
+        RETURNING "id", "isDeleted", "isNotDeleted"
+        """
+      } results: {
+        """
+        ┌───────────────────────────────────────────────────┐
+        │ Row(                                              │
+        │   id: UUID(00000000-0000-0000-0000-000000000002), │
+        │   isDeleted: true,                                │
+        │   isNotDeleted: false                             │
+        │ )                                                 │
+        └───────────────────────────────────────────────────┘
+        """
+      }
+      assertQuery(
+        Row
+          .upsert { Row.Draft(id: UUID(2), isDeleted: false) }
+          .returning(\.self)
+      ) {
+        """
+        INSERT INTO "rows"
+        ("id", "isDeleted")
+        VALUES
+        ('00000000-0000-0000-0000-000000000002', 0)
+        ON CONFLICT ("id")
+        DO UPDATE SET "isDeleted" = "excluded"."isDeleted"
+        RETURNING "id", "isDeleted", "isNotDeleted"
+        """
+      } results: {
+        """
+        ┌───────────────────────────────────────────────────┐
+        │ Row(                                              │
+        │   id: UUID(00000000-0000-0000-0000-000000000002), │
+        │   isDeleted: false,                               │
+        │   isNotDeleted: true                              │
+        │ )                                                 │
+        └───────────────────────────────────────────────────┘
+        """
+      }
       enum R: AliasName {}
       assertQuery(
         Row.as(R.self).select(\.isNotDeleted)
@@ -265,7 +309,10 @@ extension SnapshotTests {
         """
       } results: {
         """
-        no such table: rows
+        ┌──────┐
+        │ true │
+        │ true │
+        └──────┘
         """
       }
     }