Add some tests for the Patch function with composite keys

ksql.go

@@ -618,18 +618,7 @@ func normalizeIDsAsMap(idNames []string, idOrMap interface{}) (idMap map[string]
 		}
 	}
 
-	for _, idName := range idNames {
-		id, found := idMap[idName]
-		if !found {
-			return nil, fmt.Errorf("missing required id field `%s` on input record", idName)
-		}
-
-		if id == nil || reflect.ValueOf(id).IsZero() {
-			return nil, fmt.Errorf("invalid value '%v' received for id column: '%s'", id, idName)
-		}
-	}
-
-	return idMap, nil
+	return idMap, validateIfAllIdsArePresent(idNames, idMap)
 }
 
 // Update updates the given instances on the database by id.
@@ -803,6 +792,11 @@ func buildUpdateQuery(
 	numNonIDArgs := numAttrs - len(idFieldNames)
 	whereArgs := args[numNonIDArgs:]
 
+	err = validateIfAllIdsArePresent(idFieldNames, recordMap)
+	if err != nil {
+		return "", nil, err
+	}
+
 	whereQuery := make([]string, len(idFieldNames))
 	for i, fieldName := range idFieldNames {
 		whereArgs[i] = recordMap[fieldName]
@@ -847,6 +841,21 @@ func buildUpdateQuery(
 	return query, args, nil
 }
 
+func validateIfAllIdsArePresent(idNames []string, idMap map[string]interface{}) error {
+	for _, idName := range idNames {
+		id, found := idMap[idName]
+		if !found {
+			return fmt.Errorf("missing required id field `%s` on input record", idName)
+		}
+
+		if id == nil || reflect.ValueOf(id).IsZero() {
+			return fmt.Errorf("invalid value '%v' received for id column: '%s'", id, idName)
+		}
+	}
+
+	return nil
+}
+
 // Exec just runs an SQL command on the database returning no rows.
 func (c DB) Exec(ctx context.Context, query string, params ...interface{}) (Result, error) {
 	return c.db.ExecContext(ctx, query, params...)

test_adapters.go

@@ -1619,6 +1619,37 @@ func PatchTest(
 			err := c.Update(ctx, usersTable, u)
 			tt.AssertNotEqual(t, err, nil)
 		})
+
+		t.Run("should report error if the id is missing", func(t *testing.T) {
+			t.Run("with a single primary key", func(t *testing.T) {
+				db, closer := newDBAdapter(t)
+				defer closer.Close()
+
+				ctx := context.Background()
+				c := newTestDB(db, driver)
+
+				err := c.Update(ctx, usersTable, &user{
+					// Missing ID
+					Name: "Jane",
+				})
+				tt.AssertErrContains(t, err, "invalid value", "0", "'id'")
+			})
+
+			t.Run("with composite keys", func(t *testing.T) {
+				db, closer := newDBAdapter(t)
+				defer closer.Close()
+
+				ctx := context.Background()
+				c := newTestDB(db, driver)
+
+				err := c.Update(ctx, NewTable("user_permissions", "id", "user_id", "perm_id"), &userPermission{
+					ID: 1,
+					// Missing UserID
+					PermID: 42,
+				})
+				tt.AssertErrContains(t, err, "invalid value", "0", "'user_id'")
+			})
+		})
 	})
 }