def test_user_project_membership(self): u1 = User(username="******", email="*****@*****.**") u1.set_password("test_pass") p1 = Project(name="project #1", description="test project #1") db.session.add_all([u1, p1]) db.session.commit() self.assertFalse(u1.is_in_project(p1)) p1.add_to_project(u1) db.session.commit() self.assertTrue(u1.is_in_project(p1))
def new_project(): form = NewComponentForm() if form.validate_on_submit(): project = Project(name=form.name.data, description=form.description.data) db.session.add(project) project.add_to_project(current_user) db.session.commit() flash("Project - \"{}\" created successfully!".format(project.name)) return redirect(url_for('main.index')) return render_template('main/new_project.html', form=form)
def test_project_membership(self): u1 = User(username="******", email="*****@*****.**") u1.set_password("test_pass") db.session.add(u1) db.session.commit() self.assertEqual(u1.projects.all(), []) p1 = Project(name="project #1", description="test project #1") db.session.add(p1) db.session.commit() self.assertEqual(p1.members.all(), []) p1.add_to_project(u1) db.session.commit() self.assertEqual(p1.members.count(), 1) self.assertEqual(p1.members.first().username, "john123") self.assertEqual(u1.projects.count(), 1) self.assertEqual(u1.projects.first().name, "project #1") p1.remove_from_project(u1) db.session.commit() self.assertEqual(p1.members.all(), []) self.assertEqual(u1.projects.all(), [])