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)