def test_insert(self): print("Testing insert...") test_user = User(0, "dpham", "password", False) UserSQL.insert(test_user) users = UserSQL.selectAll() self.assertEqual(len(users), 3)
def setUp(self): print("Setup...") UserSQL.create_table() self.test_user1 = User(id=1, userID='admin', password='******', admin=True) self.test_user2 = User(id=2, userID='user', password='******', admin=False) UserSQL.insert(self.test_user1) UserSQL.insert(self.test_user2)
import sys sys.path.append("..") from project.user.models import User from database import UserSQL UserSQL.create_table() test_user1 = User(id=1, userID='admin', password='******', admin=True) test_user2 = User(id=2, userID='user', password='******', admin=False) UserSQL.insert(test_user1) UserSQL.insert(test_user2)