class TestAccountClient():
def setup(self):
self.client = AccountClient()
def test_add_account_success(self):
""" ACCOUNT (CLIENTS): create a new account and get information about account."""
account = account_name_generator()
type = 'USER'
ret = self.client.add_account(account, type)
assert_true(ret)
with assert_raises(Duplicate):
self.client.add_account(account, type)
with assert_raises(InvalidObject):
self.client.add_account('BAD_ACCOUNT_NAME', type)
with assert_raises(InvalidObject):
self.client.add_account('toooooooloooooonaccounnnnnnnntnammmmme', type)
acc_info = self.client.get_account(account)
assert_equal(acc_info['account'], account)
@raises(AccountNotFound)
def test_get_account_notfound(self):
""" ACCOUNT (CLIENTS): try to get information about not existing account."""
account = str(uuid())
self.client.get_account(account)
def test_list_accounts(self):
""" ACCOUNT (CLIENTS): get list of all accounts."""
dn = '/C=CH/ST=Geneva/O=CERN/OU=PH-ADP-CO/CN=DDMLAB Client Certificate/[email protected]'
acc_list = [account_name_generator() for i in xrange(5)]
for account in acc_list:
self.client.add_account(account, 'USER')
svr_list = [a['account'] for a in self.client.list_accounts(account_type='SERVICE', identity=dn)]
assert_true('root' in svr_list)
svr_list = [a['account'] for a in self.client.list_accounts(account_type='USER')]
for account in acc_list:
assert_true(account in svr_list)
class TestAccountClient(unittest.TestCase):
def setUp(self):
self.client = AccountClient()
def test_add_account_success(self):
""" ACCOUNT (CLIENTS): create a new account and get information about account."""
account = account_name_generator()
type, email = 'USER', '*****@*****.**'
ret = self.client.add_account(account, type, email)
assert ret
with pytest.raises(Duplicate):
self.client.add_account(account, type, email)
with pytest.raises(InvalidObject):
self.client.add_account('BAD_ACCOUNT_NAME', type, email)
with pytest.raises(InvalidObject):
self.client.add_account('toooooooloooooonaccounnnnnnnntnammmmme', type, email)
acc_info = self.client.get_account(account)
assert acc_info['account'] == account
def test_get_account_notfound(self):
""" ACCOUNT (CLIENTS): try to get information about not existing account."""
account = str(uuid())
with pytest.raises(AccountNotFound):
self.client.get_account(account)
def test_list_accounts(self):
""" ACCOUNT (CLIENTS): get list of all accounts."""
dn = config_get('bootstrap', 'x509_identity')
acc_list = [account_name_generator() for _ in range(5)]
for account in acc_list:
self.client.add_account(account, 'USER', '*****@*****.**')
svr_list = [a['account'] for a in self.client.list_accounts(account_type='SERVICE', identity=dn)]
assert 'root' in svr_list
svr_list = [a['account'] for a in self.client.list_accounts(account_type='USER')]
for account in acc_list:
assert account in svr_list
def test_update_account(self):
""" ACCOUNT (CLIENTS): create a new account and update it."""
account = account_name_generator()
type, email = 'USER', '*****@*****.**'
ret = self.client.add_account(account, type, email)
assert ret
self.client.update_account(account=account, key='status', value='SUSPENDED')
status = self.client.get_account(account=account)['status']
assert status == 'SUSPENDED'
self.client.update_account(account=account, key='status', value='ACTIVE')
status = self.client.get_account(account=account)['status']
assert status == 'ACTIVE'
self.client.update_account(account=account, key='email', value='test')
email = self.client.get_account(account=account)['email']
assert email == 'test'
class TestAccountClient():
def __init__(self):
self.client = AccountClient()
def test_add_account_success(self):
""" ACCOUNT (CLIENTS): create a new account and get information about account."""
account = account_name_generator()
type, email = 'USER', '*****@*****.**'
ret = self.client.add_account(account, type, email)
assert_true(ret)
with assert_raises(Duplicate):
self.client.add_account(account, type, email)
with assert_raises(InvalidObject):
self.client.add_account('BAD_ACCOUNT_NAME', type, email)
with assert_raises(InvalidObject):
self.client.add_account('toooooooloooooonaccounnnnnnnntnammmmme',
type, email)
acc_info = self.client.get_account(account)
assert_equal(acc_info['account'], account)
@raises(AccountNotFound)
def test_get_account_notfound(self):
""" ACCOUNT (CLIENTS): try to get information about not existing account."""
account = str(uuid())
self.client.get_account(account)
def test_list_accounts(self):
""" ACCOUNT (CLIENTS): get list of all accounts."""
dn = config_get('bootstrap', 'x509_identity')
acc_list = [account_name_generator() for _ in xrange(5)]
for account in acc_list:
self.client.add_account(account, 'USER', '*****@*****.**')
svr_list = [
a['account']
for a in self.client.list_accounts(account_type='SERVICE',
identity=dn)
]
assert_true('root' in svr_list)
svr_list = [
a['account']
for a in self.client.list_accounts(account_type='USER')
]
for account in acc_list:
assert_true(account in svr_list)
def test_ban_unban_account(self):
""" ACCOUNT (CLIENTS): create a new account and ban/unban it."""
account = account_name_generator()
type, email = 'USER', '*****@*****.**'
ret = self.client.add_account(account, type, email)
assert_true(ret)
self.client.set_account_status(account=account, status='SUSPENDED')
status = self.client.get_account(account=account)['status']
assert_equal(status, 'SUSPENDED')
self.client.set_account_status(account=account, status='ACTIVE')
status = self.client.get_account(account=account)['status']
assert_equal(status, 'ACTIVE')
