Skip to content

Commit 8d342f0

Browse files
committed
zstd test has been added
1 parent bc6e977 commit 8d342f0

2 files changed

Lines changed: 27 additions & 0 deletions

File tree

Binary file not shown.

ydb/tests/fq/s3/test_compressions.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,33 @@ def test_compression(self, kikimr, s3, client, filename, compression, unique_pre
6363
result_set = data.result.result_set
6464
self.validate_result(result_set)
6565

66+
@yq_all
67+
def test_zstd_unknown_frame_descriptor(self, kikimr, s3, client, unique_prefix):
68+
self.create_bucket_and_upload_file("unknown_frame_descriptor.json.zst", s3, kikimr)
69+
storage_connection_name = unique_prefix + "fruitbucket"
70+
client.create_storage_connection(storage_connection_name, "fbucket")
71+
72+
sql = '''
73+
SELECT count(*)
74+
FROM `{}`.`unknown_frame_descriptor.json.zst`
75+
WITH (format=json_each_row, compression="zstd", SCHEMA (
76+
a String NOT NULL
77+
));
78+
'''.format(storage_connection_name)
79+
80+
query_id = client.create_query("simple", sql, type=fq.QueryContent.QueryType.ANALYTICS).result.query_id
81+
client.wait_query_status(query_id, fq.QueryMeta.COMPLETED)
82+
83+
data = client.get_result_data(query_id)
84+
result_set = data.result.result_set
85+
86+
logging.debug(str(result_set))
87+
assert len(result_set.columns) == 1
88+
assert result_set.columns[0].name == "column0"
89+
assert result_set.columns[0].type.type_id == ydb.Type.UINT64
90+
assert len(result_set.rows) == 1
91+
assert result_set.rows[0].items[0].uint64_value == 5458
92+
6693
@yq_all
6794
@pytest.mark.parametrize("client", [{"folder_id": "my_folder"}], indirect=True)
6895
def test_invalid_compression(self, kikimr, s3, client, unique_prefix):

0 commit comments

Comments
 (0)