def testAdd(self): users = Users() before_count = len(users.all()) self.assertTrue(users.add(users.new(), persist=False)) after_count = len(users.all()) self.assertGreater(after_count, before_count) self.assertEqual(before_count + 1, after_count)
def test_InsertQuery(self): users = Users() entity = users.new() expected = OrderedDict([ ('id', None), ('user_group', 'user'), ('username', None), ('name', None), ('email', None), ('password', None), ('virt_auth_data', None) ]) data = users._generate_query_data(entity) self.assertEqual(expected, data) expected = 'INSERT INTO users(id,user_group,username,name,email,password,virt_auth_data) VALUES (?,?,?,?,?,?,?)' self.assertEqual(users._generate_insert_query(data), expected)
def testDelete(self): users = Users() entity = users.new() expected = 'DELETE FROM users WHERE id = ?' self.assertEqual(users._generate_delete_query(entity.name), expected)
def testNew(self): users = Users() self.assertIsInstance(users.new(), User) e = users.new(name="lol") self.assertEqual(e.name, 'lol')
def testFirst(self): users = Users() self.assertEqual(users.first(), None) entity = users.new() users.add(entity, persist=False) self.assertEqual(users.first(), entity)