update tests

panda-official · Dec 14, 2023 · 7992638 · 7992638
1 parent 9434944
commit 7992638
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,17 +7,21 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Fixed
+
+- Query timeout and improve error messages, [PR-18](https://github.com/panda-official/DriftCLI/pull/18)
+
 ## 0.9.0 - 2023-09-14
 
 ### Added
 
-- support for TypedData for CSV export, [PR-17](https://github.com/panda-official/DriftCLI/pull/17)
+- Support for TypedData for CSV export, [PR-17](https://github.com/panda-official/DriftCLI/pull/17)
 
 ## 0.8.0 - 2023-07-11
 
 ### Added
 
-- Supports for labels in metadata, [PR-15](https://github.com/panda-official/DriftCLI/pull/15)
+- Support for labels in metadata, [PR-15](https://github.com/panda-official/DriftCLI/pull/15)
 
 ## 0.7.1 - 2023-06-29
 

diff --git a/tests/export_test.py b/tests/export_test.py
@@ -203,6 +203,12 @@ def test__export_raw_data(runner, client, conf, export_path, topics, timeseries)
     assert (export_path / topics[1] / "1.dp").exists()
     assert (export_path / topics[1] / "2.dp").exists()
 
+    assert client.walk.call_count == 2
+    assert client.walk.call_args_list[0][0][0] == topics[0]
+    assert client.walk.call_args_list[0][1]["start"] == 1640991600.0
+    assert client.walk.call_args_list[0][1]["stop"] == 1641078000.0
+    assert client.walk.call_args_list[0][1]["ttl"] == 360
+
 
 @pytest.mark.usefixtures("set_alias")
 def test__export_raw_data_with_metadata(