def setUp(self):
super(FilterSchedulerTestCase, self).setUp()
self.mock_placement_client = mock.Mock()
p = mock.patch('zun.scheduler.client.report.SchedulerReportClient',
return_value=self.mock_placement_client)
p.start()
self.addCleanup(p.stop)
self.context = context.RequestContext('fake_user', 'fake_project')
self.driver = self.driver_cls()
def make_context(*args, **kwargs):
# If context hasn't been constructed with token_info
if not kwargs.get('auth_token_info'):
kwargs['auth_token_info'] = copy.deepcopy(token_info)
if not kwargs.get('project_id'):
kwargs['project_id'] = 'fake_project'
if not kwargs.get('user_id'):
kwargs['user_id'] = 'fake_user'
context = zun_context.RequestContext(*args, **kwargs)
return zun_context.RequestContext.from_dict(context.to_dict())
def _create_context(self, is_admin=True, roles=None):
return zun_context.RequestContext(auth_token='auth_token1',
domain_id='domain_id1',
domain_name='domain_name1',
user_name='user1',
user_id='user-id1',
project_name='tenant1',
project_id='tenant-id1',
roles=roles,
is_admin=is_admin,
read_only=True,
show_deleted=True,
request_id='request_id1',
trust_id='trust_id1',
auth_token_info='token_info1')
def setUp(self):
super(TestCase, self).setUp()
token_info = {
'token': {
'project': {
'id': 'fake_project'
},
'user': {
'id': 'fake_user'
}
}
}
self.context = zun_context.RequestContext(
auth_token_info=token_info,
project_id='fake_project',
is_admin=False,
user_id='fake_user')
def make_context(*args, **kwargs):
# If context hasn't been constructed with token_info
if not kwargs.get('auth_token_info'):
kwargs['auth_token_info'] = copy.deepcopy(token_info)
if not kwargs.get('project_id'):
kwargs['project_id'] = 'fake_project'
if not kwargs.get('user_id'):
kwargs['user_id'] = 'fake_user'
if not kwargs.get('is_admin'):
kwargs['is_admin'] = False
context = zun_context.RequestContext(*args, **kwargs)
return zun_context.RequestContext.from_dict(context.to_dict())
p = mock.patch.object(zun_context, 'make_context',
side_effect=make_context)
self.mock_make_context = p.start()
self.addCleanup(p.stop)
self.policy = self.useFixture(policy_fixture.PolicyFixture())
self.useFixture(conf_fixture.ConfFixture())
self._base_test_obj_backup = copy.copy(
objects_base.ZunObjectRegistry._registry._obj_classes)
self.addCleanup(self._restore_obj_registry)
def reset_pecan():
pecan.set_config({}, overwrite=True)
self.addCleanup(reset_pecan)
def setUp(self):
super(TestDiskFilter, self).setUp()
self.context = context.RequestContext('fake_user', 'fake_project')
def test_with_alternate_context(self):
context1 = zun_context.RequestContext('foo', 'foo')
context2 = zun_context.RequestContext('bar', project_id='alternate')
obj = MyObj.query(context1)
obj.update_test(context2)
self.assertEqual('alternate-context', obj.bar)
def setUp(self):
super(FilterSchedulerTestCase, self).setUp()
self.context = context.RequestContext('fake_user', 'fake_project')
self.driver = self.driver_cls()
def setUp(self):
super(TestAvailabilityZoneFilter, self).setUp()
self.context = context.RequestContext('fake_user', 'fake_project')
