From fb7c7aa73bd1d1e32ca1c4c1d16f3e89ae44ba72 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dar=C3=ADo=20Lodeiros?= <dario@commitsun.com>
Date: Fri, 29 Jan 2021 08:58:22 +0100
Subject: [PATCH] [FIX]pms: test checkin partner count only test reservations

---
 pms/tests/test_pms_checkin_partner.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/pms/tests/test_pms_checkin_partner.py b/pms/tests/test_pms_checkin_partner.py
index c91337724..c3c647f44 100644
--- a/pms/tests/test_pms_checkin_partner.py
+++ b/pms/tests/test_pms_checkin_partner.py
@@ -46,7 +46,6 @@ class TestPmsCheckinPartner(TestHotel):
         # ACTION
         self.arrange_single_checkin()
         checkins_count = len(self.reservation_1.checkin_partner_ids)
-
         # ASSERT
         self.assertEqual(
             checkins_count,
@@ -349,7 +348,7 @@ class TestPmsCheckinPartner(TestHotel):
         freezer.start()
         PmsReservation.auto_no_show()
 
-        no_show_reservations = PmsReservation.search([("state", "=", "no_show")])
+        no_show_reservations = self.folio_1.filtered(lambda r: r.state == "no_show")
 
         # ASSERT
         self.assertEqual(
@@ -371,8 +370,8 @@ class TestPmsCheckinPartner(TestHotel):
         freezer.start()
         PmsReservation.auto_no_checkout()
 
-        no_checkout_reservations = PmsReservation.search(
-            [("state", "=", "no_checkout")]
+        no_checkout_reservations = self.folio_1.filtered(
+            lambda r: r.state == "no_checkout"
         )
         freezer.stop()
         # ASSERT