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)
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)