def test_disabling_tproxy_support(self, get_json_via_rest_api_mock,
is_server_runing_mock, get_silent_mock,
get_ambari_properties_mock,
perform_changes_via_rest_api_mock):
out = StringIO.StringIO()
sys.stdout = out
get_json_via_rest_api_mock.return_value = (200, {})
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = False
properties = Properties()
get_ambari_properties_mock.return_value = properties
options = self._create_empty_options_mock()
options.tproxy_enabled = 'false'
setup_trusted_proxy(options)
self.assertTrue(perform_changes_via_rest_api_mock.called)
requestCall = perform_changes_via_rest_api_mock.call_args_list[0]
args, kwargs = requestCall
requestMethod = args[4]
self.assertTrue(isinstance(requestMethod, str))
self.assertEqual(requestMethod, "DELETE")
sys.stdout = sys.__stdout__
pass
def test_tproxy_setup_should_fail_if_server_is_not_running(
self, is_server_runing_mock):
out = StringIO.StringIO()
sys.stdout = out
is_server_runing_mock.return_value = (False, 0)
options = self._create_empty_options_mock()
try:
setup_trusted_proxy(options)
self.fail("Should fail with non-fatal exception")
except FatalException as e:
self.assertTrue("Ambari Server is not running" in e.reason)
pass
sys.stdout = sys.__stdout__
pass
def test_silent_mode_is_not_allowed(self, is_server_runing_mock,
get_silent_mock):
out = StringIO.StringIO()
sys.stdout = out
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = True
options = self._create_empty_options_mock()
try:
setup_trusted_proxy(options)
self.fail("Should fail with fatal exception")
except NonFatalException as e:
self.assertTrue(
"setup-trusted-proxy is not enabled in silent mode." in
e.reason)
pass
sys.stdout = sys.__stdout__
pass
def test_invalid_tproxy_enabled_cli_option_should_result_in_error(
self, is_server_runing_mock, get_silent_mock):
out = StringIO.StringIO()
sys.stdout = out
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = False
options = self._create_empty_options_mock()
options.tproxy_enabled = 'not_true_or_false'
try:
setup_trusted_proxy(options)
self.fail("Should fail with fatal exception")
except FatalException as e:
self.assertTrue(
"--tproxy-enabled should be to either 'true' or 'false'" in
e.reason)
pass
sys.stdout = sys.__stdout__
pass
def test_enable_tproxy_support_using_configuration_file_path_from_command_line_should_fail_if_file_does_not_exist(
self, isfile_mock, is_server_runing_mock, get_silent_mock):
out = StringIO.StringIO()
sys.stdout = out
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = False
isfile_mock.return_value = False
options = self._create_empty_options_mock()
options.tproxy_enabled = 'true'
options.tproxy_configuration_file_path = 'samplePath'
try:
setup_trusted_proxy(options)
self.fail("Should fail with fatal exception")
except FatalException as e:
self.assertTrue(
"--tproxy-configuration-file-path is set to a non-existing file"
in e.reason)
pass
sys.stdout = sys.__stdout__
pass
def test_enable_tproxy_support_using_configuration_file_path_from_command_line(
self, open_mock, isfile_mock, get_json_via_rest_api_mock,
is_server_runing_mock, get_silent_mock, get_ambari_properties_mock,
perform_changes_via_rest_api_mock):
out = StringIO.StringIO()
sys.stdout = out
get_json_via_rest_api_mock.return_value = (200, {})
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = False
properties = Properties()
get_ambari_properties_mock.return_value = properties
isfile_mock.return_value = True
tproxy_configurations = "["\
" {"\
" \"proxyuser\" : \"knox\"," \
" \"hosts\" : \"host1\"," \
" \"users\" : \"user1\"," \
" \"groups\" : \"group1\"" \
" }," \
" {"\
" \"proxyuser\": \"admin\"," \
" \"hosts\" : \"host2\"," \
" \"users\" : \"user2\"," \
" \"groups\" : \"group2\"" \
" }" \
"]"
mock_file = MagicFile(tproxy_configurations)
open_mock.side_effect = [mock_file]
options = self._create_empty_options_mock()
options.tproxy_enabled = 'true'
options.tproxy_configuration_file_path = 'samplePath'
setup_trusted_proxy(options)
self.assertTrue(perform_changes_via_rest_api_mock.called)
requestCall = perform_changes_via_rest_api_mock.call_args_list[0]
args, kwargs = requestCall
requestData = args[5]
self.assertTrue(isinstance(requestData, dict))
tproxyProperties = requestData['Configuration']['properties']
self.assertEqual(tproxyProperties[TPROXY_SUPPORT_ENABLED], 'true')
user_name1 = "knox"
self.assertEqual(tproxyProperties[PROXYUSER_HOSTS.format(user_name1)],
"host1")
self.assertEqual(tproxyProperties[PROXYUSER_USERS.format(user_name1)],
"user1")
self.assertEqual(tproxyProperties[PROXYUSER_GROUPS.format(user_name1)],
"group1")
user_name2 = "admin"
self.assertEqual(tproxyProperties[PROXYUSER_HOSTS.format(user_name2)],
"host2")
self.assertEqual(tproxyProperties[PROXYUSER_USERS.format(user_name2)],
"user2")
self.assertEqual(tproxyProperties[PROXYUSER_GROUPS.format(user_name2)],
"group2")
sys.stdout = sys.__stdout__
pass
def test_tproxy_is_enabled_for_two_proxy_users(
self, get_json_via_rest_api_mock, is_server_runing_mock,
get_silent_mock, get_ambari_properties_mock,
get_validated_string_input_mock, get_YN_input_mock,
perform_changes_via_rest_api_mock):
out = StringIO.StringIO()
sys.stdout = out
get_json_via_rest_api_mock.return_value = (200, {})
is_server_runing_mock.return_value = (True, 0)
get_silent_mock.return_value = False
get_ambari_properties_mock.return_value = Properties()
user_name1 = 'knox'
hosts1 = 'knox_hosts'
users1 = 'knox_users'
groups1 = 'knox_groups'
user_name2 = 'admin'
hosts2 = 'admin_hosts'
users2 = 'admin_users'
groups2 = 'admin_groups'
get_validated_string_input_mock.side_effect = [
user_name1, hosts1, users1, groups1, user_name2, hosts2, users2,
groups2
]
get_YN_input_mock.side_effect = [
True, False
] #answer 'True' for the first time when asking for a new proxy user addition and then 'False' (indicating we do not want to add more proxy users)
options = self._create_empty_options_mock()
options.tproxy_enabled = 'true'
setup_trusted_proxy(options)
self.assertTrue(perform_changes_via_rest_api_mock.called)
requestCall = perform_changes_via_rest_api_mock.call_args_list[0]
args, kwargs = requestCall
requestData = args[5]
self.assertTrue(isinstance(requestData, dict))
tproxyProperties = requestData['Configuration']['properties']
self.assertEqual(tproxyProperties[TPROXY_SUPPORT_ENABLED], 'true')
self.assertEqual(tproxyProperties[PROXYUSER_HOSTS.format(user_name1)],
hosts1)
self.assertEqual(tproxyProperties[PROXYUSER_USERS.format(user_name1)],
users1)
self.assertEqual(tproxyProperties[PROXYUSER_GROUPS.format(user_name1)],
groups1)
self.assertEqual(tproxyProperties[PROXYUSER_HOSTS.format(user_name2)],
hosts2)
self.assertEqual(tproxyProperties[PROXYUSER_USERS.format(user_name2)],
users2)
self.assertEqual(tproxyProperties[PROXYUSER_GROUPS.format(user_name2)],
groups2)
sys.stdout = sys.__stdout__
pass