def setUp(self): self.context = mock.Mock() self.pool = 'default_pool' self.executor = mock.Mock() self.task = zone.ZoneAction(self.executor, self.context, self.pool, mock.Mock(), 'CREATE') self.task._wait_for_nameservers = mock.Mock()
def test_wait_for_nameservers(self, mock_time): # It is just a time.sleep :( task = zone.ZoneAction( self.executor, self.context, self.pool, mock.Mock(), 'CREATE' ) task._wait_for_nameservers() mock_time.sleep.assert_called_with(task.delay)
def test_call_on_delete(self): myzone = mock.Mock() task = zone.ZoneAction(self.executor, self.context, self.pool, myzone, 'DELETE') task._zone_action_on_targets = mock.Mock(return_value=True) task._poll_for_zone = mock.Mock(return_value=True) task._wait_for_nameservers = mock.Mock() assert task() assert myzone.serial == 0
def _do_zone_action(self, context, zone): pool = self.get_pool(zone.pool_id) all_tasks = [] all_tasks.append( zonetasks.ZoneAction(self.executor, context, pool, zone, zone.action)) # Send a NOTIFY to each also-notifies for also_notify in pool.also_notifies: notify_target = AlsoNotifyTask() notify_target.options = { 'host': also_notify.host, 'port': also_notify.port } all_tasks.append( zonetasks.SendNotify(self.executor, zone, notify_target)) return self.executor.run(all_tasks)
def _do_zone_action(self, context, zone): pool = self.get_pool(zone.pool_id) task = zonetasks.ZoneAction(self.executor, context, pool, zone, zone.action) return self.executor.run(task)