コード例 #1
0
ファイル: routes.py プロジェクト: WyrickM/E-commerce-Website
def report(user_id):
    user = User.query.get(user_id)
    form = ReportForm()
    if form.validate_on_submit():
        report = Report(title = form.title.data, username = current_user.username, body = form.body.data)
        db.session.add(report)
        db.session.commit()
        report.add_report(user, current_user)
        db.session.commit()
        flash('Your report has been recorded.')
        return redirect('/clear_search')
    return render_template('report.html', title='Report User', form=form)
コード例 #2
0
    def test_report(self):
        seller = User(username="******",
                      email="*****@*****.**",
                      first_name="SellerFirst",
                      last_name="SellerLast",
                      address="Seller Address",
                      card_number="1111111111111111",
                      security_code="111",
                      is_seller=True)
        seller.set_password('Seller')
        db.session.add(seller)
        db.session.commit()
        buyer = User(username="******",
                     email="*****@*****.**",
                     first_name="BuyerFirst",
                     last_name="BuyerLast",
                     address="Buyer Address",
                     card_number="2222222222222222",
                     security_code="222",
                     is_seller=False)
        buyer.set_password("Buyer")
        db.session.add(buyer)
        db.session.commit()
        report = Report(body='This is a test report for a test seller!')
        db.session.add(report)
        db.session.commit()

        self.assertEqual(report.body,
                         'This is a test report for a test seller!')

        self.assertEqual(seller.reports.all(), [])
        self.assertFalse(report in seller.reports)
        report.add_report(seller, buyer)

        self.assertEqual(seller.reports.count(), 1)

        self.assertEqual(seller.reports.first().username, 'ABuyer')
        self.assertEqual(seller.reports.first().body,
                         'This is a test report for a test seller!')
        self.assertEqual(seller.reports.first().reported_id, seller.id)