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")
