def test_scheduler_event_handlers(mocker):
sched = mocker.Mock()
driver = mocker.Mock()
proxy = SchedulerProxy(sched)
proxy.executorLost(driver, mesos_pb2.ExecutorInfo(),
mesos_pb2.ExecutorID(), 1)
proxy.frameworkMessage(driver, mesos_pb2.ExecutorID(),
mesos_pb2.SlaveID(), 'message')
proxy.offerRescinded(driver, mesos_pb2.OfferID())
proxy.registered(driver, mesos_pb2.FrameworkID(), mesos_pb2.MasterInfo())
proxy.resourceOffers(driver, [mesos_pb2.Offer(), mesos_pb2.Offer()])
proxy.slaveLost(driver, mesos_pb2.SlaveID())
proxy.statusUpdate(driver, mesos_pb2.TaskStatus())
proxy.reregistered(driver, mesos_pb2.MasterInfo())
proxy.error(driver, 'message')
proxy.disconnected(driver)
sched.on_executor_lost.assert_called_once()
sched.on_message.assert_called_once()
sched.on_rescinded.assert_called_once()
sched.on_registered.assert_called_once()
sched.on_offers.assert_called_once()
sched.on_slave_lost.assert_called_once()
sched.on_update.assert_called_once()
sched.on_reregistered.assert_called_once()
sched.on_error.assert_called_once()
sched.on_disconnected.assert_called_once()
def get_task_log_param(name,slave_id):
task_log={}
task_log['slave_id']=mesos_pb2.SlaveID()
task_log['slave_id'].value= slave_id
task_log['task_name']=name
task_log['slave_ip']=''
return task_log
def _query_slave(self, task_id):
assert task_id in self._mesos_slaves
log.info('Querying MySQL slave %s for its log position (epoch=%s)' %
(str((task_id, self._mesos_slaves[task_id])), self._epoch))
# Because the elector re-sends messages, it's necessary to use the epoch to differentiate
# responses for each election.
self._driver.sendFrameworkMessage(
mesos_pb2.ExecutorID(value=task_id),
mesos_pb2.SlaveID(value=self._mesos_slaves[task_id]),
str(self._epoch)
) # Send the slave the epoch so it can be included in the response.
def send_data(self, pid, type, data):
if self.driver.aborted:
raise RuntimeError('driver already aborted')
msg = pickle.dumps((pid, type, data))
for slave_id, procs in self.slave_to_proc.iteritems():
if pid in procs:
self.driver.sendFrameworkMessage(
self.executor.executor_id,
mesos_pb2.SlaveID(value=slave_id), msg)
return
raise RuntimeError('Cannot find slave for pid %s' % (pid, ))
def setUp(self):
self.framework_id = mesos_pb2.FrameworkID(value=self.FRAMEWORK_ID)
self.framework_info = mesos_pb2.FrameworkInfo(
user='******',
name='fake_framework_name',
)
self.command_info = mesos_pb2.CommandInfo(value='fake-command')
self.executor_id = mesos_pb2.ExecutorID(value='fake-executor-id')
self.executor_info = mesos_pb2.ExecutorInfo(
executor_id=self.executor_id,
framework_id=self.framework_id,
command=self.command_info,
)
self.slave_id = mesos_pb2.SlaveID(value='fake-slave-id')
self.offer_id = mesos_pb2.OfferID(value='1')
