Allow to ask for query count

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/Network/KRPCSpec.hs b/tests/Network/KRPCSpec.hs
index 756c6855..e695a646 100644
--- a/tests/Network/KRPCSpec.hs
+++ b/tests/Network/KRPCSpec.hs
@@ -33,6 +33,15 @@ spec = do
          query servAddr (Echo int))
        `shouldReturn` Echo int
 
+    it "count transactions properly" $ do
+      (withManager opts servAddr handlers $ runReaderT $ do
+         listen
+         _ <- query servAddr (Echo (0xabcd :: Int))
+         _ <- query servAddr (Echo (0xabcd :: Int))
+         getQueryCount
+       )
+        `shouldReturn` 2
+
     it "throw timeout exception" $ do
       (withManager opts servAddr handlers $ runReaderT $ do
          query servAddr (Echo (0xabcd :: Int))