From 967fcbf9f6db6b30787729726e8b7e657e40ad18 Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Wed, 30 Aug 2023 16:14:22 +0200
Subject: [PATCH] Fix tests

---
 tests/test_elements_worker/test_elements.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index 1d7b7a28..273e41ba 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -1848,8 +1848,8 @@ def test_list_element_children_with_cache(
     filters,
     expected_ids,
 ):
-    # Check we have 2 elements already present in database
-    assert CachedElement.select().count() == 3
+    # Check we have 5 elements already present in database
+    assert CachedElement.select().count() == 5
 
     # Query database through cache
     elements = mock_elements_worker_with_cache.list_element_children(**filters)
@@ -2296,6 +2296,9 @@ def test_list_element_parents_with_cache(
     filters,
     expected_id,
 ):
+    # Check we have 5 elements already present in database
+    assert CachedElement.select().count() == 5
+
     # Query database through cache
     elements = mock_elements_worker_with_cache.list_element_parents(**filters)
     assert elements.count() == 1
-- 
GitLab