def test_load_credentials_keyerror(mock_open, ): credentials = {} mock_open.side_effect = mock.mock_open(read_data=json.dumps(credentials)) with pytest.raises(KeyError): assert load_credentials()
def test_load_credentials_keyerror(mock_open,): credentials = {} mock_open.side_effect = mock.mock_open(read_data=json.dumps(credentials)) with pytest.raises(KeyError): assert load_credentials()
def when_setup_service_initiated(context): config = { 'master': '%s' % get_service_connection_string('mesosmaster'), 'scheme': 'http', 'response_timeout': 5, } with contextlib.nested( mock.patch( 'paasta_tools.bounce_lib.get_happy_tasks', autospec=True, # Wrap function call so we can select a subset of tasks or test # intermediate steps, like when an app is not completely up side_effect=lambda app, _, __, ___, **kwargs: get_happy_tasks( app, context.service, "fake_nerve_ns", context.system_paasta_config)[:context.max_tasks], ), mock.patch('paasta_tools.bounce_lib.bounce_lock_zookeeper', autospec=True), mock.patch('paasta_tools.bounce_lib.create_app_lock', autospec=True), mock.patch('paasta_tools.bounce_lib.time.sleep', autospec=True), mock.patch('paasta_tools.setup_marathon_job.load_system_paasta_config', autospec=True), mock.patch('paasta_tools.setup_marathon_job._log', autospec=True), mock.patch('paasta_tools.marathon_tools.get_config_hash', autospec=True, return_value='confighash'), mock.patch('paasta_tools.marathon_tools.get_code_sha_from_dockerurl', autospec=True, return_value='newapp'), mock.patch('paasta_tools.marathon_tools.get_docker_url', autospec=True, return_value='busybox'), mock.patch('paasta_tools.paasta_maintenance.load_credentials', autospec=True), mock.patch.object(mesos.cli.master, 'CFG', config), ) as ( _, _, _, _, mock_load_system_paasta_config, _, _, _, _, mock_load_credentials, _, ): mock_load_credentials.side_effect = paasta_maintenance.load_credentials(mesos_secrets='/etc/mesos-slave-secret') mock_load_system_paasta_config.return_value.get_cluster = mock.Mock(return_value=context.cluster) # 120 * 0.5 = 60 seconds for _ in xrange(120): try: (code, message) = setup_marathon_job.setup_service( service=context.service, instance=context.instance, client=context.marathon_client, service_marathon_config=context.new_marathon_service_config, soa_dir='/nail/etc/services', ) assert code == 0, message return except MarathonHttpError: time.sleep(0.5) raise Exception("Unable to acquire app lock for setup_marathon_job.setup_service")
def test_load_credentials( mock_open, ): credentials = { 'principal': 'username', 'secret': 'password' } mock_open.side_effect = mock.mock_open(read_data=json.dumps(credentials)) assert load_credentials() == ('username', 'password')
def mark_host_at_risk(context, host): start = paasta_maintenance.datetime_to_nanoseconds(paasta_maintenance.now()) duration = paasta_maintenance.parse_timedelta('1h') config = { 'master': '%s' % get_service_connection_string('mesosmaster'), 'scheme': 'http', 'response_timeout': 5, } with contextlib.nested( mock.patch('paasta_tools.paasta_maintenance.load_credentials', autospec=True), mock.patch.object(mesos.cli.master, 'CFG', config), ) as ( mock_load_credentials, _, ): mock_load_credentials.side_effect = paasta_maintenance.load_credentials(mesos_secrets='/etc/mesos-slave-secret') paasta_maintenance.drain([host], start, duration) context.at_risk_host = host
def run_until_number_tasks(context, number): config = { 'master': '%s' % get_service_connection_string('mesosmaster'), 'scheme': 'http', 'response_timeout': 5, } for _ in xrange(20): with contextlib.nested( mock.patch('paasta_tools.paasta_maintenance.load_credentials', autospec=True), mock.patch.object(mesos.cli.master, 'CFG', config), ) as ( mock_load_credentials, _, ): mock_load_credentials.side_effect = paasta_maintenance.load_credentials( mesos_secrets='/etc/mesos-slave-secret', ) run_setup_marathon_job(context) sleep(0.5) if context.marathon_client.get_app(context.app_id).instances == number: return assert context.marathon_client.get_app(context.app_id).instances == number
def test_load_credentials_missing_file(mock_open,): with pytest.raises(IOError): assert load_credentials()
def test_load_credentials(mock_open,): credentials = {"principal": "username", "secret": "password"} mock_open.side_effect = mock.mock_open(read_data=json.dumps(credentials)) assert load_credentials() == ("username", "password")
def test_load_credentials_missing_file(mock_open, ): with pytest.raises(IOError): assert load_credentials()
def test_load_credentials(mock_open, ): credentials = {'principal': 'username', 'secret': 'password'} mock_open.side_effect = mock.mock_open(read_data=json.dumps(credentials)) assert load_credentials() == ('username', 'password')