Skip to content

Commit 39dc22e

Browse files
committed
Added integration tests
1 parent dfb1ed9 commit 39dc22e

File tree

2 files changed

+34
-1
lines changed

2 files changed

+34
-1
lines changed

ydb/tests/fq/s3/test_bindings.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -586,3 +586,36 @@ def test_count_for_pg_binding(self, kikimr, s3, client, pg_syntax):
586586
else:
587587
assert result_set.columns[0].type.type_id == ydb.Type.UINT64
588588
assert result_set.rows[0].items[0].uint64_value == 1
589+
590+
@yq_all
591+
@pytest.mark.parametrize("client", [{"folder_id": "my_folder"}], indirect=True)
592+
def test_ast_in_failed_query_compilation(self, kikimr, s3, client):
593+
resource = boto3.resource(
594+
"s3",
595+
endpoint_url=s3.s3_url,
596+
aws_access_key_id="key",
597+
aws_secret_access_key="secret_key"
598+
)
599+
600+
bucket = resource.Bucket("bindbucket")
601+
bucket.create(ACL='public-read')
602+
bucket.objects.all().delete()
603+
604+
connection_id = client.create_storage_connection("bb", "bindbucket").result.connection_id
605+
606+
data_column = ydb.Column(name="data", type=ydb.Type(type_id=ydb.Type.PrimitiveTypeId.STRING))
607+
client.create_object_storage_binding(name="s3binding",
608+
path="/",
609+
format="raw",
610+
connection_id=connection_id,
611+
columns=[data_column])
612+
613+
sql = R'''
614+
SELECT some_unknown_column FROM bindings.`s3binding`;
615+
'''
616+
617+
query_id = client.create_query("simple", sql, type=fq.QueryContent.QueryType.ANALYTICS).result.query_id
618+
client.wait_query_status(query_id, fq.QueryMeta.FAILED)
619+
620+
ast = str(client.describe_query(query_id).result.query.ast)
621+
assert ast != "", "Query ast not found"

ydb/tests/fq/yds/test_select_1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def test_compile_error(self, client, yq_version):
120120
assert "Failed to parse query" in describe_string, describe_string
121121

122122
@yq_all
123-
def test_ast_in_failed_query(self, client):
123+
def test_ast_in_failed_query_runtime(self, client):
124124
sql = "SELECT unwrap(1 / 0)"
125125

126126
query_id = client.create_query("simple", sql, type=fq.QueryContent.QueryType.ANALYTICS).result.query_id

0 commit comments

Comments
 (0)