class TestAccountClient(): @classmethod def setUpClass(cls): # Add test account cls.account = ''.join(random.choice(string.ascii_uppercase) for x in range(10)) add_account(account=cls.account, type=AccountType.USER) # Add test RSE cls.rse1 = 'MOCK' cls.rse2 = 'MOCK2' cls.rse1_id = get_rse(cls.rse1).id cls.rse2_id = get_rse(cls.rse2).id def setup(self): self.client = AccountClient() self.alclient = AccountLimitClient() def test_listing_account_limits(self): """ ACCOUNT (CLIENTS): Test listing account limits """ account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=12345) account_limit.set_account_limit(account=self.account, rse_id=self.rse2_id, bytes=12345) limits = self.client.get_account_limits(account=self.account) assert_in((self.rse1, 12345), limits.items()) assert_in((self.rse2, 12345), limits.items()) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.delete_account_limit(account=self.account, rse_id=self.rse2_id) def test_listing_account_limit(self): """ ACCOUNT (CLIENTS): Test listing account limit """ account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=333) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit, {self.rse1: 333}) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_setting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test setting account limit """ self.alclient.set_account_limit(account=self.account, rse=self.rse1, bytes=987) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit[self.rse1], 987) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_deleting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test deleting account limit """ self.alclient.set_account_limit(account=self.account, rse=self.rse1, bytes=786) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit, {self.rse1: 786}) self.alclient.delete_account_limit(account=self.account, rse=self.rse1) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit[self.rse1], float('Inf')) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id)
class TestAccountClient(): @classmethod def setUpClass(cls): if config_get_bool('common', 'multi_vo', raise_exception=False, default=False): cls.vo = {'vo': 'tst'} else: cls.vo = {} # Add test account cls.account = InternalAccount( ''.join(random.choice(string.ascii_uppercase) for x in range(10)), **cls.vo) add_account(account=cls.account, type=AccountType.USER, email='*****@*****.**') # Add test RSE cls.rse1 = 'MOCK' cls.rse2 = 'MOCK2' cls.rse1_id = get_rse_id(rse=cls.rse1, **cls.vo) cls.rse2_id = get_rse_id(rse=cls.rse2, **cls.vo) def setup(self): self.client = AccountClient() self.alclient = AccountLimitClient() def test_listing_account_limits(self): """ ACCOUNT (CLIENTS): Test listing account limits """ account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=12345) account_limit.set_account_limit(account=self.account, rse_id=self.rse2_id, bytes=12345) limits = self.client.get_account_limits(account=self.account.external) assert_in((self.rse1, 12345), limits.items()) assert_in((self.rse2, 12345), limits.items()) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.delete_account_limit(account=self.account, rse_id=self.rse2_id) def test_listing_account_limit(self): """ ACCOUNT (CLIENTS): Test listing account limit """ account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=333) limit = self.client.get_account_limit(account=self.account.external, rse=self.rse1) assert_equal(limit, {self.rse1: 333}) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_setting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test setting account limit """ self.alclient.set_account_limit(account=self.account.external, rse=self.rse1, bytes=987) limit = self.client.get_account_limit(account=self.account.external, rse=self.rse1) assert_equal(limit[self.rse1], 987) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_deleting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test deleting account limit """ self.alclient.set_account_limit(account=self.account.external, rse=self.rse1, bytes=786) limit = self.client.get_account_limit(account=self.account.external, rse=self.rse1) assert_equal(limit, {self.rse1: 786}) self.alclient.delete_account_limit(account=self.account.external, rse=self.rse1) limit = self.client.get_account_limit(account=self.account.external, rse=self.rse1) assert_equal(limit[self.rse1], None) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id)
class TestAccountClient(): @classmethod def setUpClass(cls): # Add test account cls.account = ''.join( random.choice(string.ascii_uppercase) for x in range(10)) add_account(account=cls.account, type=AccountType.USER, email='*****@*****.**') # Add test RSE cls.rse1 = 'MOCK' cls.rse2 = 'MOCK2' cls.rse1_id = get_rse_id(rse=cls.rse1) cls.rse2_id = get_rse_id(rse=cls.rse2) def setup(self): self.client = AccountClient() self.alclient = AccountLimitClient() def test_listing_account_limits(self): """ ACCOUNT (CLIENTS): Test listing account limits """ account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=12345) account_limit.set_account_limit(account=self.account, rse_id=self.rse2_id, bytes=12345) limits = self.client.get_account_limits(account=self.account) assert_in((self.rse1, 12345), limits.items()) assert_in((self.rse2, 12345), limits.items()) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.delete_account_limit(account=self.account, rse_id=self.rse2_id) def test_listing_account_limit(self): """ ACCOUNT (CLIENTS): Test listing account limit """ account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) account_limit.set_account_limit(account=self.account, rse_id=self.rse1_id, bytes=333) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit, {self.rse1: 333}) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_setting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test setting account limit """ self.alclient.set_account_limit(account=self.account, rse=self.rse1, bytes=987) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit[self.rse1], 987) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id) def test_deleting_account_limit(self): """ ACCOUNTLIMIT (CLIENTS): Test deleting account limit """ self.alclient.set_account_limit(account=self.account, rse=self.rse1, bytes=786) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit, {self.rse1: 786}) self.alclient.delete_account_limit(account=self.account, rse=self.rse1) limit = self.client.get_account_limit(account=self.account, rse=self.rse1) assert_equal(limit[self.rse1], None) account_limit.delete_account_limit(account=self.account, rse_id=self.rse1_id)