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)