snapshot of https://github.com/ActivityWatch/aw-core.git (master, 2026-09-16; blobs > 131072B omitted)

tests/utils.py

@@ -0,0 +1,52 @@
+import logging
+import random
+from datetime import datetime, timezone
+
+from aw_datastore import Datastore, get_storage_methods
+
+logging.basicConfig(level=logging.DEBUG)
+
+# Useful when you just want some placeholder time in your events, saves typing
+now = datetime.now(timezone.utc)
+
+
+class TempTestBucket:
+    """Context manager for creating a test bucket"""
+
+    def __init__(self, datastore):
+        self.ds = datastore
+        self.bucket_id = f"test-{random.randint(0, 10**4)}"
+
+    def __enter__(self):
+        self.ds.create_bucket(
+            bucket_id=self.bucket_id,
+            type="testtype",
+            client="testclient",
+            hostname="testhost",
+            name="testname",
+        )
+        return self.ds[self.bucket_id]
+
+    def __exit__(self, *_):
+        self.ds.delete_bucket(bucket_id=self.bucket_id)
+
+    def __repr__(self):
+        return f"<TempTestBucket using {self.ds.storage_strategy.__class__.__name__}>"
+
+
+_storage_methods = get_storage_methods()
+
+
+def param_datastore_objects():
+    return [
+        Datastore(storage_strategy=strategy, testing=True)
+        for name, strategy in _storage_methods.items()
+    ]
+
+
+def param_testing_buckets_cm():
+    datastores = [
+        Datastore(storage_strategy=strategy, testing=True)
+        for name, strategy in _storage_methods.items()
+    ]
+    return [TempTestBucket(ds) for ds in datastores]