def setUp(self): """Do at the beginning of every test""" # Get the Flask test client self.client = app.test_client() # Show Flask errors that happen during tests app.config['TESTING'] = True # Connect to test database connect_to_db(app, "postgresql:///testdb") # Create tables from model db.create_all() # Import different types of data from seed file seed.load_users() seed.load_groups() seed.load_usergroups() seed.load_patterns() seed.load_invites() # Reset auto incrementing primary keys to start after seed data seed.set_val_user_id() seed.set_val_group_id() seed.set_val_usergroup_id() seed.set_val_pattern_id() seed.set_val_invite_id() with self.client as c: with c.session_transaction() as sess: sess['user_id'] = 1 c.set_cookie('localhost', 'MYCOOKIE', 'cookie_value')
def setUp(cls): """Do once before all tests in this class""" # Get the Flask test client cls.client = app.test_client() # Show Flask errors that happen during tests app.config['TESTING'] = True # Connect to test database connect_to_db(app, "postgresql:///testdb") # Create tables from model db.create_all() # Import different types of data from seed file seed.load_users() seed.load_groups() seed.load_usergroups() seed.load_patterns() seed.load_invites() seed.set_val_user_id() seed.set_val_group_id() seed.set_val_usergroup_id() seed.set_val_pattern_id() seed.set_val_invite_id() with cls.client as c: with c.session_transaction() as sess: sess['user_id'] = 1 c.set_cookie('localhost', 'MYCOOKIE', 'cookie_value')
def setUp(self): """Set up prior to every test""" self.client = app.test_client() app.config["TESTING"] = True # Set fake login with self.client as c: with c.session_transaction() as sess: sess['user_id'] = 1 # Connect to test database connect_to_db(app, "postgresql:///testdb") # Create tables if not already created. Delete all entries in tables db.create_all() seed.User_Trail.query.delete() seed.Trip_User.query.delete() seed.Trip_Trail.query.delete() seed.Trip.query.delete() seed.Trail.query.delete() seed.User.query.delete() # Seed sample data into the database, and set the PK id values seed.load_sample_users() seed.load_sample_trails() seed.load_sample_user_trails() seed.load_sample_trips() seed.load_sample_trip_users() seed.load_sample_trip_trails() seed.set_val_user_id() seed.set_val_user_trail_id() seed.set_val_trip_id() seed.set_val_trip_user_id() seed.set_val_trip_trail_id()
def setUp(self): # Get the Flask test client self.client = app.test_client() # Show Flask errors that happen during tests app.config['TESTING'] = True # Connect to test database connect_to_db(app, "postgresql:///testdb") # Create tables from model db.create_all() # Import different types of data from seed file seed.load_users() seed.load_groups() seed.load_usergroups() seed.load_patterns() seed.load_invites() # Reset auto incrementing primary keys to start after seed data seed.set_val_user_id() seed.set_val_group_id() seed.set_val_usergroup_id() seed.set_val_pattern_id() seed.set_val_invite_id()
def create_testdb(): connect_to_db(app, "postgresql:///testdb") db.create_all() # Import different types of data seed.delete_everything() seed.load_users() seed.load_tasks() seed.load_kaos() seed.load_collects() seed.load_used_kaos() seed.set_val_user_id() seed.set_val_task_id() seed.set_val_collect_id() seed.set_val_kao_id()
def add_new_user(): """Add new user to database.""" # Process sign-up form and set to respective variables email = request.form.get("email") password = request.form.get("password") # Check if user exists in database and return user object user = db.session.query(User).filter(User.email == email).first() # Want to check if user is a user object # If user is None, go into else statement if user: if user.email == email: flash("you already logged in before silly willy") pass else: # Calling function set_val_user_id() from seed.py fil # to prevent conflicting id's user_id = seed.set_val_user_id() # Instantiating user in the User class user = User(user_id=user_id, email=email, password=password) # We need to add to the transaction or it won't be stored db.session.add(user) # Once we're done, we should commit our work db.session.commit() # To keep user logged in, holding onto user_id in a flask session session['current_user'] = user.user_id flash('You were successfully signed up') return redirect('/')