def add_interaction():
if not g.user:
flash("Please sign up to access user functionality")
return redirect("/signup")
form = InteractionForm()
reps = [(rep.id, rep.full_name) for rep in g.user.representatives]
repid = request.args["repId"]
form.representative.choices = reps
form.representative.default = repid
if form.validate_on_submit():
interaction_date = form.interaction_date.data
representative = form.representative.data
medium = form.medium.data
topic = form.topic.data
content = form.content.data
rep = Representative.query.get(representative)
dist = District.query.get(rep.district.id)
Interaction.add_intertaction(
user=g.user,
representative=rep,
district=dist,
interaction_date=interaction_date,
medium=medium,
topic=topic,
content=content,
)
flash("Interaction sucessfully added")
return redirect("/user/interactions")
form.process()
return render_template("add-interaction.html", form=form, user=g.user)
def test_add_interaction(self):
interaction = Interaction.add_intertaction(self.user1, self.rep1,
self.rep1.district,
'2020-08-26', 'email',
'stuff', 'nonsense')
testinteraction = Interaction.query.get(1)
self.assertIsNotNone(testinteraction)
self.assertEqual(testinteraction.user, self.user1)
self.assertEqual(testinteraction.representative, self.rep1)
self.assertEqual(testinteraction.district, self.rep1.district)
# self.assertEqual(testinteraction.interaction_date, '2020-08-26')
self.assertEqual(testinteraction.medium, 'email')
self.assertEqual(testinteraction.topic, 'stuff')
self.assertEqual(testinteraction.content, 'nonsense')