def test_find_value(self):
"""
Find row in table according serching pattern.
"""
self.key_driver.create_table()
first_acc_info = ["test_account", "test_key_string"]
second_acc_info = ["test_account2", "test_key_string2"]
answer1 = [('test_account', 1L, 'test_key_string')]
answer2 = [('test_account2', 2L, 'test_key_string2')]
for data_acc in [first_acc_info, second_acc_info]:
key_info_table.insert().execute(account=data_acc[0],
encryption_key=data_acc[1])
# check, that result empty , if such information not exist in DataBase
self.assertFalse(self.key_driver.find_value("account", "my_account"))
# check, results for currect find requests
# First answer
res = self.key_driver.find_value("account", first_acc_info[0])
self.assertEquals(res, answer1)
res = self.key_driver.find_value("encryption_key", first_acc_info[1])
self.assertEquals(res, answer1)
res = self.key_driver.find_value("key_id", "1")
self.assertEquals(res, answer1)
# Second answer
res = self.key_driver.find_value("account", second_acc_info[0])
self.assertEquals(res, answer2)
res = self.key_driver.find_value("encryption_key", second_acc_info[1])
self.assertEquals(res, answer2)
res = self.key_driver.find_value("key_id", "2")
self.assertEquals(res, answer2)
def test_get_key(self):
"""
Check key value for different account and key_id values.
"""
self.key_driver.create_table()
# create 2 account with key_id
acc_info = ["acc1", "acc2"]
for acc in acc_info:
key_info_table.insert().execute(account=acc)
# check key for first account
key1 = self.key_driver.get_key(1)
key2 = self.key_driver.get_key(1)
self.assertEqual(key1, key2)
# check key for second account
key3 = self.key_driver.get_key(2)
self.assertNotEqual(key2, key3)
# check raise, if incorrect id (no in table)
self.assertRaises(StandardError, self.key_driver.get_key, 100)
# check raise, if incorrect id (string and have not only digits)
self.assertRaises(ValueError, self.key_driver.get_key, "id100")
# check raise, if incorrect id (not string or int)
test = [[2], {"test": "test"}, (222, 2)]
for val in test:
self.assertRaises(TypeError, self.key_driver.get_key, val)
