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 test_datetime_to_nanoseconds(): dt = datetime.datetime(2016, 4, 16, 0, 23, 40, 157145, tzinfo=tz.tzutc()) expected = 1460766220000000000 assert datetime_to_nanoseconds(dt) == expected