def __get_account_from_row(self, row):
assert row is not None
tbl_a = self._table_map['account']
if row[tbl_a.c.host] is None:
return None
if row[tbl_a.c.keyfile]:
account = Account(row[tbl_a.c.name], key = row[tbl_a.c.keyfile])
else:
account = Account(row[tbl_a.c.name])
account.set_password(row[tbl_a.c.password])
account.set_authorization_password(row[tbl_a.c.authorization_password])
#TODO: account.set_skey_password(row[tbl_a.c.skey_password])
return account
def __get_account_from_row(self, row):
assert row is not None
tbl_a = self._table_map['account']
if row[tbl_a.c.host] is None:
return None
if row[tbl_a.c.keyfile]:
account = Account(row[tbl_a.c.name], key=row[tbl_a.c.keyfile])
else:
account = Account(row[tbl_a.c.name])
account.set_password(row[tbl_a.c.password])
account.set_authorization_password(row[tbl_a.c.authorization_password])
#TODO: account.set_skey_password(row[tbl_a.c.skey_password])
return account
class AccountTest(unittest.TestCase):
CORRELATE = Account
def setUp(self):
self.user = '******'
self.password1 = 'test1'
self.password2 = 'test2'
self.key = PrivateKey()
self.account = Account(self.user,
self.password1,
self.password2,
self.key)
def testConstructor(self):
key = PrivateKey()
account = Account(self.user, self.password1, key=key)
self.assertEqual(account.get_key(), key)
self.assertEqual(account.get_password(),
account.get_authorization_password())
account = Account(self.user, self.password1, self.password2)
self.assertNotEqual(account.get_password(),
account.get_authorization_password())
def testContext(self):
with self.account as account:
account.release()
account.acquire()
self.account.acquire()
with self.account.context() as context:
context.release()
context.acquire()
with context.context() as subcontext:
subcontext.release()
subcontext.acquire()
with subcontext.context() as subsubcontext:
subsubcontext.release()
subsubcontext.acquire()
with self.account:
pass
def testAcquire(self):
self.account.acquire()
self.account.release()
self.account.acquire()
self.account.release()
def testRelease(self):
self.assertRaises(Exception, self.account.release)
def testSetName(self):
self.assertEqual(self.user, self.account.get_name())
self.account.set_name('foo')
self.assertEqual('foo', self.account.get_name())
def testGetName(self):
self.assertEqual(self.user, self.account.get_name())
def testSetPassword(self):
self.assertEqual(self.password1, self.account.get_password())
self.account.set_password('foo')
self.assertEqual('foo', self.account.get_password())
def testGetPassword(self):
self.assertEqual(self.password1, self.account.get_password())
def testSetAuthorizationPassword(self):
self.assertEqual(self.password2,
self.account.get_authorization_password())
self.account.set_authorization_password('foo')
self.assertEqual('foo', self.account.get_authorization_password())
def testGetAuthorizationPassword(self):
self.assertEqual(self.password2,
self.account.get_authorization_password())
def testGetKey(self):
self.assertEqual(self.key, self.account.get_key())
class AccountTest(unittest.TestCase):
CORRELATE = Account
def setUp(self):
self.user = '******'
self.password1 = 'test1'
self.password2 = 'test2'
self.key = PrivateKey()
self.account = Account(self.user,
self.password1,
self.password2,
self.key)
def testConstructor(self):
key = PrivateKey()
account = Account(self.user, self.password1, key = key)
self.assertEqual(account.get_key(), key)
self.assertEqual(account.get_password(),
account.get_authorization_password())
account = Account(self.user, self.password1, self.password2)
self.failIfEqual(account.get_password(),
account.get_authorization_password())
def testContext(self):
with self.account as account:
account.release()
account.acquire()
self.account.acquire()
with self.account.context() as context:
context.release()
context.acquire()
with context.context() as subcontext:
subcontext.release()
subcontext.acquire()
with subcontext.context() as subsubcontext:
subsubcontext.release()
subsubcontext.acquire()
with self.account:
pass
def testAcquire(self):
self.account.acquire()
self.account.release()
self.account.acquire()
self.account.release()
def testRelease(self):
self.assertRaises(Exception, self.account.release)
def testSetName(self):
self.assertEqual(self.user, self.account.get_name())
self.account.set_name('foo')
self.assertEqual('foo', self.account.get_name())
def testGetName(self):
self.assertEqual(self.user, self.account.get_name())
def testSetPassword(self):
self.assertEqual(self.password1, self.account.get_password())
self.account.set_password('foo')
self.assertEqual('foo', self.account.get_password())
def testGetPassword(self):
self.assertEqual(self.password1, self.account.get_password())
def testSetAuthorizationPassword(self):
self.assertEqual(self.password2,
self.account.get_authorization_password())
self.account.set_authorization_password('foo')
self.assertEqual('foo', self.account.get_authorization_password())
def testGetAuthorizationPassword(self):
self.assertEqual(self.password2,
self.account.get_authorization_password())
def testGetKey(self):
self.assertEqual(self.key, self.account.get_key())
