def test_create_user_no_duplicate(self): """Makes sure users can't have the same username and email.""" Role.create(role_name='tester') create_user('Tester', 'test', '0000000000', 'tester', 'ttest', 'testing', '*****@*****.**') from peewee import IntegrityError with self.assertRaises(IntegrityError): create_user('Tester', 'test', '0000000000', 'tester', 'ttest', 'testing', '*****@*****.**') delete_user = User.delete().where(User.username == 'ttest') delete_user.execute() delete_role = Role.delete().where(Role.role_name == 'tester') delete_role.execute()
def test_create_user(self): """Create an user.""" Role.create(role_name='tester') create_user('Tester', 'test', '0000000000', 'tester', 'ttest', 'testing', '*****@*****.**') user = User.select().where(User.username == 'ttest') delete_user = User.delete().where(User.username == 'ttest') delete_user.execute() delete_role = Role.delete().where(Role.role_name == 'tester') delete_role.execute() self.assertEqual(str(user), ("<class 'models.User'> SELECT `t1`.`id`, `t1`.`first_name`," " `t1`.`second_name`, `t1`.`phone`, `t1`.`role_id`," " `t1`.`username`, `t1`.`password`, `t1`.`email` FROM `user`" " AS t1 WHERE (`t1`.`username` = %s) ['ttest']"))
def test_create_outgoing(self): """Create an outgoing record.""" Category.create(category_id='001', description='Testing Stock') Project.create(project_id='001', project_description="Testing") Role.create(role_name='Admin') User.create(first_name='Jay', second_name='Palm', phone='9783978223', role='Admin', username='******', password='******', email='*****@*****.**') create_stock('001', 'Testing stock', 1, '001', 9.99) create_outgoing_stock(stock='001', project_id='001', date="2015-07-22", quantity='7', username='******') OutgoingStock.get(OutgoingStock.stock == '001').delete_instance() Stock.get(Stock.category == '001').delete_instance() User.get(User.username == 'JayPalm').delete_instance() Role.get(Role.role_name == 'Admin').delete_instance() Project.get(Project.project_id == '001').delete_instance() Category.get(Category.category_id == '001').delete_instance()
def init(): ''' Initialize the database ''' print('Initializing') barrel.db.init(app) Role.create(name='admin') Role.create(name='mod') Role.create(name='user') Group.create( abbr=u'ACME', name=u'Administration, Control and Management Environment') add_user('admin', 'nidma', 'admin', group=Group.get_admin_group()) return 'Database initialized successfully'
def create_role(role_name): """Takes the role name as an argument and create it.""" db.connect() Role.create(role_name=role_name) db.close()