almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-05-06 05:01:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update the tests to reflect last changes in the mock

Browse source
This commit is contained in:
Alexis Métaireau 2021-04-13 22:42:27 +02:00 committed by Glandos
parent a9128261f1
commit e2e9fe0f52
2 changed files with 110 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

107
ihatemoney/tests/budget_test.py
View file

@ -4,6 +4,7 @@ import json
import re
from time import sleep
import unittest
from unittest.mock import MagicMock
from flask import session
from werkzeug.security import check_password_hash, generate_password_hash
@ -1443,6 +1444,112 @@ class BudgetTestCase(IhatemoneyTestCase):
member = models.Person.query.filter(models.Person.id == 1).one_or_none()
self.assertEqual(member, None)
def test_currency_switch(self):
mock_data = {"USD": 1, "EUR": 0.8, "CAD": 1.2}
converter = CurrencyConverter()
converter.get_rates = MagicMock(return_value=mock_data)
# A project should be editable
self.post_project("raclette")
# add members
self.client.post("/raclette/members/add", data={"name": "zorglub"})
self.client.post("/raclette/members/add", data={"name": "fred"})
self.client.post("/raclette/members/add", data={"name": "tata"})
# create bills
self.client.post(
"/raclette/add",
data={
"date": "2016-12-31",
"what": "fromage à raclette",
"payer": 1,
"payed_for": [1, 2, 3],
"amount": "10.0",
},
)
self.client.post(
"/raclette/add",
data={
"date": "2016-12-31",
"what": "red wine",
"payer": 2,
"payed_for": [1, 3],
"amount": "20",
},
)
self.client.post(
"/raclette/add",
data={
"date": "2017-01-01",
"what": "refund",
"payer": 3,
"payed_for": [2],
"amount": "13.33",
},
)
project = models.Project.query.get("raclette")
# First, no currency
project.switch_currency(CurrencyConverter.no_currency)
bills = project.get_bills()
for bill in bills:
self.assertEqual(bill.original_currency, CurrencyConverter.no_currency)
# Switch to USD.
project.switch_currency("USD")
bills = project.get_bills()
for bill in bills:
self.assertEqual(bill.original_currency, "USD")
# Add bill in EUR.
self.client.post(
"/raclette/add",
data={
"date": "2017-01-01",
"what": "other country",
"payer": 3,
"payed_for": [2],
"amount": "13",
"original_currency": "EUR",
},
)
# Add one bill in CAD.
self.client.post(
"/raclette/add",
data={
"date": "2017-01-01",
"what": "other country",
"payer": 3,
"payed_for": [2],
"amount": "10",
"original_currency": "CAD",
},
)
# Check that the amount is entered as 10CAD and converted to USD.
assert project.get_bills().first().converted_amount == 8.33
# If we switch back to CAD.
project.switch_currency("CAD")
converted = [
(b.amount, b.original_currency, b.converted_amount)
for b in project.get_bills()
]
assert converted == [
(10.0, "CAD", 10.0),
(13.0, "EUR", 19.5),
(13.33, "USD", 16.0),
(20.0, "USD", 24.0),
(10.0, "USD", 12.0),
]
if __name__ == "__main__":
unittest.main()

6
ihatemoney/tests/main_test.py
View file

@ -240,7 +240,7 @@ class EmailFailureTestCase(IhatemoneyTestCase):
class TestCurrencyConverter(unittest.TestCase):
converter = CurrencyConverter()
mock_data = {"USD": 1, "EUR": 0.8115}
mock_data = {"USD": 1, "EUR": 0.8, "CAD": 1.2}
converter.get_rates = MagicMock(return_value=mock_data)
def test_only_one_instance(self):
@ -249,11 +249,11 @@ class TestCurrencyConverter(unittest.TestCase):
self.assertEqual(one, two)
def test_get_currencies(self):
self.assertCountEqual(self.converter.get_currencies(), ["USD", "EUR"])
self.assertCountEqual(self.converter.get_currencies(), ["USD", "EUR", "CAD"])
def test_exchange_currency(self):
result = self.converter.exchange_currency(100, "USD", "EUR")
self.assertEqual(result, 81.15)
self.assertEqual(result, 80.0)
if __name__ == "__main__":