def test_username_password_success(self, private_bus, environ_get): """Test the RegisterWithUsernamePasswordTask - success.""" # register server proxy register_server_proxy = Mock() # private register proxy get_proxy = private_bus.return_value.__enter__.return_value.get_proxy private_register_proxy = get_proxy.return_value # instantiate the task and run it task = RegisterWithUsernamePasswordTask( rhsm_register_server_proxy=register_server_proxy, username="******", password="******") task.run() # check the private register proxy Register method was called correctly private_register_proxy.Register.assert_called_once_with( "", "foo_user", "bar_password", {}, {}, "en_US.UTF-8")
def test_username_password_failure(self, private_bus, environ_get): """Test the RegisterWithUsernamePasswordTask - failure.""" # register server proxy register_server_proxy = Mock() # private register proxy get_proxy = private_bus.return_value.__enter__.return_value.get_proxy private_register_proxy = get_proxy.return_value # raise DBusError with error message in JSON json_error = '{"message": "Registration failed."}' private_register_proxy.Register.side_effect = DBusError(json_error) # instantiate the task and run it task = RegisterWithUsernamePasswordTask( rhsm_register_server_proxy=register_server_proxy, username="******", password="******") with self.assertRaises(RegistrationError): task.run() # check private register proxy Register method was called correctly private_register_proxy.Register.assert_called_with( "", "foo_user", "bar_password", {}, {}, "en_US.UTF-8")