def test_add_project_to_db(client, clean_db):
entry = Project()
entry.client = "project client"
entry.name = "project name"
entry.job_number = "project job number"
entry.timestamp = datetime.datetime.now()
entry.last_active = datetime.datetime.now()
db.session.add(entry)
db.session.commit()
db_entry = Project.query.first_or_404()
assert entry == db_entry
def add():
form = AddProject()
if form.validate_on_submit():
entry = Project()
entry.job_number = form.job_number.data
entry.client = form.client.data
entry.name = form.project.data
entry.timestamp = datetime.datetime.now()
entry.company = current_user.company.id
db.session.add(entry)
db.session.commit()
current_user.company.add_asset(entry.asset)
db.session.commit()
flash("Project added")
return redirect(url_for("project.view", asset=entry.asset))
return render_template("project/add.html", form=form)
