285 lines
10 KiB
Go
285 lines
10 KiB
Go
package psql
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func simpleInsert(t *testing.T) {
|
|
gql := `mutation {
|
|
user(insert: $data) {
|
|
id
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "users" AS (WITH "input" AS (SELECT {{data}}::json AS j) INSERT INTO "users" ("full_name", "email") SELECT "full_name", "email" FROM input i, json_populate_record(NULL::users, i.j) t RETURNING *) SELECT json_object_agg('user', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "users_0"."id" AS "id") AS "sel_0")) AS "sel_json_0" FROM (SELECT "users"."id" FROM "users") AS "users_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"data": json.RawMessage(`{"email": "reannagreenholt@orn.com", "full_name": "Flo Barton"}`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func singleInsert(t *testing.T) {
|
|
gql := `mutation {
|
|
product(id: 15, insert: $insert) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{insert}}::json AS j) INSERT INTO "products" ("name", "description", "user_id") SELECT "name", "description", "user_id" FROM input i, json_populate_record(NULL::products, i.j) t RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"insert": json.RawMessage(` { "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc", "user_id": 5 }`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "anon")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func bulkInsert(t *testing.T) {
|
|
gql := `mutation {
|
|
product(name: "test", id: 15, insert: $insert) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{insert}}::json AS j) INSERT INTO "products" ("name", "description") SELECT "name", "description" FROM input i, json_populate_recordset(NULL::products, i.j) t RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"insert": json.RawMessage(` [{ "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc" }]`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "anon")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func singleUpsert(t *testing.T) {
|
|
gql := `mutation {
|
|
product(id: 15, upsert: $upsert) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{upsert}}::json AS j) INSERT INTO "products" ("name", "description") SELECT "name", "description" FROM input i, json_populate_record(NULL::products, i.j) t ON CONFLICT DO (id) DO UPDATE SET name = EXCLUDED.name, description = EXCLUDED.description RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"upsert": json.RawMessage(` { "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc" }`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func bulkUpsert(t *testing.T) {
|
|
gql := `mutation {
|
|
product(id: 15, upsert: $upsert) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{upsert}}::json AS j) INSERT INTO "products" ("name", "description") SELECT "name", "description" FROM input i, json_populate_recordset(NULL::products, i.j) t ON CONFLICT DO (id) DO UPDATE SET name = EXCLUDED.name, description = EXCLUDED.description RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"upsert": json.RawMessage(` [{ "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc" }]`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func singleUpdate(t *testing.T) {
|
|
gql := `mutation {
|
|
product(id: 15, update: $update, where: { id: { eq: 1 } }) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{update}}::json AS j) UPDATE "products" SET ("name", "description") = (SELECT "name", "description" FROM input i, json_populate_record(NULL::products, i.j) t) WHERE (("products"."id") = 1) AND (("products"."id") = 15) RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"update": json.RawMessage(` { "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc" }`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "anon")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func delete(t *testing.T) {
|
|
gql := `mutation {
|
|
product(delete: true, where: { id: { eq: 1 } }) {
|
|
id
|
|
name
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (DELETE FROM "products" WHERE (("products"."price") > 0) AND (("products"."price") < 8) AND (("products"."id") = 1) RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id", "products_0"."name" AS "name") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id", "products"."name" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"update": json.RawMessage(` { "name": "my_name", "woo": { "hoo": "goo" }, "description": "my_desc" }`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func blockedInsert(t *testing.T) {
|
|
gql := `mutation {
|
|
user(insert: $data) {
|
|
id
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "users" AS (WITH "input" AS (SELECT {{data}}::json AS j) INSERT INTO "users" ("full_name", "email") SELECT "full_name", "email" FROM input i, json_populate_record(NULL::users, i.j) t WHERE false RETURNING *) SELECT json_object_agg('user', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "users_0"."id" AS "id") AS "sel_0")) AS "sel_json_0" FROM (SELECT "users"."id" FROM "users") AS "users_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"data": json.RawMessage(`{"email": "reannagreenholt@orn.com", "full_name": "Flo Barton"}`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "bad_dude")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func blockedUpdate(t *testing.T) {
|
|
gql := `mutation {
|
|
user(where: { id: { lt: 5 } }, update: $data) {
|
|
id
|
|
email
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "users" AS (WITH "input" AS (SELECT {{data}}::json AS j) UPDATE "users" SET ("full_name", "email") = (SELECT "full_name", "email" FROM input i, json_populate_record(NULL::users, i.j) t) WHERE false RETURNING *) SELECT json_object_agg('user', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "users_0"."id" AS "id", "users_0"."email" AS "email") AS "sel_0")) AS "sel_json_0" FROM (SELECT "users"."id", "users"."email" FROM "users") AS "users_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"data": json.RawMessage(`{"email": "reannagreenholt@orn.com", "full_name": "Flo Barton"}`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "bad_dude")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func simpleInsertWithPresets(t *testing.T) {
|
|
gql := `mutation {
|
|
product(insert: $data) {
|
|
id
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{data}}::json AS j) INSERT INTO "products" ("name", "price", "created_at", "updated_at", "user_id") SELECT "name", "price", 'now', 'now', '$user_id' FROM input i, json_populate_record(NULL::products, i.j) t RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"data": json.RawMessage(`{"name": "Tomato", "price": 5.76}`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func simpleUpdateWithPresets(t *testing.T) {
|
|
gql := `mutation {
|
|
product(update: $data) {
|
|
id
|
|
}
|
|
}`
|
|
|
|
sql := `WITH "products" AS (WITH "input" AS (SELECT {{data}}::json AS j) UPDATE "products" SET ("name", "price", "updated_at") = (SELECT "name", "price", 'now' FROM input i, json_populate_record(NULL::products, i.j) t) WHERE (("products"."user_id") = {{user_id}}) RETURNING *) SELECT json_object_agg('product', sel_json_0) FROM (SELECT row_to_json((SELECT "sel_0" FROM (SELECT "products_0"."id" AS "id") AS "sel_0")) AS "sel_json_0" FROM (SELECT "products"."id" FROM "products") AS "products_0") AS "done_1337"`
|
|
|
|
vars := map[string]json.RawMessage{
|
|
"data": json.RawMessage(`{"name": "Apple", "price": 1.25}`),
|
|
}
|
|
|
|
resSQL, err := compileGQLToPSQL(gql, vars, "user")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(string(resSQL))
|
|
|
|
if string(resSQL) != sql {
|
|
t.Fatal(errNotExpected)
|
|
}
|
|
}
|
|
|
|
func TestCompileMutate(t *testing.T) {
|
|
t.Run("simpleInsert", simpleInsert)
|
|
t.Run("singleInsert", singleInsert)
|
|
t.Run("bulkInsert", bulkInsert)
|
|
t.Run("singleUpdate", singleUpdate)
|
|
t.Run("singleUpsert", singleUpsert)
|
|
t.Run("bulkUpsert", bulkUpsert)
|
|
t.Run("delete", delete)
|
|
t.Run("blockedInsert", blockedInsert)
|
|
t.Run("blockedUpdate", blockedUpdate)
|
|
t.Run("simpleInsertWithPresets", simpleInsertWithPresets)
|
|
t.Run("simpleUpdateWithPresets", simpleUpdateWithPresets)
|
|
|
|
}
|