Improve revolut tests

This commit is contained in:
Gabriel Augendre 2020-02-26 18:32:31 +01:00
parent bfa815f08d
commit 2ac1139d61
No known key found for this signature in database
GPG key ID: 1E693F4CE4AEE7B4

View file

@ -8,61 +8,66 @@ from ofx_processor.processors.revolut import (
class RevolutProcessorTestCase(unittest.TestCase): class RevolutProcessorTestCase(unittest.TestCase):
def test_process_amount_with_one_decimal_place(self): def test_convert_amount_with_one_decimal_place(self):
amount = "3,4" amount = "3,4"
expected = 3.4 expected = 3.4
self.assertEqual(_amount_str_to_float(amount), expected) self.assertEqual(_amount_str_to_float(amount), expected)
def test_process_amount_with_two_decimal_places(self): def test_convert_amount_with_two_decimal_places(self):
amount = "3,41" amount = "3,41"
expected = 3.41 expected = 3.41
self.assertEqual(_amount_str_to_float(amount), expected) self.assertEqual(_amount_str_to_float(amount), expected)
def test_process_amount_with_empty_string(self): def test_convert_amount_with_empty_string(self):
amount = "" amount = ""
expected = "" expected = ""
self.assertEqual(_amount_str_to_float(amount), expected) self.assertEqual(_amount_str_to_float(amount), expected)
def test_process_memo_with_category_and_rate(self): def test_get_memo_with_category_and_rate(self):
line = {"Category": "category name", "Exchange Rate": "exchange rate"} line = {"Category": "category name", "Exchange Rate": "exchange rate"}
expected = "category name - exchange rate" expected = "category name - exchange rate"
self.assertEqual(RevolutLine(line).get_memo(), expected) self.assertEqual(RevolutLine(line).get_memo(), expected)
def test_process_memo_with_only_category(self): def test_get_memo_with_only_category(self):
line = {"Category": "category name", "Exchange Rate": ""} line = {"Category": "category name", "Exchange Rate": ""}
expected = "category name" expected = "category name"
self.assertEqual(RevolutLine(line).get_memo(), expected) self.assertEqual(RevolutLine(line).get_memo(), expected)
def test_process_memo_with_only_rate(self): def test_get_memo_with_only_rate(self):
line = {"Category": "", "Exchange Rate": "exchange rate"} line = {"Category": "", "Exchange Rate": "exchange rate"}
expected = "exchange rate" expected = "exchange rate"
self.assertEqual(RevolutLine(line).get_memo(), expected) self.assertEqual(RevolutLine(line).get_memo(), expected)
def test_process_memo_with_missing_keys(self): def test_get_memo_with_missing_keys(self):
line = {"Category": "category name"} line = {"Category": "category name"}
expected = "category name" expected = "category name"
self.assertEqual(RevolutLine(line).get_memo(), expected) self.assertEqual(RevolutLine(line).get_memo(), expected)
def test_process_date(self): def test_get_date(self):
line = {"Completed Date": "January 16"} line = {"Completed Date": "January 16"}
current_year = datetime.date.today().year current_year = datetime.date.today().year
expected = f"{current_year}-01-16" expected = f"{current_year}-01-16"
self.assertEqual(RevolutLine(line).get_date(), expected) self.assertEqual(RevolutLine(line).get_date(), expected)
def test_process_date_other_year(self): def test_get_date_other_year(self):
line = {"Completed Date": "January 16 2019"} line = {"Completed Date": "January 16 2019"}
expected = f"2019-01-16" expected = f"2019-01-16"
self.assertEqual(RevolutLine(line).get_date(), expected) self.assertEqual(RevolutLine(line).get_date(), expected)
def test_process_inflow(self): def test_get_amount_inflow(self):
line = {"Paid In (EUR)": "3,42"} line = {"Paid In (EUR)": "3,42"}
expected = 3.42 expected = 3420
self.assertEqual(RevolutLine(line)._process_inflow(), expected) self.assertEqual(RevolutLine(line).get_amount(), expected)
def test_process_outflow(self): def test_get_amount_outflow(self):
line = {"Paid Out (EUR)": "8,42"} line = {"Paid Out (EUR)": "8,42"}
expected = 8.42 expected = -8420
self.assertEqual(RevolutLine(line)._process_outflow(), expected) self.assertEqual(RevolutLine(line).get_amount(), expected)
def test_get_payee(self):
line = {"Reference": "Hello world"}
expected = line["Reference"]
self.assertEqual(RevolutLine(line).get_payee(), expected)
if __name__ == "__main__": if __name__ == "__main__":