Fix unit tests

gbtami · Dec 25, 2024 · 86ba024 · 86ba024
1 parent e41b643
commit 86ba024
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/server/auto_pair.py b/server/auto_pair.py
@@ -31,6 +31,7 @@ def add_to_auto_pairings(app_state, user, data):
             variant_tc[1][2],
         )
         variant, chess960, base, inc, byoyomi_period = variant_tc
+        print(variant_tc)
         # We don't want to create non byo variant with byo TC combinations
         if (byoyomi_period > 0 and variant not in BYOS) or variant.startswith("bughouse"):
             continue

diff --git a/tests/test_auto_pairing.py b/tests/test_auto_pairing.py
@@ -96,7 +96,7 @@ def test_add_to_auto_pairings(self):
             app_state, self.bplayer, DATA["chess"]
         )
 
-        self.assertEqual(auto_variant_tc, variant_tc)
+        self.assertIn(variant_tc, app_state.auto_pairings)
         self.assertEqual(app_state.auto_pairing_users[self.bplayer], (-10000, 10000))
         self.assertIn(self.bplayer, app_state.auto_pairings[variant_tc])
         self.assertTrue(self.bplayer.ready_for_auto_pairing)
@@ -106,7 +106,7 @@ def test_add_to_auto_pairings(self):
             app_state, self.aplayer, DATA["chess960"]
         )
 
-        self.assertEqual(auto_variant_tc, variant_tc)
+        self.assertIn(variant_tc, app_state.auto_pairings)
         self.assertIsNone(matching_user)
         self.assertIsNone(matching_seek)
         self.assertTrue(self.aplayer.ready_for_auto_pairing)