diff options
| author | Oleg Doronin <[email protected]> | 2024-07-23 16:33:51 +0200 |
|---|---|---|
| committer | GitHub <[email protected]> | 2024-07-23 17:33:51 +0300 |
| commit | 9564ab2b008dbcd69f9bc3bce5cc0b03e68b3e1b (patch) | |
| tree | a533a60b1b98e8f331d501471f5c735fd53e3479 | |
| parent | 4cdd4a79d817f20343e45a6749211204e28cebe9 (diff) | |
test_kill_pq_bill has been improved (#6984)
| -rw-r--r-- | ydb/tests/fq/yds/test_kill_pq_bill.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ydb/tests/fq/yds/test_kill_pq_bill.py b/ydb/tests/fq/yds/test_kill_pq_bill.py index 04900344200..e43f310ec37 100644 --- a/ydb/tests/fq/yds/test_kill_pq_bill.py +++ b/ydb/tests/fq/yds/test_kill_pq_bill.py @@ -3,6 +3,7 @@ import json import os import pytest +import time from ydb.tests.tools.fq_runner.kikimr_utils import yq_v1 from ydb.tests.tools.datastreams_helpers.test_yds_base import TestYdsBase @@ -36,6 +37,14 @@ class TestKillPqBill(TestYdsBase): self.write_stream(data_1mb) self.read_stream(message_count) + # TODO: fix this place. We need to correct to account for the ingress bytes in case of an aborted query. + for _ in range(20): + stat = json.loads(client.describe_query(query_id).result.query.statistics.json) + graph_name = "Graph=0" + if graph_name in stat and "IngressBytes" in stat[graph_name]: + break + time.sleep(1) + client.abort_query(query_id) client.wait_query_status(query_id, fq.QueryMeta.ABORTED_BY_USER) |