mirror of
https://github.com/launchbadge/sqlx.git
synced 2025-10-03 07:45:30 +00:00
ed56622526
* reproduce incorrect nullability for materialized views * split ephemeral/index-only table handling from real table handling * add test for literal null, expect nullability to be identified from table information * gather interpreter state into a struct, no change in behaviour * prevent infinite loops that could arise once branching is supported * track nullability alongside the datatype instead of in a separate lookup * implement basic comprehension of branching opcodes * fix datatype calculation of aggregates which are never 'stepped' through * implement coroutine and return operations, including tracking of 'actual' integer value stored in the register by Integer/InitCoroutine/Yield operations. * strip unnecessary history field out * Modify variable test to expect bind-variable outputs to be nullable, rather than unknown * add partially commented-out union tests, simplify code to satisfy simplest union case * fix unit test incorrectly expecting primary keys to be implicitly not-null * add failing test for recursive tables * add logging of query explain plan * track explain plan execution history * broken RowData implementation (doesn't alias) * Implement OpenPseudo tables as an alias of a register value * fix comment * clean up logging code warnings * use cfg to omit QueryPlanLogger unless sqlite feature is used
383 lines
12 KiB
Rust
383 lines
12 KiB
Rust
use sqlx::error::DatabaseError;
|
|
use sqlx::sqlite::{SqliteConnectOptions, SqliteError};
|
|
use sqlx::ConnectOptions;
|
|
use sqlx::TypeInfo;
|
|
use sqlx::{sqlite::Sqlite, Column, Executor};
|
|
use sqlx_test::new;
|
|
use std::env;
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_simple() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let info = conn.describe("SELECT * FROM tweet").await?;
|
|
let columns = info.columns();
|
|
|
|
assert_eq!(columns[0].name(), "id");
|
|
assert_eq!(columns[1].name(), "text");
|
|
assert_eq!(columns[2].name(), "is_sent");
|
|
assert_eq!(columns[3].name(), "owner_id");
|
|
|
|
assert_eq!(columns[0].ordinal(), 0);
|
|
assert_eq!(columns[1].ordinal(), 1);
|
|
assert_eq!(columns[2].ordinal(), 2);
|
|
assert_eq!(columns[3].ordinal(), 3);
|
|
|
|
assert_eq!(info.nullable(0), Some(false));
|
|
assert_eq!(info.nullable(1), Some(false));
|
|
assert_eq!(info.nullable(2), Some(false));
|
|
assert_eq!(info.nullable(3), Some(true)); // owner_id
|
|
|
|
assert_eq!(columns[0].type_info().name(), "INTEGER");
|
|
assert_eq!(columns[1].type_info().name(), "TEXT");
|
|
assert_eq!(columns[2].type_info().name(), "BOOLEAN");
|
|
assert_eq!(columns[3].type_info().name(), "INTEGER");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_variables() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
// without any context, we resolve to NULL
|
|
let info = conn.describe("SELECT ?1").await?;
|
|
|
|
assert_eq!(info.columns()[0].type_info().name(), "NULL");
|
|
assert_eq!(info.nullable(0), Some(true)); // nothing prevents the value from being bound to null
|
|
|
|
// context can be provided by using CAST(_ as _)
|
|
let info = conn.describe("SELECT CAST(?1 AS REAL)").await?;
|
|
|
|
assert_eq!(info.columns()[0].type_info().name(), "REAL");
|
|
assert_eq!(info.nullable(0), Some(true)); // nothing prevents the value from being bound to null
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_expression() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let d = conn
|
|
.describe("SELECT 1 + 10, 5.12 * 2, 'Hello', x'deadbeef', null")
|
|
.await?;
|
|
|
|
let columns = d.columns();
|
|
|
|
assert_eq!(columns[0].type_info().name(), "INTEGER");
|
|
assert_eq!(columns[0].name(), "1 + 10");
|
|
assert_eq!(d.nullable(0), Some(false)); // literal constant
|
|
|
|
assert_eq!(columns[1].type_info().name(), "REAL");
|
|
assert_eq!(columns[1].name(), "5.12 * 2");
|
|
assert_eq!(d.nullable(1), Some(false)); // literal constant
|
|
|
|
assert_eq!(columns[2].type_info().name(), "TEXT");
|
|
assert_eq!(columns[2].name(), "'Hello'");
|
|
assert_eq!(d.nullable(2), Some(false)); // literal constant
|
|
|
|
assert_eq!(columns[3].type_info().name(), "BLOB");
|
|
assert_eq!(columns[3].name(), "x'deadbeef'");
|
|
assert_eq!(d.nullable(3), Some(false)); // literal constant
|
|
|
|
assert_eq!(columns[4].type_info().name(), "NULL");
|
|
assert_eq!(columns[4].name(), "null");
|
|
assert_eq!(d.nullable(4), Some(true)); // literal null
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_expression_from_empty_table() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
conn.execute("CREATE TEMP TABLE _temp_empty ( name TEXT NOT NULL, a INT )")
|
|
.await?;
|
|
|
|
let d = conn
|
|
.describe("SELECT COUNT(*), a + 1, name, 5.12, 'Hello' FROM _temp_empty")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns()[0].type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(false)); // COUNT(*)
|
|
|
|
assert_eq!(d.columns()[1].type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(1), Some(true)); // `a+1` is nullable, because a is nullable
|
|
|
|
assert_eq!(d.columns()[2].type_info().name(), "TEXT");
|
|
assert_eq!(d.nullable(2), Some(true)); // `name` is not nullable, but the query can be null due to zero rows
|
|
|
|
assert_eq!(d.columns()[3].type_info().name(), "REAL");
|
|
assert_eq!(d.nullable(3), Some(false)); // literal constant
|
|
|
|
assert_eq!(d.columns()[4].type_info().name(), "TEXT");
|
|
assert_eq!(d.nullable(4), Some(false)); // literal constant
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_expression_from_empty_table_with_star() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
conn.execute("CREATE TEMP TABLE _temp_empty ( name TEXT, a INT )")
|
|
.await?;
|
|
|
|
let d = conn
|
|
.describe("SELECT *, 5, 'Hello' FROM _temp_empty")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns()[0].type_info().name(), "TEXT");
|
|
assert_eq!(d.columns()[1].type_info().name(), "INTEGER");
|
|
assert_eq!(d.columns()[2].type_info().name(), "INTEGER");
|
|
assert_eq!(d.columns()[3].type_info().name(), "TEXT");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_insert() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let d = conn
|
|
.describe("INSERT INTO tweet (id, text) VALUES (2, 'Hello')")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns().len(), 0);
|
|
|
|
let d = conn
|
|
.describe("INSERT INTO tweet (id, text) VALUES (2, 'Hello'); SELECT last_insert_rowid();")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns().len(), 1);
|
|
assert_eq!(d.columns()[0].type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(false));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_insert_with_read_only() -> anyhow::Result<()> {
|
|
sqlx_test::setup_if_needed();
|
|
|
|
let mut options: SqliteConnectOptions = env::var("DATABASE_URL")?.parse().unwrap();
|
|
options = options.read_only(true);
|
|
|
|
let mut conn = options.connect().await?;
|
|
|
|
let d = conn
|
|
.describe("INSERT INTO tweet (id, text) VALUES (2, 'Hello')")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns().len(), 0);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_insert_with_returning() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let d = conn
|
|
.describe("INSERT INTO tweet (id, text) VALUES (2, 'Hello') RETURNING *")
|
|
.await?;
|
|
|
|
assert_eq!(d.columns().len(), 4);
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.column(1).type_info().name(), "TEXT");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_bad_statement() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let err = conn.describe("SELECT 1 FROM not_found").await.unwrap_err();
|
|
let err = err
|
|
.as_database_error()
|
|
.unwrap()
|
|
.downcast_ref::<SqliteError>();
|
|
|
|
assert_eq!(err.message(), "no such table: not_found");
|
|
assert_eq!(err.code().as_deref(), Some("1"));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_left_join() -> anyhow::Result<()> {
|
|
let mut conn = new::<Sqlite>().await?;
|
|
|
|
let d = conn.describe("select accounts.id from accounts").await?;
|
|
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(false));
|
|
|
|
let d = conn
|
|
.describe("select tweet.id from accounts left join tweet on owner_id = accounts.id")
|
|
.await?;
|
|
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(true));
|
|
|
|
let d = conn
|
|
.describe(
|
|
"select tweet.id, accounts.id from accounts left join tweet on owner_id = accounts.id",
|
|
)
|
|
.await?;
|
|
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(true));
|
|
|
|
assert_eq!(d.column(1).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(1), Some(false));
|
|
|
|
let d = conn
|
|
.describe(
|
|
"select tweet.id, accounts.id from accounts inner join tweet on owner_id = accounts.id",
|
|
)
|
|
.await?;
|
|
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(false));
|
|
|
|
assert_eq!(d.column(1).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(1), Some(false));
|
|
|
|
let d = conn
|
|
.describe(
|
|
"select tweet.id, accounts.id from accounts left join tweet on tweet.id = accounts.id",
|
|
)
|
|
.await?;
|
|
|
|
assert_eq!(d.column(0).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(0), Some(true));
|
|
|
|
assert_eq!(d.column(1).type_info().name(), "INTEGER");
|
|
assert_eq!(d.nullable(1), Some(false));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_literal_subquery() -> anyhow::Result<()> {
|
|
async fn assert_literal_described(
|
|
conn: &mut sqlx::SqliteConnection,
|
|
query: &str,
|
|
) -> anyhow::Result<()> {
|
|
let info = conn.describe(query).await?;
|
|
|
|
assert_eq!(info.column(0).type_info().name(), "TEXT", "{}", query);
|
|
assert_eq!(info.nullable(0), Some(false), "{}", query);
|
|
assert_eq!(info.column(1).type_info().name(), "NULL", "{}", query);
|
|
assert_eq!(info.nullable(1), Some(true), "{}", query);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
let mut conn = new::<Sqlite>().await?;
|
|
assert_literal_described(&mut conn, "SELECT 'a', NULL").await?;
|
|
assert_literal_described(&mut conn, "SELECT * FROM (SELECT 'a', NULL)").await?;
|
|
assert_literal_described(
|
|
&mut conn,
|
|
"WITH cte AS (SELECT 'a', NULL) SELECT * FROM cte",
|
|
)
|
|
.await?;
|
|
assert_literal_described(
|
|
&mut conn,
|
|
"WITH cte AS MATERIALIZED (SELECT 'a', NULL) SELECT * FROM cte",
|
|
)
|
|
.await?;
|
|
assert_literal_described(
|
|
&mut conn,
|
|
"WITH RECURSIVE cte(a,b) AS (SELECT 'a', NULL UNION ALL SELECT a||a, NULL FROM cte WHERE length(a)<3) SELECT * FROM cte",
|
|
)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_table_subquery() -> anyhow::Result<()> {
|
|
async fn assert_tweet_described(
|
|
conn: &mut sqlx::SqliteConnection,
|
|
query: &str,
|
|
) -> anyhow::Result<()> {
|
|
let info = conn.describe(query).await?;
|
|
let columns = info.columns();
|
|
|
|
assert_eq!(columns[0].name(), "id", "{}", query);
|
|
assert_eq!(columns[1].name(), "text", "{}", query);
|
|
assert_eq!(columns[2].name(), "is_sent", "{}", query);
|
|
assert_eq!(columns[3].name(), "owner_id", "{}", query);
|
|
|
|
assert_eq!(columns[0].ordinal(), 0, "{}", query);
|
|
assert_eq!(columns[1].ordinal(), 1, "{}", query);
|
|
assert_eq!(columns[2].ordinal(), 2, "{}", query);
|
|
assert_eq!(columns[3].ordinal(), 3, "{}", query);
|
|
|
|
assert_eq!(info.nullable(0), Some(false), "{}", query);
|
|
assert_eq!(info.nullable(1), Some(false), "{}", query);
|
|
assert_eq!(info.nullable(2), Some(false), "{}", query);
|
|
assert_eq!(info.nullable(3), Some(true), "{}", query);
|
|
|
|
assert_eq!(columns[0].type_info().name(), "INTEGER", "{}", query);
|
|
assert_eq!(columns[1].type_info().name(), "TEXT", "{}", query);
|
|
assert_eq!(columns[2].type_info().name(), "BOOLEAN", "{}", query);
|
|
assert_eq!(columns[3].type_info().name(), "INTEGER", "{}", query);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
let mut conn = new::<Sqlite>().await?;
|
|
assert_tweet_described(&mut conn, "SELECT * FROM tweet").await?;
|
|
assert_tweet_described(&mut conn, "SELECT * FROM (SELECT * FROM tweet)").await?;
|
|
assert_tweet_described(
|
|
&mut conn,
|
|
"WITH cte AS (SELECT * FROM tweet) SELECT * FROM cte",
|
|
)
|
|
.await?;
|
|
assert_tweet_described(
|
|
&mut conn,
|
|
"WITH cte AS MATERIALIZED (SELECT * FROM tweet) SELECT * FROM cte",
|
|
)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx_macros::test]
|
|
async fn it_describes_union() -> anyhow::Result<()> {
|
|
async fn assert_union_described(
|
|
conn: &mut sqlx::SqliteConnection,
|
|
query: &str,
|
|
) -> anyhow::Result<()> {
|
|
let info = conn.describe(query).await?;
|
|
|
|
assert_eq!(info.column(0).type_info().name(), "TEXT", "{}", query);
|
|
assert_eq!(info.nullable(0), Some(false), "{}", query);
|
|
assert_eq!(info.column(1).type_info().name(), "TEXT", "{}", query);
|
|
assert_eq!(info.nullable(1), Some(true), "{}", query);
|
|
assert_eq!(info.column(2).type_info().name(), "INTEGER", "{}", query);
|
|
assert_eq!(info.nullable(2), Some(true), "{}", query);
|
|
//TODO: mixed type columns not handled correctly
|
|
//assert_eq!(info.column(3).type_info().name(), "NULL", "{}", query);
|
|
//assert_eq!(info.nullable(3), Some(false), "{}", query);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
let mut conn = new::<Sqlite>().await?;
|
|
assert_union_described(
|
|
&mut conn,
|
|
"SELECT 'txt','a',null,'b' UNION ALL SELECT 'int',NULL,1,2 ",
|
|
)
|
|
.await?;
|
|
//TODO: insert into temp-table not merging datatype/nullable of all operations - currently keeping last-writer
|
|
//assert_union_described(&mut conn, "SELECT 'txt','a',null,'b' UNION SELECT 'int',NULL,1,2 ").await?;
|
|
|
|
Ok(())
|
|
}
|