add queue to test

guangrei · Sep 21, 2024 · 2fefa67 · 2fefa67
1 parent e0dd28a
commit 2fefa67
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from zcache import Cache, SmartRequest
+from zcache import Cache, SmartRequest, Queue
 import requests
 import time
 import unittest
@@ -46,6 +46,18 @@ def test_request(self):
                          cache_path='/tmp/request2.cache')
         self.assertEqual(r.response["body"], s.response["body"])
 
+    def test_queue(self):
+        q = Queue()
+        id = q.put("test")
+        self.assertEqual(q.exists(id), True)
+        self.assertEqual(q.peek(), "test")
+        self.assertEqual(q.size(), 1)
+        self.assertEqual(q.empty(), False)
+        self.assertEqual(q.get(), "test")
+        self.assertEqual(q.size(), 0)
+        self.assertEqual(q.empty(), True)
+        self.assertEqual(q.exists(id), False)
+
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/zcache/Extras/SmartRequest.py b/zcache/Extras/SmartRequest.py
@@ -33,7 +33,7 @@ class SmartRequest:
     A class for making Smart HTTP requests with caching capabilities using PyZCache.
     """
 
-    def __init__(self, url, cache_path=None, cache_time=120, offline_ttl=604800, **kwargs):
+    def __init__(self, url, cache_path="smartrequest.json", cache_time=120, offline_ttl=604800, **kwargs):
         if not isinstance(url, str):
             cache_name = url.url
         else: