Remove hardcoded ids from history tests

This commit is contained in:
Andrew Dickinson 2020-04-13 19:32:04 -04:00
parent 931b3f84da
commit d008e88f31

View file

@ -2428,37 +2428,45 @@ class HistoryTestCase(IhatemoneyTestCase):
) )
self.assertEqual(resp.status_code, 200) self.assertEqual(resp.status_code, 200)
user_id = models.Person.query.one().id
# create a bill # create a bill
resp = self.client.post( resp = self.client.post(
"/demo/add", "/demo/add",
data={ data={
"date": "2011-08-10", "date": "2011-08-10",
"what": "fromage à raclette", "what": "fromage à raclette",
"payer": 1, "payer": user_id,
"payed_for": [1], "payed_for": [user_id],
"amount": "25", "amount": "25",
}, },
follow_redirects=True, follow_redirects=True,
) )
self.assertEqual(resp.status_code, 200) self.assertEqual(resp.status_code, 200)
bill_id = models.Bill.query.one().id
# edit the bill # edit the bill
resp = self.client.post( resp = self.client.post(
"/demo/edit/1", "/demo/edit/%i" % bill_id,
data={ data={
"date": "2011-08-10", "date": "2011-08-10",
"what": "fromage à raclette", "what": "fromage à raclette",
"payer": 1, "payer": user_id,
"payed_for": [1], "payed_for": [user_id],
"amount": "10", "amount": "10",
}, },
follow_redirects=True, follow_redirects=True,
) )
self.assertEqual(resp.status_code, 200) self.assertEqual(resp.status_code, 200)
# delete the bill # delete the bill
resp = self.client.get("/demo/delete/1", follow_redirects=True) resp = self.client.get("/demo/delete/%i" % bill_id, follow_redirects=True)
self.assertEqual(resp.status_code, 200) self.assertEqual(resp.status_code, 200)
# delete user using POST method # delete user using POST method
resp = self.client.post("/demo/members/1/delete", follow_redirects=True) resp = self.client.post(
"/demo/members/%i/delete" % user_id, follow_redirects=True
)
self.assertEqual(resp.status_code, 200) self.assertEqual(resp.status_code, 200)
def test_disable_clear_no_new_records(self): def test_disable_clear_no_new_records(self):