def test_main_deploy_secondary(mocker): parent_mock = mocker.Mock() mocker.patch("nuke_from_orbit.utils.nuke_utils.set_variables").return_value = MOCK_USER_CONFIG mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_gke") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_test_container_image") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_persistent_disk") mocker.patch("nuke_from_orbit.utils.nuke_utils.get_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.collect_kube_yaml_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.render_kubernetes_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_looker_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_oauth_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_external") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_locust") context_mock = mocker.patch("nuke_from_orbit.utils.nuke_utils.set_kubernetes_context") deploy_mock = mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_secondary") parent_mock.attach_mock(context_mock, "context_mock") parent_mock.attach_mock(deploy_mock, "deploy_mock") setup_commands.main(config_file="mock_config.yaml", external=True, persistence=False) # determine if context call occurs before deployment expected_call_order = [mocker.call.context_mock(MOCK_USER_CONFIG), mocker.call.deploy_mock()] assert parent_mock.mock_calls == expected_call_order
def test_main_set_k8s_context(mocker): mocker.patch("nuke_from_orbit.utils.nuke_utils.set_variables").return_value = MOCK_USER_CONFIG mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_gke") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_test_container_image") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_persistent_disk") mocker.patch("nuke_from_orbit.utils.nuke_utils.get_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.collect_kube_yaml_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.render_kubernetes_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.set_kubernetes_context") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_looker_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_oauth_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_external") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_locust") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_secondary") setup_commands.main(config_file="mock_config.yaml", external=True, persistence=False) nuke_utils.set_kubernetes_context.assert_called_with(MOCK_USER_CONFIG)
def test_main_run_threads_no_persistence_no_external(mocker): mocker.patch("nuke_from_orbit.utils.nuke_utils.set_variables").return_value = MOCK_USER_CONFIG mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_gke") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_test_container_image") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_persistent_disk") mocker.patch("nuke_from_orbit.utils.nuke_utils.get_ip_address") mocker.patch("nuke_from_orbit.utils.nuke_utils.collect_kube_yaml_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.render_kubernetes_templates") mocker.patch("nuke_from_orbit.utils.nuke_utils.set_kubernetes_context") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_looker_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_oauth_secret") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_external") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_locust") mocker.patch("nuke_from_orbit.utils.nuke_utils.deploy_secondary") setup_commands.main(config_file="mock_config.yaml", external=False, persistence=False) nuke_utils.deploy_gke.assert_called_with(MOCK_USER_CONFIG) nuke_utils.deploy_test_container_image.assert_called_with(MOCK_USER_CONFIG) # deploy ip and persistent disk shouldn't be called in multithread nuke_utils.deploy_ip_address.assert_not_called() nuke_utils.deploy_persistent_disk.assert_not_called()
def setup(**kwargs): setup_commands.main(**kwargs)