Fix unit test

gonicus · Jan 5, 2024 · c5fe1ef · c5fe1ef
1 parent ba01f05
commit c5fe1ef
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/test/test_client_wrapper.py b/test/test_client_wrapper.py
@@ -123,22 +123,17 @@ def setUp(self) -> None:
     def test_login(self):
         self.__ami_client.login_response = FutureResponseMock(
             ResponseMock("Error", True))
-        self.assertRaises(Exception, self.__client.login, "", "", 1)
+        self.assertRaises(Exception, self.__client.login, "", "", 1, 1)
 
         self.__ami_client.login_response = FutureResponseMock(
             ResponseMock("Error", False))
-        self.assertRaisesRegex(
-            Exception,
-            "Unable to login AMI client",
-            self.__client.login,
-            "",
-            "",
-            0.01)
+        self.assertRaisesRegex(Exception, "Unable to login AMI client", self.__client.login, "", "", 0, 0)
 
         self.__ami_client.login_response = FutureResponseMock(
             ResponseMock("Success", False))
         self.__client._ClientWrapper__is_login_validated = True
-        self.__client.login("<username>", "<secret>", 0)
+        self.__client._ClientWrapper__is_asterisk_fully_booted = True
+        self.__client.login("<username>", "<secret>", 0, 0)
 
         self.assertEqual(self.__ami_client.logged_in_username, "<username>")
         self.assertEqual(self.__ami_client.logged_in_secret, "<secret>")