From bb1bdb147807ff17e7a4174b9e20df632cb01754 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Sat, 27 Feb 2021 10:07:02 +0100
Subject: [PATCH] Improve test-coverage of pairlocks

---
 tests/plugins/test_pairlocks.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/plugins/test_pairlocks.py b/tests/plugins/test_pairlocks.py
index dfcbff0ed..fce3a8cd1 100644
--- a/tests/plugins/test_pairlocks.py
+++ b/tests/plugins/test_pairlocks.py
@@ -73,9 +73,13 @@ def test_PairLocks(use_db):
     assert PairLocks.is_pair_locked('XRP/USDT', lock_time + timedelta(minutes=-50))
 
     if use_db:
-        assert len(PairLock.query.all()) > 0
+        locks = PairLocks.get_all_locks()
+        locks_db = PairLock.query.all()
+        assert len(locks) == len(locks_db)
+        assert len(locks_db) > 0
     else:
         # Nothing was pushed to the database
+        assert len(PairLocks.get_all_locks()) > 0
         assert len(PairLock.query.all()) == 0
     # Reset use-db variable
     PairLocks.reset_locks()