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)
