Fix keepermap test on 24.3

Aiven-Open · Oct 24, 2024 · ff9639f · ff9639f
1 parent 7d39946
commit ff9639f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/astacus/coordinator/plugins/clickhouse/client.py b/astacus/coordinator/plugins/clickhouse/client.py
@@ -78,7 +78,7 @@ async def execute(self, query: bytes, timeout: float | None = None, session_id:
         if self.password is not None:
             headers.append(("X-ClickHouse-Key", self.password))
         netloc = build_netloc(self.host, self.port)
-        params = {"wait_end_of_query": "1"}
+        params = {"wait_end_of_query": "1", "http_write_exception_in_output_format": "0"}
         if session_id is not None:
             params["session_id"] = session_id
         response = await httpx_request(

diff --git a/tests/integration/coordinator/plugins/clickhouse/test_plugin.py b/tests/integration/coordinator/plugins/clickhouse/test_plugin.py
@@ -625,4 +625,4 @@ async def test_restores_keeper_map_tables(restored_cluster: Sequence[ClickHouseC
         assert result == []
         with pytest.raises(ClickHouseClientQueryError) as e:
             await client.execute(b"SELECT * FROM default.keeper_map_dropped")
-        assert "Table default.keeper_map_dropped does not exist." in e.value.args[0]
+        assert "UNKNOWN_TABLE" in e.value.args[0]