def astakos_user(user): """ Context manager to mock astakos response. usage: with astakos_user("*****@*****.**"): .... make api calls .... """ with patch("snf_django.lib.api.get_token") as get_token: get_token.return_value = "DummyToken" with patch("astakosclient.AstakosClient.authenticate") as m2: m2.return_value = { "access": { "token": { "expires": "2013-06-19T15:23:59.975572+00:00", "id": "DummyToken", "tenant": {"id": text.udec(user, "utf8"), "name": "Firstname Lastname"}, }, "serviceCatalog": [], "user": { "roles_links": [], "id": text.udec(user, "utf8"), "roles": [{"id": 1, "name": "default"}], "name": "Firstname Lastname", }, } } with patch("astakosclient.AstakosClient.get_quotas") as m3: m3.return_value = { "system": {"pithos.diskspace": {"usage": 0, "limit": 1073741824, "pending": 0}} # 1GB } issue_fun = "astakosclient.AstakosClient.issue_one_commission" with patch(issue_fun) as m3: serials = [] append = serials.append def get_serial(*args, **kwargs): global serial serial += 1 append(serial) return serial m3.side_effect = get_serial resolv_fun = "astakosclient.AstakosClient.resolve_commissions" with patch(resolv_fun) as m4: m4.return_value = {"accepted": serials, "rejected": [], "failed": []} users_fun = "astakosclient.AstakosClient.get_usernames" with patch(users_fun) as m5: def get_usernames(*args, **kwargs): uuids = args[-1] return dict((uuid, uuid) for uuid in uuids) m5.side_effect = get_usernames yield
def setUp(self): self.client = PithosTestClient() # Override default block size to spead up tests pithos_settings.BACKEND_BLOCK_SIZE = TEST_BLOCK_SIZE pithos_settings.BACKEND_HASH_ALGORITHM = TEST_HASH_ALGORITHM self.user = '******' self.pithos_path = join_urls( get_service_path(pithos_settings.pithos_services, 'object-store')) # patch astakosclient.AstakosClient.validate_token mock_validate_token = self.create_patch( 'astakosclient.AstakosClient.validate_token') mock_validate_token.return_value = { 'access': { 'user': { 'id': text.udec(self.user, 'utf8') } } } # patch astakosclient.AstakosClient.get_token mock_get_token = self.create_patch( 'astakosclient.AstakosClient.get_token') mock_get_token.return_value = {'access_token': 'valid_token'} # patch astakosclient.AstakosClient.api_oa2_auth mock_api_oauth2_auth = self.create_patch( 'astakosclient.AstakosClient.oauth2_url', new_callable=PropertyMock) mock_api_oauth2_auth.return_value = '/astakos/oauth2/' mock_service_get_quotas = self.create_patch( 'astakosclient.AstakosClient.service_get_quotas') mock_service_get_quotas.return_value = { self.user: { "system": { "pithos.diskspace": { "usage": 0, "limit": 1073741824, # 1GB "pending": 0 } } } }
def setUp(self): self.client = PithosTestClient() # Override default block size to spead up tests pithos_settings.BACKEND_BLOCK_SIZE = TEST_BLOCK_SIZE pithos_settings.BACKEND_HASH_ALGORITHM = TEST_HASH_ALGORITHM self.user = '******' self.pithos_path = join_urls(get_service_path( pithos_settings.pithos_services, 'object-store')) # patch astakosclient.AstakosClient.validate_token mock_validate_token = self.create_patch( 'astakosclient.AstakosClient.validate_token') mock_validate_token.return_value = { 'access': {'user': {'id': text.udec(self.user, 'utf8')}}} # patch astakosclient.AstakosClient.get_token mock_get_token = self.create_patch( 'astakosclient.AstakosClient.get_token') mock_get_token.return_value = {'access_token': 'valid_token'} # patch astakosclient.AstakosClient.api_oa2_auth mock_api_oauth2_auth = self.create_patch( 'astakosclient.AstakosClient.oauth2_url', new_callable=PropertyMock) mock_api_oauth2_auth.return_value = '/astakos/oauth2/' mock_service_get_quotas = self.create_patch( 'astakosclient.AstakosClient.service_get_quotas') mock_service_get_quotas.return_value = { self.user: { "system": { "pithos.diskspace": { "usage": 0, "limit": 1073741824, # 1GB "pending": 0}}}}
def stringnify(obj): if isinstance(obj, (unicode, str)): return udec(obj) else: return str(obj)
def astakos_user(user): """ Context manager to mock astakos response. usage: with astakos_user("*****@*****.**"): .... make api calls .... """ with patch("snf_django.lib.api.get_token") as get_token: get_token.return_value = "DummyToken" with patch('astakosclient.AstakosClient.authenticate') as m2: m2.return_value = { "access": { "token": { "expires": "2013-06-19T15:23:59.975572+00:00", "id": "DummyToken", "tenant": { "id": text.udec(user, 'utf8'), "name": "Firstname Lastname" } }, "serviceCatalog": [], "user": { "roles_links": [], "id": text.udec(user, 'utf8'), "roles": [{ "id": 1, "name": "default" }], "name": "Firstname Lastname" } } } with patch('astakosclient.AstakosClient.get_quotas') as m3: m3.return_value = { "system": { "pithos.diskspace": { "usage": 0, "limit": 1073741824, # 1GB "pending": 0 } } } issue_fun = \ "astakosclient.AstakosClient.issue_one_commission" with patch(issue_fun) as m3: serials = [] append = serials.append def get_serial(*args, **kwargs): global serial serial += 1 append(serial) return serial m3.side_effect = get_serial resolv_fun = \ 'astakosclient.AstakosClient.resolve_commissions' with patch(resolv_fun) as m4: m4.return_value = { 'accepted': serials, 'rejected': [], 'failed': [] } users_fun = \ 'astakosclient.AstakosClient.get_usernames' with patch(users_fun) as m5: def get_usernames(*args, **kwargs): uuids = args[-1] return dict((uuid, uuid) for uuid in uuids) m5.side_effect = get_usernames yield