mirror of
https://github.com/spiral-project/ihatemoney.git
synced 2025-05-05 20:51:49 +02:00
separate import tests
This commit is contained in:
Youe Graillot
parent
9352ad64dc
commit
ef92181786
2 changed files with 735 additions and 726 deletions
|
|
@ -1,5 +1,4 @@
|
||||||
from collections import defaultdict
|
from collections import defaultdict
|
||||||
import json
|
|
||||||
import re
|
import re
|
||||||
from time import sleep
|
from time import sleep
|
||||||
import unittest
|
import unittest
|
||||||
|
|
@ -1092,731 +1091,6 @@ class BudgetTestCase(IhatemoneyTestCase):
|
||||||
msg=f"{t['amount']} is equal to zero after rounding",
|
msg=f"{t['amount']} is equal to zero after rounding",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_export(self):
|
|
||||||
# Export a simple project without currencies
|
|
||||||
|
|
||||||
self.post_project("raclette")
|
|
||||||
|
|
||||||
# add participants
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "fred"})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "tata"})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "pépé"})
|
|
||||||
|
|
||||||
# create bills
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage à raclette",
|
|
||||||
"payer": 1,
|
|
||||||
"payed_for": [1, 2, 3, 4],
|
|
||||||
"amount": "10.0",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"payer": 2,
|
|
||||||
"payed_for": [1, 3],
|
|
||||||
"amount": "200",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"payer": 3,
|
|
||||||
"payed_for": [2],
|
|
||||||
"amount": "13.33",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
# generate json export of bills
|
|
||||||
resp = self.client.get("/raclette/export/bills.json")
|
|
||||||
expected = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"amount": 200.0,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage \xe0 raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "p\xe9p\xe9"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
|
||||||
|
|
||||||
# generate csv export of bills
|
|
||||||
resp = self.client.get("/raclette/export/bills.csv")
|
|
||||||
expected = [
|
|
||||||
"date,what,amount,currency,payer_name,payer_weight,owers",
|
|
||||||
"2017-01-01,refund,XXX,13.33,tata,1.0,fred",
|
|
||||||
'2016-12-31,red wine,XXX,200.0,fred,1.0,"zorglub, tata"',
|
|
||||||
'2016-12-31,fromage à raclette,10.0,XXX,zorglub,2.0,"zorglub, fred, tata, pépé"',
|
|
||||||
]
|
|
||||||
received_lines = resp.data.decode("utf-8").split("\n")
|
|
||||||
|
|
||||||
for i, line in enumerate(expected):
|
|
||||||
self.assertEqual(
|
|
||||||
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
|
||||||
)
|
|
||||||
|
|
||||||
# generate json export of transactions
|
|
||||||
resp = self.client.get("/raclette/export/transactions.json")
|
|
||||||
expected = [
|
|
||||||
{
|
|
||||||
"amount": 2.00,
|
|
||||||
"currency": "XXX",
|
|
||||||
"receiver": "fred",
|
|
||||||
"ower": "p\xe9p\xe9",
|
|
||||||
},
|
|
||||||
{"amount": 55.34, "currency": "XXX", "receiver": "fred", "ower": "tata"},
|
|
||||||
{
|
|
||||||
"amount": 127.33,
|
|
||||||
"currency": "XXX",
|
|
||||||
"receiver": "fred",
|
|
||||||
"ower": "zorglub",
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
|
||||||
|
|
||||||
# generate csv export of transactions
|
|
||||||
resp = self.client.get("/raclette/export/transactions.csv")
|
|
||||||
|
|
||||||
expected = [
|
|
||||||
"amount,currency,receiver,ower",
|
|
||||||
"2.0,XXX,fred,pépé",
|
|
||||||
"55.34,XXX,fred,tata",
|
|
||||||
"127.33,XXX,fred,zorglub",
|
|
||||||
]
|
|
||||||
received_lines = resp.data.decode("utf-8").split("\n")
|
|
||||||
|
|
||||||
for i, line in enumerate(expected):
|
|
||||||
self.assertEqual(
|
|
||||||
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
|
||||||
)
|
|
||||||
|
|
||||||
# wrong export_format should return a 404
|
|
||||||
resp = self.client.get("/raclette/export/transactions.wrong")
|
|
||||||
self.assertEqual(resp.status_code, 404)
|
|
||||||
|
|
||||||
def test_export_with_currencies(self):
|
|
||||||
self.post_project("raclette", default_currency="EUR")
|
|
||||||
|
|
||||||
# add participants
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "fred"})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "tata"})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "pépé"})
|
|
||||||
|
|
||||||
# create bills
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage à raclette",
|
|
||||||
"payer": 1,
|
|
||||||
"payed_for": [1, 2, 3, 4],
|
|
||||||
"amount": "10.0",
|
|
||||||
"original_currency": "EUR",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "poutine from Québec",
|
|
||||||
"payer": 2,
|
|
||||||
"payed_for": [1, 3],
|
|
||||||
"amount": "100",
|
|
||||||
"original_currency": "CAD",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"payer": 3,
|
|
||||||
"payed_for": [2],
|
|
||||||
"amount": "13.33",
|
|
||||||
"original_currency": "EUR",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
# generate json export of bills
|
|
||||||
resp = self.client.get("/raclette/export/bills.json")
|
|
||||||
expected = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "poutine from Qu\xe9bec",
|
|
||||||
"amount": 100.0,
|
|
||||||
"currency": "CAD",
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage \xe0 raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "p\xe9p\xe9"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
|
||||||
|
|
||||||
# generate csv export of bills
|
|
||||||
resp = self.client.get("/raclette/export/bills.csv")
|
|
||||||
expected = [
|
|
||||||
"date,what,amount,currency,payer_name,payer_weight,owers",
|
|
||||||
"2017-01-01,refund,13.33,EUR,tata,1.0,fred",
|
|
||||||
'2016-12-31,poutine from Québec,100.0,CAD,fred,1.0,"zorglub, tata"',
|
|
||||||
'2016-12-31,fromage à raclette,10.0,EUR,zorglub,2.0,"zorglub, fred, tata, pépé"',
|
|
||||||
]
|
|
||||||
received_lines = resp.data.decode("utf-8").split("\n")
|
|
||||||
|
|
||||||
for i, line in enumerate(expected):
|
|
||||||
self.assertEqual(
|
|
||||||
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
|
||||||
)
|
|
||||||
|
|
||||||
# generate json export of transactions (in EUR!)
|
|
||||||
resp = self.client.get("/raclette/export/transactions.json")
|
|
||||||
expected = [
|
|
||||||
{
|
|
||||||
"amount": 2.00,
|
|
||||||
"currency": "EUR",
|
|
||||||
"receiver": "fred",
|
|
||||||
"ower": "p\xe9p\xe9",
|
|
||||||
},
|
|
||||||
{"amount": 10.89, "currency": "EUR", "receiver": "fred", "ower": "tata"},
|
|
||||||
{"amount": 38.45, "currency": "EUR", "receiver": "fred", "ower": "zorglub"},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
|
||||||
|
|
||||||
# generate csv export of transactions
|
|
||||||
resp = self.client.get("/raclette/export/transactions.csv")
|
|
||||||
|
|
||||||
expected = [
|
|
||||||
"amount,currency,receiver,ower",
|
|
||||||
"2.0,EUR,fred,pépé",
|
|
||||||
"10.89,EUR,fred,tata",
|
|
||||||
"38.45,EUR,fred,zorglub",
|
|
||||||
]
|
|
||||||
received_lines = resp.data.decode("utf-8").split("\n")
|
|
||||||
|
|
||||||
for i, line in enumerate(expected):
|
|
||||||
self.assertEqual(
|
|
||||||
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
|
||||||
)
|
|
||||||
|
|
||||||
# Change project currency to CAD
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
project.switch_currency("CAD")
|
|
||||||
|
|
||||||
# generate json export of transactions (now in CAD!)
|
|
||||||
resp = self.client.get("/raclette/export/transactions.json")
|
|
||||||
expected = [
|
|
||||||
{
|
|
||||||
"amount": 3.00,
|
|
||||||
"currency": "CAD",
|
|
||||||
"receiver": "fred",
|
|
||||||
"ower": "p\xe9p\xe9",
|
|
||||||
},
|
|
||||||
{"amount": 16.34, "currency": "CAD", "receiver": "fred", "ower": "tata"},
|
|
||||||
{"amount": 57.67, "currency": "CAD", "receiver": "fred", "ower": "zorglub"},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
|
||||||
|
|
||||||
# generate csv export of transactions
|
|
||||||
resp = self.client.get("/raclette/export/transactions.csv")
|
|
||||||
|
|
||||||
expected = [
|
|
||||||
"amount,currency,receiver,ower",
|
|
||||||
"3.0,CAD,fred,pépé",
|
|
||||||
"16.34,CAD,fred,tata",
|
|
||||||
"57.67,CAD,fred,zorglub",
|
|
||||||
]
|
|
||||||
received_lines = resp.data.decode("utf-8").split("\n")
|
|
||||||
|
|
||||||
for i, line in enumerate(expected):
|
|
||||||
self.assertEqual(
|
|
||||||
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
|
||||||
)
|
|
||||||
|
|
||||||
def test_import_currencies_in_empty_project_with_currency(self):
|
|
||||||
# Import JSON with currencies in an empty project with a default currency
|
|
||||||
|
|
||||||
self.post_project("raclette", default_currency="EUR")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "poutine from québec",
|
|
||||||
"amount": 50.0,
|
|
||||||
"currency": "CAD",
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.import_project("raclette", json_to_import)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check if all bills have been added
|
|
||||||
self.assertEqual(len(bills), len(json_to_import))
|
|
||||||
|
|
||||||
# Check if name of bills are ok
|
|
||||||
b = [e["what"] for e in bills]
|
|
||||||
b.sort()
|
|
||||||
ref = [e["what"] for e in json_to_import]
|
|
||||||
ref.sort()
|
|
||||||
|
|
||||||
self.assertEqual(b, ref)
|
|
||||||
|
|
||||||
# Check if other informations in bill are ok
|
|
||||||
for j in json_to_import:
|
|
||||||
for b in bills:
|
|
||||||
if b["what"] == j["what"]:
|
|
||||||
self.assertEqual(b["payer_name"], j["payer_name"])
|
|
||||||
self.assertEqual(b["amount"], j["amount"])
|
|
||||||
self.assertEqual(b["currency"], j["currency"])
|
|
||||||
self.assertEqual(b["payer_weight"], j["payer_weight"])
|
|
||||||
self.assertEqual(b["date"], j["date"])
|
|
||||||
|
|
||||||
list_project = [ower for ower in b["owers"]]
|
|
||||||
list_project.sort()
|
|
||||||
list_json = [ower for ower in j["owers"]]
|
|
||||||
list_json.sort()
|
|
||||||
|
|
||||||
self.assertEqual(list_project, list_json)
|
|
||||||
|
|
||||||
def test_import_single_currency_in_empty_project_without_currency(self):
|
|
||||||
# Import JSON with a single currency in an empty project with no
|
|
||||||
# default currency. It should work by stripping the currency from
|
|
||||||
# bills.
|
|
||||||
|
|
||||||
self.post_project("raclette")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.import_project("raclette", json_to_import)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check if all bills have been added
|
|
||||||
self.assertEqual(len(bills), len(json_to_import))
|
|
||||||
|
|
||||||
# Check if name of bills are ok
|
|
||||||
b = [e["what"] for e in bills]
|
|
||||||
b.sort()
|
|
||||||
ref = [e["what"] for e in json_to_import]
|
|
||||||
ref.sort()
|
|
||||||
|
|
||||||
self.assertEqual(b, ref)
|
|
||||||
|
|
||||||
# Check if other informations in bill are ok
|
|
||||||
for i in json_to_import:
|
|
||||||
for j in bills:
|
|
||||||
if j["what"] == i["what"]:
|
|
||||||
self.assertEqual(j["payer_name"], i["payer_name"])
|
|
||||||
self.assertEqual(j["amount"], i["amount"])
|
|
||||||
# Currency should have been stripped
|
|
||||||
self.assertEqual(j["currency"], "XXX")
|
|
||||||
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
|
||||||
self.assertEqual(j["date"], i["date"])
|
|
||||||
|
|
||||||
list_project = [ower for ower in j["owers"]]
|
|
||||||
list_project.sort()
|
|
||||||
list_json = [ower for ower in i["owers"]]
|
|
||||||
list_json.sort()
|
|
||||||
|
|
||||||
self.assertEqual(list_project, list_json)
|
|
||||||
|
|
||||||
def test_import_multiple_currencies_in_empty_project_without_currency(self):
|
|
||||||
# Import JSON with multiple currencies in an empty project with no
|
|
||||||
# default currency. It should fail.
|
|
||||||
|
|
||||||
self.post_project("raclette")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "poutine from québec",
|
|
||||||
"amount": 50.0,
|
|
||||||
"currency": "CAD",
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "EUR",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
# Import should fail
|
|
||||||
self.import_project("raclette", json_to_import, 400)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check that there are no bills
|
|
||||||
self.assertEqual(len(bills), 0)
|
|
||||||
|
|
||||||
def test_import_no_currency_in_empty_project_with_currency(self):
|
|
||||||
# Import JSON without currencies (from ihatemoney < 5) in an empty
|
|
||||||
# project with a default currency.
|
|
||||||
|
|
||||||
self.post_project("raclette", default_currency="EUR")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"amount": 200.0,
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.import_project("raclette", json_to_import)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check if all bills have been added
|
|
||||||
self.assertEqual(len(bills), len(json_to_import))
|
|
||||||
|
|
||||||
# Check if name of bills are ok
|
|
||||||
b = [e["what"] for e in bills]
|
|
||||||
b.sort()
|
|
||||||
ref = [e["what"] for e in json_to_import]
|
|
||||||
ref.sort()
|
|
||||||
|
|
||||||
self.assertEqual(b, ref)
|
|
||||||
|
|
||||||
# Check if other informations in bill are ok
|
|
||||||
for i in json_to_import:
|
|
||||||
for j in bills:
|
|
||||||
if j["what"] == i["what"]:
|
|
||||||
self.assertEqual(j["payer_name"], i["payer_name"])
|
|
||||||
self.assertEqual(j["amount"], i["amount"])
|
|
||||||
# All bills are converted to default project currency
|
|
||||||
self.assertEqual(j["currency"], "EUR")
|
|
||||||
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
|
||||||
self.assertEqual(j["date"], i["date"])
|
|
||||||
|
|
||||||
list_project = [ower for ower in j["owers"]]
|
|
||||||
list_project.sort()
|
|
||||||
list_json = [ower for ower in i["owers"]]
|
|
||||||
list_json.sort()
|
|
||||||
|
|
||||||
self.assertEqual(list_project, list_json)
|
|
||||||
|
|
||||||
def test_import_no_currency_in_empty_project_without_currency(self):
|
|
||||||
# Import JSON without currencies (from ihatemoney < 5) in an empty
|
|
||||||
# project with no default currency.
|
|
||||||
|
|
||||||
self.post_project("raclette")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"amount": 200.0,
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.import_project("raclette", json_to_import)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check if all bills have been added
|
|
||||||
self.assertEqual(len(bills), len(json_to_import))
|
|
||||||
|
|
||||||
# Check if name of bills are ok
|
|
||||||
b = [e["what"] for e in bills]
|
|
||||||
b.sort()
|
|
||||||
ref = [e["what"] for e in json_to_import]
|
|
||||||
ref.sort()
|
|
||||||
|
|
||||||
self.assertEqual(b, ref)
|
|
||||||
|
|
||||||
# Check if other informations in bill are ok
|
|
||||||
for i in json_to_import:
|
|
||||||
for j in bills:
|
|
||||||
if j["what"] == i["what"]:
|
|
||||||
self.assertEqual(j["payer_name"], i["payer_name"])
|
|
||||||
self.assertEqual(j["amount"], i["amount"])
|
|
||||||
self.assertEqual(j["currency"], "XXX")
|
|
||||||
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
|
||||||
self.assertEqual(j["date"], i["date"])
|
|
||||||
|
|
||||||
list_project = [ower for ower in j["owers"]]
|
|
||||||
list_project.sort()
|
|
||||||
list_json = [ower for ower in i["owers"]]
|
|
||||||
list_json.sort()
|
|
||||||
|
|
||||||
self.assertEqual(list_project, list_json)
|
|
||||||
|
|
||||||
def test_import_partial_project(self):
|
|
||||||
# Import a JSON in a project with already existing data
|
|
||||||
|
|
||||||
self.post_project("raclette")
|
|
||||||
self.login("raclette")
|
|
||||||
|
|
||||||
project = self.get_project("raclette")
|
|
||||||
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "fred"})
|
|
||||||
self.client.post("/raclette/members/add", data={"name": "tata"})
|
|
||||||
self.client.post(
|
|
||||||
"/raclette/add",
|
|
||||||
data={
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"payer": 2,
|
|
||||||
"payed_for": [1, 3],
|
|
||||||
"amount": "200",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
json_to_import = [
|
|
||||||
{
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"amount": 13.33,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
},
|
|
||||||
{ # This expense does not have to be present twice.
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "red wine",
|
|
||||||
"amount": 200.0,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "fred",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["zorglub", "tata"],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"date": "2016-12-31",
|
|
||||||
"what": "fromage a raclette",
|
|
||||||
"amount": 10.0,
|
|
||||||
"currency": "XXX",
|
|
||||||
"payer_name": "zorglub",
|
|
||||||
"payer_weight": 2.0,
|
|
||||||
"owers": ["zorglub", "fred", "tata", "pepe"],
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
self.import_project("raclette", json_to_import)
|
|
||||||
|
|
||||||
bills = project.get_pretty_bills()
|
|
||||||
|
|
||||||
# Check if all bills have been added
|
|
||||||
self.assertEqual(len(bills), len(json_to_import))
|
|
||||||
|
|
||||||
# Check if name of bills are ok
|
|
||||||
b = [e["what"] for e in bills]
|
|
||||||
b.sort()
|
|
||||||
ref = [e["what"] for e in json_to_import]
|
|
||||||
ref.sort()
|
|
||||||
|
|
||||||
self.assertEqual(b, ref)
|
|
||||||
|
|
||||||
# Check if other informations in bill are ok
|
|
||||||
for i in json_to_import:
|
|
||||||
for j in bills:
|
|
||||||
if j["what"] == i["what"]:
|
|
||||||
self.assertEqual(j["payer_name"], i["payer_name"])
|
|
||||||
self.assertEqual(j["amount"], i["amount"])
|
|
||||||
self.assertEqual(j["currency"], i["currency"])
|
|
||||||
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
|
||||||
self.assertEqual(j["date"], i["date"])
|
|
||||||
|
|
||||||
list_project = [ower for ower in j["owers"]]
|
|
||||||
list_project.sort()
|
|
||||||
list_json = [ower for ower in i["owers"]]
|
|
||||||
list_json.sort()
|
|
||||||
|
|
||||||
self.assertEqual(list_project, list_json)
|
|
||||||
|
|
||||||
def test_import_wrong_json(self):
|
|
||||||
self.post_project("raclette")
|
|
||||||
self.login("raclette")
|
|
||||||
self.get_project("raclette")
|
|
||||||
|
|
||||||
json_1 = [
|
|
||||||
{ # wrong keys
|
|
||||||
"checked": False,
|
|
||||||
"dimensions": {"width": 5, "height": 10},
|
|
||||||
"id": 1,
|
|
||||||
"name": "A green door",
|
|
||||||
"price": 12.5,
|
|
||||||
"tags": ["home", "green"],
|
|
||||||
}
|
|
||||||
]
|
|
||||||
|
|
||||||
json_2 = [
|
|
||||||
{ # amount missing
|
|
||||||
"date": "2017-01-01",
|
|
||||||
"what": "refund",
|
|
||||||
"payer_name": "tata",
|
|
||||||
"payer_weight": 1.0,
|
|
||||||
"owers": ["fred"],
|
|
||||||
}
|
|
||||||
]
|
|
||||||
|
|
||||||
for json_to_import in [json_1, json_2]:
|
|
||||||
# Import should fail
|
|
||||||
self.import_project("raclette", json_to_import, 400)
|
|
||||||
|
|
||||||
def test_access_other_projects(self):
|
def test_access_other_projects(self):
|
||||||
"""Test that accessing or editing bills and participants from another project fails"""
|
"""Test that accessing or editing bills and participants from another project fails"""
|
||||||
# Create project
|
# Create project
|
||||||
|
|
|
||||||
735
ihatemoney/tests/import_test.py
Normal file
735
ihatemoney/tests/import_test.py
Normal file
|
|
@ -0,0 +1,735 @@
|
||||||
|
import json
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from ihatemoney.tests.common.ihatemoney_testcase import IhatemoneyTestCase
|
||||||
|
|
||||||
|
|
||||||
|
class ImportTestCase(IhatemoneyTestCase):
|
||||||
|
def test_export(self):
|
||||||
|
# Export a simple project without currencies
|
||||||
|
|
||||||
|
self.post_project("raclette")
|
||||||
|
|
||||||
|
# add participants
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "fred"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "tata"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "pépé"})
|
||||||
|
|
||||||
|
# create bills
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage à raclette",
|
||||||
|
"payer": 1,
|
||||||
|
"payed_for": [1, 2, 3, 4],
|
||||||
|
"amount": "10.0",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"payer": 2,
|
||||||
|
"payed_for": [1, 3],
|
||||||
|
"amount": "200",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"payer": 3,
|
||||||
|
"payed_for": [2],
|
||||||
|
"amount": "13.33",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
# generate json export of bills
|
||||||
|
resp = self.client.get("/raclette/export/bills.json")
|
||||||
|
expected = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"amount": 200.0,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage \xe0 raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "p\xe9p\xe9"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
||||||
|
|
||||||
|
# generate csv export of bills
|
||||||
|
resp = self.client.get("/raclette/export/bills.csv")
|
||||||
|
expected = [
|
||||||
|
"date,what,amount,currency,payer_name,payer_weight,owers",
|
||||||
|
"2017-01-01,refund,XXX,13.33,tata,1.0,fred",
|
||||||
|
'2016-12-31,red wine,XXX,200.0,fred,1.0,"zorglub, tata"',
|
||||||
|
'2016-12-31,fromage à raclette,10.0,XXX,zorglub,2.0,"zorglub, fred, tata, pépé"',
|
||||||
|
]
|
||||||
|
received_lines = resp.data.decode("utf-8").split("\n")
|
||||||
|
|
||||||
|
for i, line in enumerate(expected):
|
||||||
|
self.assertEqual(
|
||||||
|
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
||||||
|
)
|
||||||
|
|
||||||
|
# generate json export of transactions
|
||||||
|
resp = self.client.get("/raclette/export/transactions.json")
|
||||||
|
expected = [
|
||||||
|
{
|
||||||
|
"amount": 2.00,
|
||||||
|
"currency": "XXX",
|
||||||
|
"receiver": "fred",
|
||||||
|
"ower": "p\xe9p\xe9",
|
||||||
|
},
|
||||||
|
{"amount": 55.34, "currency": "XXX", "receiver": "fred", "ower": "tata"},
|
||||||
|
{
|
||||||
|
"amount": 127.33,
|
||||||
|
"currency": "XXX",
|
||||||
|
"receiver": "fred",
|
||||||
|
"ower": "zorglub",
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
||||||
|
|
||||||
|
# generate csv export of transactions
|
||||||
|
resp = self.client.get("/raclette/export/transactions.csv")
|
||||||
|
|
||||||
|
expected = [
|
||||||
|
"amount,currency,receiver,ower",
|
||||||
|
"2.0,XXX,fred,pépé",
|
||||||
|
"55.34,XXX,fred,tata",
|
||||||
|
"127.33,XXX,fred,zorglub",
|
||||||
|
]
|
||||||
|
received_lines = resp.data.decode("utf-8").split("\n")
|
||||||
|
|
||||||
|
for i, line in enumerate(expected):
|
||||||
|
self.assertEqual(
|
||||||
|
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
||||||
|
)
|
||||||
|
|
||||||
|
# wrong export_format should return a 404
|
||||||
|
resp = self.client.get("/raclette/export/transactions.wrong")
|
||||||
|
self.assertEqual(resp.status_code, 404)
|
||||||
|
|
||||||
|
def test_export_with_currencies(self):
|
||||||
|
self.post_project("raclette", default_currency="EUR")
|
||||||
|
|
||||||
|
# add participants
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "fred"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "tata"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "pépé"})
|
||||||
|
|
||||||
|
# create bills
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage à raclette",
|
||||||
|
"payer": 1,
|
||||||
|
"payed_for": [1, 2, 3, 4],
|
||||||
|
"amount": "10.0",
|
||||||
|
"original_currency": "EUR",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "poutine from Québec",
|
||||||
|
"payer": 2,
|
||||||
|
"payed_for": [1, 3],
|
||||||
|
"amount": "100",
|
||||||
|
"original_currency": "CAD",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"payer": 3,
|
||||||
|
"payed_for": [2],
|
||||||
|
"amount": "13.33",
|
||||||
|
"original_currency": "EUR",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
# generate json export of bills
|
||||||
|
resp = self.client.get("/raclette/export/bills.json")
|
||||||
|
expected = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "poutine from Qu\xe9bec",
|
||||||
|
"amount": 100.0,
|
||||||
|
"currency": "CAD",
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage \xe0 raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "p\xe9p\xe9"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
||||||
|
|
||||||
|
# generate csv export of bills
|
||||||
|
resp = self.client.get("/raclette/export/bills.csv")
|
||||||
|
expected = [
|
||||||
|
"date,what,amount,currency,payer_name,payer_weight,owers",
|
||||||
|
"2017-01-01,refund,13.33,EUR,tata,1.0,fred",
|
||||||
|
'2016-12-31,poutine from Québec,100.0,CAD,fred,1.0,"zorglub, tata"',
|
||||||
|
'2016-12-31,fromage à raclette,10.0,EUR,zorglub,2.0,"zorglub, fred, tata, pépé"',
|
||||||
|
]
|
||||||
|
received_lines = resp.data.decode("utf-8").split("\n")
|
||||||
|
|
||||||
|
for i, line in enumerate(expected):
|
||||||
|
self.assertEqual(
|
||||||
|
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
||||||
|
)
|
||||||
|
|
||||||
|
# generate json export of transactions (in EUR!)
|
||||||
|
resp = self.client.get("/raclette/export/transactions.json")
|
||||||
|
expected = [
|
||||||
|
{
|
||||||
|
"amount": 2.00,
|
||||||
|
"currency": "EUR",
|
||||||
|
"receiver": "fred",
|
||||||
|
"ower": "p\xe9p\xe9",
|
||||||
|
},
|
||||||
|
{"amount": 10.89, "currency": "EUR", "receiver": "fred", "ower": "tata"},
|
||||||
|
{"amount": 38.45, "currency": "EUR", "receiver": "fred", "ower": "zorglub"},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
||||||
|
|
||||||
|
# generate csv export of transactions
|
||||||
|
resp = self.client.get("/raclette/export/transactions.csv")
|
||||||
|
|
||||||
|
expected = [
|
||||||
|
"amount,currency,receiver,ower",
|
||||||
|
"2.0,EUR,fred,pépé",
|
||||||
|
"10.89,EUR,fred,tata",
|
||||||
|
"38.45,EUR,fred,zorglub",
|
||||||
|
]
|
||||||
|
received_lines = resp.data.decode("utf-8").split("\n")
|
||||||
|
|
||||||
|
for i, line in enumerate(expected):
|
||||||
|
self.assertEqual(
|
||||||
|
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
||||||
|
)
|
||||||
|
|
||||||
|
# Change project currency to CAD
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
project.switch_currency("CAD")
|
||||||
|
|
||||||
|
# generate json export of transactions (now in CAD!)
|
||||||
|
resp = self.client.get("/raclette/export/transactions.json")
|
||||||
|
expected = [
|
||||||
|
{
|
||||||
|
"amount": 3.00,
|
||||||
|
"currency": "CAD",
|
||||||
|
"receiver": "fred",
|
||||||
|
"ower": "p\xe9p\xe9",
|
||||||
|
},
|
||||||
|
{"amount": 16.34, "currency": "CAD", "receiver": "fred", "ower": "tata"},
|
||||||
|
{"amount": 57.67, "currency": "CAD", "receiver": "fred", "ower": "zorglub"},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.assertEqual(json.loads(resp.data.decode("utf-8")), expected)
|
||||||
|
|
||||||
|
# generate csv export of transactions
|
||||||
|
resp = self.client.get("/raclette/export/transactions.csv")
|
||||||
|
|
||||||
|
expected = [
|
||||||
|
"amount,currency,receiver,ower",
|
||||||
|
"3.0,CAD,fred,pépé",
|
||||||
|
"16.34,CAD,fred,tata",
|
||||||
|
"57.67,CAD,fred,zorglub",
|
||||||
|
]
|
||||||
|
received_lines = resp.data.decode("utf-8").split("\n")
|
||||||
|
|
||||||
|
for i, line in enumerate(expected):
|
||||||
|
self.assertEqual(
|
||||||
|
set(line.split(",")), set(received_lines[i].strip("\r").split(","))
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_import_currencies_in_empty_project_with_currency(self):
|
||||||
|
# Import JSON with currencies in an empty project with a default currency
|
||||||
|
|
||||||
|
self.post_project("raclette", default_currency="EUR")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "poutine from québec",
|
||||||
|
"amount": 50.0,
|
||||||
|
"currency": "CAD",
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.import_project("raclette", json_to_import)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check if all bills have been added
|
||||||
|
self.assertEqual(len(bills), len(json_to_import))
|
||||||
|
|
||||||
|
# Check if name of bills are ok
|
||||||
|
b = [e["what"] for e in bills]
|
||||||
|
b.sort()
|
||||||
|
ref = [e["what"] for e in json_to_import]
|
||||||
|
ref.sort()
|
||||||
|
|
||||||
|
self.assertEqual(b, ref)
|
||||||
|
|
||||||
|
# Check if other informations in bill are ok
|
||||||
|
for j in json_to_import:
|
||||||
|
for b in bills:
|
||||||
|
if b["what"] == j["what"]:
|
||||||
|
self.assertEqual(b["payer_name"], j["payer_name"])
|
||||||
|
self.assertEqual(b["amount"], j["amount"])
|
||||||
|
self.assertEqual(b["currency"], j["currency"])
|
||||||
|
self.assertEqual(b["payer_weight"], j["payer_weight"])
|
||||||
|
self.assertEqual(b["date"], j["date"])
|
||||||
|
|
||||||
|
list_project = [ower for ower in b["owers"]]
|
||||||
|
list_project.sort()
|
||||||
|
list_json = [ower for ower in j["owers"]]
|
||||||
|
list_json.sort()
|
||||||
|
|
||||||
|
self.assertEqual(list_project, list_json)
|
||||||
|
|
||||||
|
def test_import_single_currency_in_empty_project_without_currency(self):
|
||||||
|
# Import JSON with a single currency in an empty project with no
|
||||||
|
# default currency. It should work by stripping the currency from
|
||||||
|
# bills.
|
||||||
|
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.import_project("raclette", json_to_import)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check if all bills have been added
|
||||||
|
self.assertEqual(len(bills), len(json_to_import))
|
||||||
|
|
||||||
|
# Check if name of bills are ok
|
||||||
|
b = [e["what"] for e in bills]
|
||||||
|
b.sort()
|
||||||
|
ref = [e["what"] for e in json_to_import]
|
||||||
|
ref.sort()
|
||||||
|
|
||||||
|
self.assertEqual(b, ref)
|
||||||
|
|
||||||
|
# Check if other informations in bill are ok
|
||||||
|
for i in json_to_import:
|
||||||
|
for j in bills:
|
||||||
|
if j["what"] == i["what"]:
|
||||||
|
self.assertEqual(j["payer_name"], i["payer_name"])
|
||||||
|
self.assertEqual(j["amount"], i["amount"])
|
||||||
|
# Currency should have been stripped
|
||||||
|
self.assertEqual(j["currency"], "XXX")
|
||||||
|
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
||||||
|
self.assertEqual(j["date"], i["date"])
|
||||||
|
|
||||||
|
list_project = [ower for ower in j["owers"]]
|
||||||
|
list_project.sort()
|
||||||
|
list_json = [ower for ower in i["owers"]]
|
||||||
|
list_json.sort()
|
||||||
|
|
||||||
|
self.assertEqual(list_project, list_json)
|
||||||
|
|
||||||
|
def test_import_multiple_currencies_in_empty_project_without_currency(self):
|
||||||
|
# Import JSON with multiple currencies in an empty project with no
|
||||||
|
# default currency. It should fail.
|
||||||
|
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "poutine from québec",
|
||||||
|
"amount": 50.0,
|
||||||
|
"currency": "CAD",
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "EUR",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
# Import should fail
|
||||||
|
self.import_project("raclette", json_to_import, 400)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check that there are no bills
|
||||||
|
self.assertEqual(len(bills), 0)
|
||||||
|
|
||||||
|
def test_import_no_currency_in_empty_project_with_currency(self):
|
||||||
|
# Import JSON without currencies (from ihatemoney < 5) in an empty
|
||||||
|
# project with a default currency.
|
||||||
|
|
||||||
|
self.post_project("raclette", default_currency="EUR")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"amount": 200.0,
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.import_project("raclette", json_to_import)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check if all bills have been added
|
||||||
|
self.assertEqual(len(bills), len(json_to_import))
|
||||||
|
|
||||||
|
# Check if name of bills are ok
|
||||||
|
b = [e["what"] for e in bills]
|
||||||
|
b.sort()
|
||||||
|
ref = [e["what"] for e in json_to_import]
|
||||||
|
ref.sort()
|
||||||
|
|
||||||
|
self.assertEqual(b, ref)
|
||||||
|
|
||||||
|
# Check if other informations in bill are ok
|
||||||
|
for i in json_to_import:
|
||||||
|
for j in bills:
|
||||||
|
if j["what"] == i["what"]:
|
||||||
|
self.assertEqual(j["payer_name"], i["payer_name"])
|
||||||
|
self.assertEqual(j["amount"], i["amount"])
|
||||||
|
# All bills are converted to default project currency
|
||||||
|
self.assertEqual(j["currency"], "EUR")
|
||||||
|
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
||||||
|
self.assertEqual(j["date"], i["date"])
|
||||||
|
|
||||||
|
list_project = [ower for ower in j["owers"]]
|
||||||
|
list_project.sort()
|
||||||
|
list_json = [ower for ower in i["owers"]]
|
||||||
|
list_json.sort()
|
||||||
|
|
||||||
|
self.assertEqual(list_project, list_json)
|
||||||
|
|
||||||
|
def test_import_no_currency_in_empty_project_without_currency(self):
|
||||||
|
# Import JSON without currencies (from ihatemoney < 5) in an empty
|
||||||
|
# project with no default currency.
|
||||||
|
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"amount": 200.0,
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.import_project("raclette", json_to_import)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check if all bills have been added
|
||||||
|
self.assertEqual(len(bills), len(json_to_import))
|
||||||
|
|
||||||
|
# Check if name of bills are ok
|
||||||
|
b = [e["what"] for e in bills]
|
||||||
|
b.sort()
|
||||||
|
ref = [e["what"] for e in json_to_import]
|
||||||
|
ref.sort()
|
||||||
|
|
||||||
|
self.assertEqual(b, ref)
|
||||||
|
|
||||||
|
# Check if other informations in bill are ok
|
||||||
|
for i in json_to_import:
|
||||||
|
for j in bills:
|
||||||
|
if j["what"] == i["what"]:
|
||||||
|
self.assertEqual(j["payer_name"], i["payer_name"])
|
||||||
|
self.assertEqual(j["amount"], i["amount"])
|
||||||
|
self.assertEqual(j["currency"], "XXX")
|
||||||
|
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
||||||
|
self.assertEqual(j["date"], i["date"])
|
||||||
|
|
||||||
|
list_project = [ower for ower in j["owers"]]
|
||||||
|
list_project.sort()
|
||||||
|
list_json = [ower for ower in i["owers"]]
|
||||||
|
list_json.sort()
|
||||||
|
|
||||||
|
self.assertEqual(list_project, list_json)
|
||||||
|
|
||||||
|
def test_import_partial_project(self):
|
||||||
|
# Import a JSON in a project with already existing data
|
||||||
|
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.login("raclette")
|
||||||
|
|
||||||
|
project = self.get_project("raclette")
|
||||||
|
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "zorglub", "weight": 2})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "fred"})
|
||||||
|
self.client.post("/raclette/members/add", data={"name": "tata"})
|
||||||
|
self.client.post(
|
||||||
|
"/raclette/add",
|
||||||
|
data={
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"payer": 2,
|
||||||
|
"payed_for": [1, 3],
|
||||||
|
"amount": "200",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
json_to_import = [
|
||||||
|
{
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"amount": 13.33,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
},
|
||||||
|
{ # This expense does not have to be present twice.
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "red wine",
|
||||||
|
"amount": 200.0,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "fred",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["zorglub", "tata"],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2016-12-31",
|
||||||
|
"what": "fromage a raclette",
|
||||||
|
"amount": 10.0,
|
||||||
|
"currency": "XXX",
|
||||||
|
"payer_name": "zorglub",
|
||||||
|
"payer_weight": 2.0,
|
||||||
|
"owers": ["zorglub", "fred", "tata", "pepe"],
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
self.import_project("raclette", json_to_import)
|
||||||
|
|
||||||
|
bills = project.get_pretty_bills()
|
||||||
|
|
||||||
|
# Check if all bills have been added
|
||||||
|
self.assertEqual(len(bills), len(json_to_import))
|
||||||
|
|
||||||
|
# Check if name of bills are ok
|
||||||
|
b = [e["what"] for e in bills]
|
||||||
|
b.sort()
|
||||||
|
ref = [e["what"] for e in json_to_import]
|
||||||
|
ref.sort()
|
||||||
|
|
||||||
|
self.assertEqual(b, ref)
|
||||||
|
|
||||||
|
# Check if other informations in bill are ok
|
||||||
|
for i in json_to_import:
|
||||||
|
for j in bills:
|
||||||
|
if j["what"] == i["what"]:
|
||||||
|
self.assertEqual(j["payer_name"], i["payer_name"])
|
||||||
|
self.assertEqual(j["amount"], i["amount"])
|
||||||
|
self.assertEqual(j["currency"], i["currency"])
|
||||||
|
self.assertEqual(j["payer_weight"], i["payer_weight"])
|
||||||
|
self.assertEqual(j["date"], i["date"])
|
||||||
|
|
||||||
|
list_project = [ower for ower in j["owers"]]
|
||||||
|
list_project.sort()
|
||||||
|
list_json = [ower for ower in i["owers"]]
|
||||||
|
list_json.sort()
|
||||||
|
|
||||||
|
self.assertEqual(list_project, list_json)
|
||||||
|
|
||||||
|
def test_import_wrong_json(self):
|
||||||
|
self.post_project("raclette")
|
||||||
|
self.login("raclette")
|
||||||
|
self.get_project("raclette")
|
||||||
|
|
||||||
|
json_1 = [
|
||||||
|
{ # wrong keys
|
||||||
|
"checked": False,
|
||||||
|
"dimensions": {"width": 5, "height": 10},
|
||||||
|
"id": 1,
|
||||||
|
"name": "A green door",
|
||||||
|
"price": 12.5,
|
||||||
|
"tags": ["home", "green"],
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
|
json_2 = [
|
||||||
|
{ # amount missing
|
||||||
|
"date": "2017-01-01",
|
||||||
|
"what": "refund",
|
||||||
|
"payer_name": "tata",
|
||||||
|
"payer_weight": 1.0,
|
||||||
|
"owers": ["fred"],
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
|
for json_to_import in [json_1, json_2]:
|
||||||
|
# Import should fail
|
||||||
|
self.import_project("raclette", json_to_import, 400)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
Loading…
Reference in a new issue