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(), [])
