def testAdd(self):
keys = Keys()
before_count = len(keys.all())
self.assertTrue(keys.add(keys.new(), persist=False))
after_count = len(keys.all())
self.assertGreater(after_count, before_count)
self.assertEqual(before_count + 1, after_count)
def test_InsertQuery(self):
keys = Keys()
entity = keys.new()
expected = OrderedDict([('name', None), ('key', None)])
data = keys._generate_query_data(entity)
self.assertEqual(expected, data)
expected = 'INSERT INTO keys(name,key) VALUES (?,?)'
self.assertEqual(keys._generate_insert_query(data), expected)
def testDelete(self):
keys = Keys()
entity = keys.new()
expected = 'DELETE FROM keys WHERE name = ?'
self.assertEqual(keys._generate_delete_query(entity.name), expected)
def testNew(self):
keys = Keys()
self.assertIsInstance(keys.new(), Key)
e = keys.new(name="lol")
self.assertEqual(e.name, 'lol')
def testFirst(self):
keys = Keys()
self.assertEqual(keys.first(), None)
entity = keys.new()
keys.add(entity, persist=False)
self.assertEqual(keys.first(), entity)
