chainqueue

commit 485b33866b11d5d93e39eed1d91249a5c5b3d088
parent 04dfb185ce0bc43364361aa369b9191e61b8064c
Author: lash <dev@holbrook.no>
Date:   Sun, 13 Mar 2022 16:04:14 +0000

Harden query tests

Diffstat:
M
chainqueue/store/base.py
 | 
3
++-
M
tests/test_integrate.py
 | 
55
+++++++++++++++++++++++++++++++++++++++++++++++--------

2 files changed, 49 insertions(+), 9 deletions(-)
diff --git a/chainqueue/store/base.py b/chainqueue/store/base.py
@@ -99,4 +99,5 @@ class Store:
 
     def fail(self, k):
         entry = QueueEntry(self, k)
-        entry.reject()
+        entry.load()
+        entry.sendfail()
diff --git a/tests/test_integrate.py b/tests/test_integrate.py
@@ -1,4 +1,5 @@
 # standard imports
+import os
 import tempfile
 import unittest
 import logging
@@ -40,24 +41,62 @@ class TestIntegrateBase(TestShepBase):
 
 
     def test_integration_valid(self):
-        self.store.put(b'foo'.hex(), b'bar'.hex(), cache_adapter=MockCacheTokenTx)
+        self.store.put(os.urandom(4).hex(), os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
 
+    
+    def test_state_default(self):
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        v = self.store.pending()
+        self.assertEqual(len(v), 1)
+        self.assertEqual(v[0], hx)
 
-    def test_state_enqueu(self):
-        hx = b'foo'.hex()
-        self.store.put(hx, b'bar'.hex(), cache_adapter=MockCacheTokenTx)
+
+    def test_state_enqueue(self):
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
         self.store.get(hx)
         self.store.enqueue(hx)
         v = self.store.upcoming()
         self.assertEqual(len(v), 1)
-        self.assertEqual(v[0], hx)
+        v = self.store.pending()
+        self.assertEqual(len(v), 0)
 
 
     def test_state_defer(self):
-        hx = b'foo'.hex()
-        self.store.put(hx, b'bar'.hex(), cache_adapter=MockCacheTokenTx)
-        self.store.get(hx)
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        self.store.fail(hx)
+        v = self.store.deferred()
+        self.assertEqual(len(v), 1)
+        self.assertEqual(v[0], hx)
+
+
+    def test_state_multiple(self):
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        self.store.fail(hx)
+        hx = os.urandom(8).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        self.store.fail(hx)
+        v = self.store.deferred()
+        self.assertEqual(len(v), 2)
+
+
+    def test_state_multiple_sort(self):
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        self.store.fail(hx)
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        self.store.enqueue(hx)
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
         self.store.fail(hx)
+        hx = os.urandom(4).hex()
+        self.store.put(hx, os.urandom(8).hex(), cache_adapter=MockCacheTokenTx)
+        v = self.store.deferred()
+        self.assertEqual(len(v), 2)
 
 
 if __name__ == '__main__':