def mark_host_at_risk(context, host): start = mesos_maintenance.datetime_to_nanoseconds(mesos_maintenance.now()) duration = mesos_maintenance.parse_timedelta("1h") with contextlib.nested( mock.patch("paasta_tools.mesos_maintenance.get_principal", autospec=True), mock.patch("paasta_tools.mesos_maintenance.get_secret", autospec=True), ) as (mock_get_principal, mock_get_secret): credentials = mesos_maintenance.load_credentials(mesos_secrets="/etc/mesos-slave-secret") mock_get_principal.return_value = credentials.principal mock_get_secret.return_value = credentials.secret mesos_maintenance.drain([host], start, duration) context.at_risk_host = host
def mark_host_at_risk(context, host): start = mesos_maintenance.datetime_to_nanoseconds(mesos_maintenance.now()) duration = mesos_maintenance.parse_timedelta('1h') with mock.patch( 'paasta_tools.mesos_maintenance.get_principal', autospec=True, ) as mock_get_principal, mock.patch( 'paasta_tools.mesos_maintenance.get_secret', autospec=True, ) as mock_get_secret: credentials = mesos_maintenance.load_credentials(mesos_secrets='/etc/mesos-slave-secret') mock_get_principal.return_value = credentials.principal mock_get_secret.return_value = credentials.secret mesos_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