@@ -14,7 +14,7 @@ import Type.Proxy (Proxy(..))
1414import Yoga.Om as Om
1515import Yoga.Postgres as PG
1616import Yoga.Postgres.ClientOm (run, runOne, exec)
17−import Yoga.Postgres.Schema (Table, PrimaryKey, AutoIncrement, Unique, Default, TimestampTZ, class CreateTableDDL, createTableDDL, from, into, select, where_, orderBy, limit, groupBy, insert, set, delete, returning, union)
17+import Yoga.Postgres.Schema (Table, PrimaryKey, AutoIncrement, Unique, Default, TimestampTZ, class CreateTableDDL, createTableDDL, from, into, update, select, where_, orderBy, limit, groupBy, insert, set, delete, returning, union)
1818
1919spec :: Spec Unit
2020spec = describe "Query builder" do
@@ -72,13 +72,13 @@ spec = describe "Query builder" do
7272 it "updates a row with SET and WHERE" $ withUsersTable do
7373 _ <- into usersTable # insert { name: "Alice", email: "a@t.com", role: "admin" } # exec {}
7474 _ <- into usersTable # insert { name: "Bob", email: "b@t.com", role: "user" } # exec {}
75− _ <- from usersTable # set { role: "mod" } # where_ @"name = $name" # exec { name: "Bob" }
75+ _ <- update usersTable # set { role: "mod" } # where_ @"name = $name" # exec { name: "Bob" }
7676 rows <- from usersTable # select @"name, role" # orderBy @"name" # run {}
7777 rows `shouldEqual` [ { name: "Alice", role: "admin" }, { name: "Bob", role: "mod" } ]
7878
7979 it "updates multiple columns" $ withUsersTable do
8080 _ <- into usersTable # insert { name: "Alice", email: "a@t.com", role: "admin" } # exec {}
81− _ <- from usersTable # set { name: "Alicia", role: "mod" } # where_ @"email = $email" # exec { email: "a@t.com" }
81+ _ <- update usersTable # set { name: "Alicia", role: "mod" } # where_ @"email = $email" # exec { email: "a@t.com" }
8282 row <- from usersTable # select @"name, role" # runOne {}
8383 r <- shouldBeJust row
8484 r `shouldEqual` { name: "Alicia", role: "mod" }