diff --git a/ballista/client/tests/context_unsupported.rs b/ballista/client/tests/context_unsupported.rs index 805e81325..347071e55 100644 --- a/ballista/client/tests/context_unsupported.rs +++ b/ballista/client/tests/context_unsupported.rs @@ -211,4 +211,45 @@ mod unsupported { assert_batches_eq!(expected, &result); } + + /// looks like `ctx.enable_url_table()` changes session context id. + /// + /// Error returned: + /// ``` + /// Failed to load SessionContext for session ID b5530099-63d1-43b1-9e11-87ac83bb33e5: + /// General error: No session for b5530099-63d1-43b1-9e11-87ac83bb33e5 found + /// ``` + #[rstest] + #[case::standalone(standalone_context())] + #[case::remote(remote_context())] + #[tokio::test] + #[should_panic] + async fn should_execute_sql_show_with_url_table( + #[future(awt)] + #[case] + ctx: SessionContext, + test_data: String, + ) { + let ctx = ctx.enable_url_table(); + + let result = ctx + .sql(&format!("select string_col, timestamp_col from '{test_data}/alltypes_plain.parquet' where id > 4")) + .await + .unwrap() + .collect() + .await + .unwrap(); + + let expected = [ + "+------------+---------------------+", + "| string_col | timestamp_col |", + "+------------+---------------------+", + "| 31 | 2009-03-01T00:01:00 |", + "| 30 | 2009-04-01T00:00:00 |", + "| 31 | 2009-04-01T00:01:00 |", + "+------------+---------------------+", + ]; + + assert_batches_eq!(expected, &result); + } }