Skip to content

Commit

Permalink
fix columns names in InvalidResultColumnNames being reported the wron…
Browse files Browse the repository at this point in the history
…g way round (#185)

fix columns names in InvalidResultColumnNames being reported the wrong way round
---------

Co-authored-by: Alberto Donato <alberto.donato@gmail.com>
  • Loading branch information
andrewjw and albertodonato committed Jan 25, 2024
1 parent 0f39657 commit 1bc8aa0
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 4 deletions.
2 changes: 1 addition & 1 deletion query_exporter/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,7 +218,7 @@ def results(self, query_results: QueryResults) -> MetricResults:
if len(expected_keys) != len(result_keys):
raise InvalidResultCount(len(expected_keys), len(result_keys))
if result_keys != expected_keys:
raise InvalidResultColumnNames(result_keys, expected_keys)
raise InvalidResultColumnNames(expected_keys, result_keys)
results = []
for row in query_results.rows:
values = dict(zip(query_results.keys, row))
Expand Down
10 changes: 7 additions & 3 deletions tests/db_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -292,8 +292,12 @@ def test_results_wrong_names_with_labels(self):
"query", ["db"], [QueryMetric("metric1", ["label1"])], ""
)
query_results = QueryResults(["one", "two"], [(1, 2)])
with pytest.raises(InvalidResultColumnNames):
with pytest.raises(InvalidResultColumnNames) as error:
query.results(query_results)
assert str(error.value) == (
"Wrong column names from query: "
"expected (label1, metric1), got (one, two)"
)


class TestQueryResults:
Expand Down Expand Up @@ -609,7 +613,7 @@ async def test_execute_query_invalid_names_with_labels(self, db):
await db.execute(query)
assert (
str(error.value)
== "Wrong column names from query: expected (foo, label), got (label, metric)"
== "Wrong column names from query: expected (label, metric), got (foo, label)"
)
assert error.value.fatal

Expand All @@ -618,7 +622,7 @@ async def test_execute_query_traceback_debug(self, caplog, mocker, db):
"""Traceback are logged as debug messages."""
query = Query(
"query",
20,
["db"],
[QueryMetric("metric", [])],
"SELECT 1 AS metric",
)
Expand Down

0 comments on commit 1bc8aa0

Please sign in to comment.