Correct tests

Author: Remy Gwaramadze

tests/test_quixstreams/test_state/test_rocksdb/test_windowed/test_state.py

@@ -71,8 +71,8 @@ def test_expire_windows(transaction_state, delete):
 
     assert len(expired) == 2
     assert expired == [
-        ((0, 10), 1, [], b"__key__"),
-        ((10, 20), 2, [], b"__key__"),
+        ((0, 10), 1, []),
+        ((10, 20), 2, []),
     ]
 
     with transaction_state() as state:
@@ -112,8 +112,8 @@ def test_expire_windows_with_collect(transaction_state, end_inclusive):
     window_1_value = ["a", "b"] if end_inclusive else ["a"]
     window_2_value = ["b", "c"] if end_inclusive else ["b"]
     assert expired == [
-        ((0, 10), None, window_1_value, b"__key__"),
-        ((10, 20), [777, None], window_2_value, b"__key__"),
+        ((0, 10), None, window_1_value),
+        ((10, 20), [777, None], window_2_value),
     ]
 
 
@@ -132,7 +132,7 @@ def test_same_keys_in_db_and_update_cache(transaction_state):
         expired = list(state.expire_windows(max_start_time=max_start_time))
 
         # Value from the cache takes precedence over the value in the db
-        assert expired == [((0, 10), 3, [], b"__key__")]
+        assert expired == [((0, 10), 3, [])]
 
 
 def test_get_latest_timestamp(windowed_rocksdb_store_factory):

tests/test_quixstreams/test_state/test_rocksdb/test_windowed/test_transaction.py

@@ -78,8 +78,8 @@ def test_expire_windows_expired(self, windowed_rocksdb_store_factory, delete):
 
         assert len(expired) == 2
         assert expired == [
-            ((0, 10), 1, [], prefix),
-            ((10, 20), 2, [], prefix),
+            ((0, 10), 1, []),
+            ((10, 20), 2, []),
         ]
 
         with store.start_partition_transaction(0) as tx:
@@ -131,8 +131,8 @@ def test_expire_windows_cached(self, windowed_rocksdb_store_factory, delete):
             )
             assert len(expired) == 2
             assert expired == [
-                ((0, 10), 1, [], prefix),
-                ((10, 20), 2, [], prefix),
+                ((0, 10), 1, []),
+                ((10, 20), 2, []),
             ]
             assert (
                 tx.get_window(start_ms=0, end_ms=10, prefix=prefix) == None
@@ -193,7 +193,7 @@ def test_expire_windows_with_grace_expired(self, windowed_rocksdb_store_factory)
             )
 
         assert len(expired) == 1
-        assert expired == [((0, 10), 1, [], prefix)]
+        assert expired == [((0, 10), 1, [])]
 
     def test_expire_windows_with_grace_empty(self, windowed_rocksdb_store_factory):
         store = windowed_rocksdb_store_factory()
@@ -328,9 +328,9 @@ def test_expire_windows_multiple_windows(self, windowed_rocksdb_store_factory):
             )
 
         assert len(expired) == 3
-        assert expired[0] == ((0, 10), 1, [], prefix)
-        assert expired[1] == ((10, 20), 1, [], prefix)
-        assert expired[2] == ((20, 30), 1, [], prefix)
+        assert expired[0] == ((0, 10), 1, [])
+        assert expired[1] == ((10, 20), 1, [])
+        assert expired[2] == ((20, 30), 1, [])
 
     def test_get_latest_timestamp_update(self, windowed_rocksdb_store_factory):
         store = windowed_rocksdb_store_factory()