示例#1
0
    def test_insert_duplicate_uid_and_rid_points_entry(self):
        """
        Test insert a normal points with same uid and rid. Expect an error message as well as data in database remains unchanged.
        """
        points = Points(uid=1, rid=12, points=10)
        db.session.add(points)
        errmsg = pointshelper.insert_points(1, 12)
        self.assertEqual(
            errmsg,
            ["Points entry already exists for given user at this restaurant."])

        points = Points.query.filter_by(uid=1, rid=12)
        self.assertEqual(points.count(), 1)
        self.assertEqual(points.first().pid, 1)
        self.assertEqual(points.first().uid, 1)
        self.assertEqual(points.first().rid, 12)
        self.assertEqual(points.first().points, 10)
示例#2
0
    def test_insert_duplicate_rid_points_entry(self):
        """
        Test insert multi normal points with same rid. Expect output to match correct data.
        """
        points = Points(uid=1, rid=12, points=10)
        db.session.add(points)
        errmsg = pointshelper.insert_points(2, 12)
        self.assertEqual(errmsg, None)

        points = Points.query.filter_by(rid=12)
        self.assertEqual(points.count(), 2)

        points = Points.query.filter_by(pid=1).first()
        self.assertNotEqual(points, None)
        self.assertEqual(points.uid, 1)
        self.assertEqual(points.rid, 12)
        self.assertEqual(points.points, 10)

        points = Points.query.filter_by(pid=2).first()
        self.assertNotEqual(points, None)
        self.assertEqual(points.uid, 2)
        self.assertEqual(points.rid, 12)
        self.assertEqual(points.points, 0)