RDBC-697 Refurbished legacy time series tests

Author: poissoncorp

ravendb/tests/jvm_migrated_tests/client_tests/time_series_tests/test_time_series_operations.py

@@ -3,7 +3,8 @@
 from ravendb.documents.operations.time_series import (
     GetTimeSeriesOperation,
     TimeSeriesOperation,
-    TimeSeriesBatchOperation, GetTimeSeriesStatisticsOperation,
+    TimeSeriesBatchOperation,
+    GetTimeSeriesStatisticsOperation,
 )
 from ravendb.tests.test_base import TestBase, User
 

ravendb/tests/operations_tests/test_timeseries_operaiotn.py

@@ -1,87 +1,87 @@
-from ravendb.tests.test_base import *
+from datetime import datetime, timedelta
+
+from ravendb.documents.operations.time_series import (
+    GetTimeSeriesOperation,
+    TimeSeriesOperation,
+    TimeSeriesBatchOperation,
+)
+
 import unittest
 
+from ravendb.tests.test_base import TestBase, User
+
 
 class TestTimeSeriesOperations(TestBase):
     def setUp(self):
         super(TestTimeSeriesOperations, self).setUp()
-        self.timeseries_range = TimeSeriesRange("Heartrate")
+        self.ts_name = "Heartrate"
         with self.store.open_session() as session:
-            user = User(name="Idan", age="31")
-            session.store(user, key="users/1-A")
+            session.store(User())
             session.save_changes()
 
     def tearDown(self):
         super(TestTimeSeriesOperations, self).tearDown()
-        TestBase.delete_all_topology_files()
 
-    def add_timeseries(self):
-        # Add time_series to user/1-A document
-        time_series_operation = TimeSeriesOperation(name="Heartrate")
-        time_series_operation.append_single(datetime.now(), 73, tag="heart/rates")
-        time_series_operation.append_single(datetime.now() + timedelta(minutes=5), 78, tag="heart/rates")
-        time_series_operation.append_single(datetime(2019, 4, 23) + timedelta(minutes=5), 789, tag="heart/rates")
+    def add_time_series(self):
+        ts_operation = TimeSeriesOperation(self.ts_name)
+
+        base = datetime.now()
+        base_24 = datetime(2023, 11, 14)
+
+        ts_operation.append(TimeSeriesOperation.AppendOperation(base, [73], "heart/rates"))
+        ts_operation.append(TimeSeriesOperation.AppendOperation(base + timedelta(minutes=5), [78], "heart/rates"))
+        ts_operation.append(TimeSeriesOperation.AppendOperation(base_24 + timedelta(minutes=5), [789], "heart/rates"))
 
-        time_series_batch_operation = TimeSeriesBatchOperation(document_id="users/1-A", operation=time_series_operation)
+        time_series_batch_operation = TimeSeriesBatchOperation("users/1-A", ts_operation)
         self.store.operations.send(time_series_batch_operation)
 
-    @unittest.skip("Time series")
-    def test_append_new_timeseries(self):
-        self.add_timeseries()
+    def test_append_new_time_series(self):
+        self.add_time_series()
+
         # Fetch all time_series from the document
-        time_series = self.store.operations.send(GetTimeSeriesOperation("users/1-A", ranges=self.timeseries_range))
-        entries = time_series["Entries"]
-        self.assertEqual(len(entries), 3)
-
-    @unittest.skip("Time series")
-    def test_remove_timeseries_with_range(self):
-        self.add_timeseries()
-        time_series_operation_remove = TimeSeriesOperation(name="Heartrate")
-        time_series_operation_remove.remove(
-            from_date=datetime.now() - timedelta(days=2),
-            to_date=datetime.now() + timedelta(days=2),
-        )
-
-        time_series_batch_operation = TimeSeriesBatchOperation(
-            document_id="users/1-A", operation=time_series_operation_remove
-        )
+        time_series_range_result = self.store.operations.send(GetTimeSeriesOperation("users/1-A", self.ts_name))
+        self.assertEqual(len(time_series_range_result.entries), 3)
+
+    def test_remove_time_series_with_range(self):
+        self.add_time_series()
+
+        base = datetime.now()
+        ts_operation = TimeSeriesOperation(self.ts_name)
+        ts_operation.delete(TimeSeriesOperation.DeleteOperation(base - timedelta(days=2), base + timedelta(days=2)))
+
+        time_series_batch_operation = TimeSeriesBatchOperation("users/1-A", ts_operation)
         self.store.operations.send(time_series_batch_operation)
-        time_series = self.store.operations.send(GetTimeSeriesOperation("users/1-A", ranges=self.timeseries_range))
-        entries = time_series["Entries"]
-        self.assertEqual(len(entries), 1)
-
-    @unittest.skip("Time series")
-    def test_remove_timeseries_without_range(self):
-        self.add_timeseries()
-        time_series_operation_remove = TimeSeriesOperation(name="Heartrate")
-        time_series_operation_remove.remove()
-
-        time_series_batch_operation = TimeSeriesBatchOperation(
-            document_id="users/1-A", operation=time_series_operation_remove
-        )
+
+        time_series_range_result = self.store.operations.send(GetTimeSeriesOperation("users/1-A", self.ts_name))
+        self.assertEqual(len(time_series_range_result.entries), 1)
+
+    def test_remove_time_series_without_range(self):
+        self.add_time_series()
+
+        time_series_operation_remove = TimeSeriesOperation(self.ts_name)
+        time_series_operation_remove.delete(TimeSeriesOperation.DeleteOperation())
+
+        time_series_batch_operation = TimeSeriesBatchOperation("users/1-A", time_series_operation_remove)
         self.store.operations.send(time_series_batch_operation)
-        time_series = self.store.operations.send(GetTimeSeriesOperation("users/1-A", ranges=self.timeseries_range))
-        self.assertIsNone(time_series)
-
-    @unittest.skip("TimeSeries")
-    def test_get_timeseries_with_range(self):
-        self.add_timeseries()
-        time_series_range = TimeSeriesRange(
-            "Heartrate",
-            from_date=datetime.now() - timedelta(days=2),
-            to_date=datetime.now() + timedelta(days=2),
-        )
-        time_series = self.store.operations.send(GetTimeSeriesOperation("users/1-A", ranges=time_series_range))
-        entries = time_series["Entries"]
-        self.assertEqual(len(entries), 2)
-
-    @unittest.skip("Time series")
-    def test_get_timeseries_without_range(self):
-        self.add_timeseries()
-        time_series_range = TimeSeriesRange("Heartrate")
-        time_series = self.store.operations.send(GetTimeSeriesOperation("users/1-A", ranges=time_series_range))
-        entries = time_series["Entries"]
-        self.assertEqual(len(entries), 3)
+
+        time_series_range_result = self.store.operations.send(GetTimeSeriesOperation("users/1-A", self.ts_name))
+        self.assertIsNone(time_series_range_result)
+
+    def test_get_time_series_with_range(self):
+        self.add_time_series()
+
+        base = datetime.now()
+        from_base = base - timedelta(days=2)
+        to_base = base + timedelta(days=2)
+
+        ts_result = self.store.operations.send(GetTimeSeriesOperation("users/1-A", self.ts_name, from_base, to_base))
+        self.assertEqual(len(ts_result.entries), 2)
+
+    def test_get_time_series_without_range(self):
+        self.add_time_series()
+
+        ts_range_result = self.store.operations.send(GetTimeSeriesOperation("users/1-A", self.ts_name))
+        self.assertEqual(len(ts_range_result.entries), 3)
 
 
 if __name__ == "__main__":