Skip to content

Commit

Permalink
add scalar coercion tests + handle coercion errors
Browse files Browse the repository at this point in the history
  • Loading branch information
@Weakky
Weakky committed Nov 30, 2023
1 parent 9390be7 commit c0f2c4c
Show file tree
Hide file tree
Showing 7 changed files with 399 additions and 26 deletions.
1 change: 1 addition & 0 deletions Cargo.lock
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions query-engine/connector-test-kit-rs/query-engine-tests/tests/queries/data_types/mod.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,3 +8,4 @@ mod float;
mod int;
mod json;
mod string;
mod through_relation;
286 changes: 286 additions & 0 deletions ...ine/connector-test-kit-rs/query-engine-tests/tests/queries/data_types/through_relation.rs
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,286 @@
use indoc::indoc;
use query_engine_tests::*;

#[test_suite]
mod scalar_relations {
fn schema_common() -> String {
let schema = indoc! {
r#"model Parent {
#id(id, Int, @id)
children Child[]
}
model Child {
#id(childId, Int, @id)
parentId Int?
parent Parent? @relation(fields: [parentId], references: [id])
string String
int Int
bInt BigInt
float Float
bytes Bytes
bool Boolean
dt DateTime
}
"#
};

schema.to_owned()
}

#[connector_test(schema(schema_common))]
async fn common_types(runner: Runner) -> TestResult<()> {
create_common_children(&runner).await?;

insta::assert_snapshot!(
run_query!(&runner, r#"{ findManyParent { id children { childId string int bInt float bytes bool dt } } }"#),
@r###"{"data":{"findManyParent":[{"id":1,"children":[{"childId":1,"string":"abc","int":1,"bInt":"1","float":1.5,"bytes":"AQID","bool":false,"dt":"1900-10-10T01:10:10.001Z"},{"childId":2,"string":"def","int":-4234234,"bInt":"14324324234324","float":-2.54367,"bytes":"FDSF","bool":true,"dt":"1999-12-12T21:12:12.121Z"}]}]}}"###
);

Ok(())
}

fn schema_json() -> String {
let schema = indoc! {
r#"model Parent {
#id(id, Int, @id)
children Child[]
}
model Child {
#id(childId, Int, @id)
parentId Int?
parent Parent? @relation(fields: [parentId], references: [id])
json Json
}
"#
};

schema.to_owned()
}

#[connector_test(schema(schema_json), capabilities(Json), exclude(Mysql(5.6)))]
async fn json_type(runner: Runner) -> TestResult<()> {
create_child(&runner, r#"{ childId: 1, json: "1" }"#).await?;
create_child(&runner, r#"{ childId: 2, json: "{}" }"#).await?;
create_child(&runner, r#"{ childId: 3, json: "{\"a\": \"b\"}" }"#).await?;
create_child(&runner, r#"{ childId: 4, json: "[]" }"#).await?;
create_child(&runner, r#"{ childId: 5, json: "[1, -1, true, {\"a\": \"b\"}]" }"#).await?;
create_parent(
&runner,
r#"{ id: 1, children: { connect: [{ childId: 1 }, { childId: 2 }, { childId: 3 }, { childId: 4 }, { childId: 5 }] } }"#,
)
.await?;

insta::assert_snapshot!(
run_query!(&runner, r#"{ findManyParent(orderBy: { id: asc }) { id children { childId json } } }"#),
@r###"{"data":{"findManyParent":[{"id":1,"children":[{"childId":1,"json":"1"},{"childId":2,"json":"{}"},{"childId":3,"json":"{\"a\":\"b\"}"},{"childId":4,"json":"[]"},{"childId":5,"json":"[1,-1,true,{\"a\":\"b\"}]"}]}]}}"###
);

Ok(())
}

fn schema_enum() -> String {
let schema = indoc! {
r#"model Parent {
#id(id, Int, @id)
children Child[]
}
model Child {
#id(childId, Int, @id)
parentId Int?
parent Parent? @relation(fields: [parentId], references: [id])
enum Color
}
enum Color {
Red
Green
Blue
}
"#
};

schema.to_owned()
}

#[connector_test(schema(schema_enum), capabilities(Enums))]
async fn enum_type(runner: Runner) -> TestResult<()> {
create_child(&runner, r#"{ childId: 1, enum: Red }"#).await?;
create_child(&runner, r#"{ childId: 2, enum: Green }"#).await?;
create_child(&runner, r#"{ childId: 3, enum: Blue }"#).await?;
create_parent(
&runner,
r#"{ id: 1, children: { connect: [{ childId: 1 }, { childId: 2 }, { childId: 3 }] } }"#,
)
.await?;

insta::assert_snapshot!(
run_query!(&runner, r#"{ findManyParent(orderBy: { id :asc }) { id children { childId enum } } }"#),
@r###"{"data":{"findManyParent":[{"id":1,"children":[{"childId":1,"enum":"Red"},{"childId":2,"enum":"Green"},{"childId":3,"enum":"Blue"}]}]}}"###
);

Ok(())
}

fn schema_decimal() -> String {
let schema = indoc! {
r#"model Parent {
#id(id, Int, @id)
children Child[]
}
model Child {
#id(childId, Int, @id)
parentId Int?
parent Parent? @relation(fields: [parentId], references: [id])
dec Decimal
}
"#
};

schema.to_owned()
}

#[connector_test(schema(schema_decimal), capabilities(DecimalType))]
async fn decimal_type(runner: Runner) -> TestResult<()> {
create_child(&runner, r#"{ childId: 1, dec: "1" }"#).await?;
create_child(&runner, r#"{ childId: 2, dec: "-1" }"#).await?;
create_child(&runner, r#"{ childId: 3, dec: "123.45678910" }"#).await?;
create_parent(
&runner,
r#"{ id: 1, children: { connect: [{ childId: 1 }, { childId: 2 }, { childId: 3 }] } }"#,
)
.await?;

insta::assert_snapshot!(
run_query!(&runner, r#"{ findManyParent(orderBy: { id: asc }) { id children { childId dec } } }"#),
@r###"{"data":{"findManyParent":[{"id":1,"children":[{"childId":1,"dec":"1"},{"childId":2,"dec":"-1"},{"childId":3,"dec":"123.4567891"}]}]}}"###
);

Ok(())
}

fn schema_scalar_lists() -> String {
let schema = indoc! {
r#"model Parent {
#id(id, Int, @id)
children Child[]
}
model Child {
#id(childId, Int, @id)
parentId Int?
parent Parent? @relation(fields: [parentId], references: [id])
string String[]
int Int[]
bInt BigInt[]
float Float[]
bytes Bytes[]
bool Boolean[]
dt DateTime[]
}
"#
};

schema.to_owned()
}

#[connector_test(schema(schema_scalar_lists), capabilities(ScalarLists))]
async fn scalar_lists(runner: Runner) -> TestResult<()> {
create_child(
&runner,
r#"{
childId: 1,
string: ["abc", "def"],
int: [1, -1, 1234567],
bInt: [1, -1, 9223372036854775807, -9223372036854775807],
float: [1.5, -1.5, 1.234567],
bytes: ["AQID", "Qk9OSk9VUg=="],
bool: [false, true],
dt: ["1900-10-10T01:10:10.001Z", "1999-12-12T21:12:12.121Z"],
}"#,
)
.await?;
create_parent(&runner, r#"{ id: 1, children: { connect: [{ childId: 1 }] } }"#).await?;

insta::assert_snapshot!(
run_query!(&runner, r#"{ findManyParent { id children { childId string int bInt float bytes bool dt } } }"#),
@r###"{"data":{"findManyParent":[{"id":1,"children":[{"childId":1,"string":["abc","def"],"int":[1,-1,1234567],"bInt":["1","-1","9223372036854775807","-9223372036854775807"],"float":[1.5,-1.5,1.234567],"bytes":["AQID","Qk9OSk9VUg=="],"bool":[false,true],"dt":["1900-10-10T01:10:10.001Z","1999-12-12T21:12:12.121Z"]}]}]}}"###
);

Ok(())
}

async fn create_common_children(runner: &Runner) -> TestResult<()> {
create_child(
&runner,
r#"{
childId: 1,
string: "abc",
int: 1,
bInt: 1,
float: 1.5,
bytes: "AQID",
bool: false,
dt: "1900-10-10T01:10:10.001Z",
}"#,
)
.await?;

create_child(
&runner,
r#"{
childId: 2,
string: "def",
int: -4234234,
bInt: 14324324234324,
float: -2.54367,
bytes: "FDSF",
bool: true,
dt: "1999-12-12T21:12:12.121Z",
}"#,
)
.await?;

create_parent(
&runner,
r#"{ id: 1, children: { connect: [{ childId: 1 }, { childId: 2 }] } }"#,
)
.await?;

Ok(())
}

async fn create_child(runner: &Runner, data: &str) -> TestResult<()> {
runner
.query(format!("mutation {{ createOneChild(data: {}) {{ childId }} }}", data))
.await?
.assert_success();
Ok(())
}

async fn create_parent(runner: &Runner, data: &str) -> TestResult<()> {
runner
.query(format!("mutation {{ createOneParent(data: {}) {{ id }} }}", data))
.await?
.assert_success();
Ok(())
}
}
1 change: 1 addition & 0 deletions query-engine/connectors/sql-query-connector/Cargo.toml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ uuid.workspace = true
opentelemetry = { version = "0.17", features = ["tokio"] }
tracing-opentelemetry = "0.17.3"
cuid = { git = "https://github.com/prisma/cuid-rust", branch = "wasm32-support" }
hex = "0.4"

[target.'cfg(not(target_arch = "wasm32"))'.dependencies]
quaint.workspace = true
Expand Down
Loading

0 comments on commit c0f2c4c

Please sign in to comment.