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)
