def test_show_widget_with_component(self): poll = Poll(question="fu") poll.widget = self.widget choice = Choice(description="bar") poll.choices.append(choice) db.session.add(poll) db.session.commit() self.login_user(self.admin_user) response = self.app.get("/admin/widget/{}".format(self.widget.id)) self.assertEquals(200, response.status_code)
def test_delete_widget_with_component(self): poll = Poll(question="fu") poll.widget = self.widget choice = Choice(description="bar") poll.choices.append(choice) db.session.add(poll) db.session.commit() self.login_user(self.admin_user) self.app.get("/admin/widget/{}/delete".format(self.widget.id)) widget = self._get_model(Widget, name="test_widget") poll = self._get_model(Poll, question="fu") choice = self._get_model(Choice, description="bar") assert widget is None assert poll is None assert choice is None