def test_task_has_command_to_launch_dworkers(self):
offer = MagicMock()
offer.url.address.ip = 'localhost'
offer.slave_id = MagicMock()
offer.slave_id.value = 'slave-id'
mesos_task = TaskDirector(offer,WorkerTask).make_task_with_id('task-id')
expect(mesos_task.command.value).to(equal('dworker localhost:8787'))
def test_task_has_command_to_launch_dcenter(self):
offer = MagicMock()
offer.hostname = 'localhost'
offer.slave_id = MagicMock()
offer.slave_id.value = 'slave-id'
mesos_task = TaskDirector(offer,SchedulerTask).make_task_with_id('task-id')
expect(mesos_task.command.value).to(equal('dscheduler'))
def setUp(self):
self.driver = MagicMock()
offer = MagicMock()
offer.hostname = 'localhost'
offer.slave_id = MagicMock()
offer.slave_id.value = 'slave-id'
offer.id = 1
self.offers = [offer for x in range(9)]
self.id1 = uuid.uuid4()
self.id2 = uuid.uuid4()
self.id_generator = iter([self.id1, self.id2]).next
def test_create_task_configures_resources_with_offers(self):
offer = MagicMock()
offer.url.address.ip = 'localhost'
offer.slave_id = MagicMock()
offer.slave_id.value = 'slave-id'
mesos_task = TaskDirector(offer,WorkerTask).make_task_with_id('task-id')
expect(mesos_task.slave_id.value).to(equal('slave-id'))
expect(mesos_task.task_id.value).to(equal('task-id'))
expect(len(mesos_task.resources)).to(equal(2))
expect(mesos_task.resources[0].name).to(equal('cpus'))
expect(mesos_task.resources[0].type).to(equal(mesos_pb2.Value.SCALAR))
expect(mesos_task.resources[0].scalar.value).to(equal(2))
expect(mesos_task.resources[1].name).to(equal('mem'))
expect(mesos_task.resources[1].type).to(equal(mesos_pb2.Value.SCALAR))
expect(mesos_task.resources[1].scalar.value).to(equal(128))
