From 04dfb185ce0bc43364361aa369b9191e61b8064c Mon Sep 17 00:00:00 2001
From: lash <dev@holbrook.no>
Date: Sun, 13 Mar 2022 15:45:48 +0000
Subject: [PATCH] Implement upcoming query on store

---
 chainqueue/store/base.py | 5 +++++
 tests/test_integrate.py  | 9 ++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/chainqueue/store/base.py b/chainqueue/store/base.py
index 93ffdec..1275d8f 100644
--- a/chainqueue/store/base.py
+++ b/chainqueue/store/base.py
@@ -95,3 +95,8 @@ class Store:
             entry.retry()
         except StateTransitionInvalid:
             entry.readysend()
+
+
+    def fail(self, k):
+        entry = QueueEntry(self, k)
+        entry.reject()
diff --git a/tests/test_integrate.py b/tests/test_integrate.py
index ccb8101..d8243c7 100644
--- a/tests/test_integrate.py
+++ b/tests/test_integrate.py
@@ -43,7 +43,7 @@ class TestIntegrateBase(TestShepBase):
         self.store.put(b'foo'.hex(), b'bar'.hex(), cache_adapter=MockCacheTokenTx)
 
 
-    def test_state_move(self):
+    def test_state_enqueu(self):
         hx = b'foo'.hex()
         self.store.put(hx, b'bar'.hex(), cache_adapter=MockCacheTokenTx)
         self.store.get(hx)
@@ -53,5 +53,12 @@ class TestIntegrateBase(TestShepBase):
         self.assertEqual(v[0], hx)
 
 
+    def test_state_defer(self):
+        hx = b'foo'.hex()
+        self.store.put(hx, b'bar'.hex(), cache_adapter=MockCacheTokenTx)
+        self.store.get(hx)
+        self.store.fail(hx)
+
+
 if __name__ == '__main__':
     unittest.main()