def test_api_error(self):
test_dir = tempfile.mkdtemp()
open(test_dir + '/blacklist', 'w+').close()
api = mock.Mock(spec_set=["allocate_jobsteps"])
api.allocate_jobsteps.side_effect = APIError("Failure")
cs = ChangesScheduler(test_dir, 'nostatefile', api=api)
driver = mock.Mock()
offer = mesos_pb2.Offer(
id=mesos_pb2.OfferID(value="offerid"),
framework_id=mesos_pb2.FrameworkID(value="frameworkid"),
slave_id=mesos_pb2.SlaveID(value="slaveid"),
hostname='hostname',
)
offer.resources.add(name="cpus",
type=mesos_pb2.Value.SCALAR,
scalar=mesos_pb2.Value.Scalar(value=4))
offer.resources.add(name="mem",
type=mesos_pb2.Value.SCALAR,
scalar=mesos_pb2.Value.Scalar(value=8192))
cs.resourceOffers(driver, [offer])
api.allocate_jobsteps.assert_called_once_with({
"resources": {"cpus": 4, "mem": 8192},
})
driver.declineOffer.assert_called_once_with(offer.id)
def test_blacklist(self):
test_dir = tempfile.mkdtemp()
state_file = test_dir + '/test.json'
blacklist = open(test_dir + '/blacklist', 'w+')
cs = ChangesScheduler(test_dir, state_file, api=mock.Mock())
driver = mock.Mock()
offer = mock.Mock()
offer.hostname = 'some_hostname.com'
offer.id = '999'
blacklist.write('some_hostname.com\n')
blacklist.close()
cs.resourceOffers(driver, [offer])
driver.declineOffer.assert_called_once_with(offer.id)
