added test for yaml read

getredash · Nov 21, 2023 · 00a441b · 00a441b
1 parent 2f9eb7f
commit 00a441b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/query_runner/test_yandex_disk.py b/tests/query_runner/test_yandex_disk.py
@@ -144,6 +144,14 @@ def test_run_query_with_empty_query(mock_yandex_disk):
     assert result == (None, "Query is empty")
 
 
+@skip_condition
+def test_run_query_nonstring_yaml(mock_yandex_disk):
+    bad_yaml_query = None
+    data, error = mock_yandex_disk.run_query(bad_yaml_query, "user")
+    assert data is None
+    assert error is not None
+
+
 @skip_condition
 def test_run_query_bad_yaml(mock_yandex_disk):
     bad_yaml_query = "unparseable = yaml"