def setUp(self):
self._role = 'mesos'
self._name = 'jimbob'
self._env = 'test'
self._job_key = JobKey(name=self._name,
environment=self._env,
role=self._role)
self._session_key = 'test_session'
self._lock = 'test_lock'
self._instance_watcher = MockObject(InstanceWatcher)
self._scheduler = MockObject(scheduler_client)
self._scheduler_proxy = FakeSchedulerProxy('test-cluster',
self._scheduler,
self._session_key)
self.init_updater(deepcopy(self.UPDATE_CONFIG))
def setUp(self):
self._role = 'mesos'
self._name = 'jimbob'
self._env = 'test'
self._job_key = JobKey(name=self._name, environment=self._env, role=self._role)
self._session_key = 'test_session'
self._lock = 'test_lock'
self._instance_watcher = MockObject(InstanceWatcher)
self._job_monitor = MockObject(JobMonitor)
self._scheduler_mux = FakeSchedulerMux()
self._scheduler = MockObject(scheduler_client)
self._scheduler_proxy = FakeSchedulerProxy(Cluster(name='test-cluster'),
self._scheduler,
self._session_key)
self._quota_check = MockObject(QuotaCheck)
self.init_updater(deepcopy(self.UPDATE_CONFIG))
self._num_cpus = 1.0
self._num_ram = 1
self._num_disk = 1
def _prepare_clone_mock(self, status):
drv = self._driver
mox = self._mox
volume = FakeVolume()
setattr(volume, 'provider_location', '127.0.0.1:/nfs')
drv._client = MockObject(suds.client.Client)
drv._client.factory = MockObject(suds.client.Factory)
drv._client.service = MockObject(suds.client.ServiceSelector)
# CloneNasFile method is generated by ServiceSelector at runtime from
# the
# XML, so mocking is impossible.
setattr(
drv._client.service, 'CloneNasFile',
types.MethodType(lambda *args, **kwargs: FakeResponce(status),
suds.client.ServiceSelector))
mox.StubOutWithMock(drv, '_get_host_ip')
mox.StubOutWithMock(drv, '_get_export_path')
drv._get_host_ip(IgnoreArg()).AndReturn('127.0.0.1')
drv._get_export_path(IgnoreArg()).AndReturn('/nfs')
return mox
