From c71f4ed04577ac5450be7993e322db74bf054c89 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 27 May 2021 16:48:13 +1000
Subject: [PATCH] Add currency exchange unit tests

---
 InvenTree/InvenTree/tests.py | 51 ++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/InvenTree/InvenTree/tests.py b/InvenTree/InvenTree/tests.py
index d65829cf8e..b7e5b98c1b 100644
--- a/InvenTree/InvenTree/tests.py
+++ b/InvenTree/InvenTree/tests.py
@@ -5,6 +5,12 @@ from django.test import TestCase
 import django.core.exceptions as django_exceptions
 from django.core.exceptions import ValidationError
 
+from django.conf import settings
+
+from djmoney.money import Money
+from djmoney.contrib.exchange.models import Rate, convert_money
+from djmoney.contrib.exchange.exceptions import MissingRate
+
 from .validators import validate_overage, validate_part_name
 from . import helpers
 from . import version
@@ -13,6 +19,8 @@ from mptt.exceptions import InvalidMove
 
 from decimal import Decimal
 
+import InvenTree.tasks
+
 from stock.models import StockLocation
 
 
@@ -308,3 +316,46 @@ class TestVersionNumber(TestCase):
         self.assertTrue(v_c > v_b)
         self.assertTrue(v_d > v_c)
         self.assertTrue(v_d > v_a)
+
+
+class CurrencyTests(TestCase):
+    """
+    Unit tests for currency / exchange rate functionality
+    """
+
+    def test_rates(self):
+
+        # Initially, there will not be any exchange rate information
+        rates = Rate.objects.all()
+
+        self.assertEqual(rates.count(), 0)
+
+        # Without rate information, we cannot convert anything...
+        with self.assertRaises(MissingRate):
+            convert_money(Money(100, 'USD'), 'AUD')
+
+        with self.assertRaises(MissingRate):
+            convert_money(Money(100, 'AUD'), 'USD')
+
+        currencies = settings.CURRENCIES
+
+        InvenTree.tasks.update_exchange_rates()
+
+        rates = Rate.objects.all()
+
+        self.assertEqual(rates.count(), len(currencies))
+
+        # Now that we have some exchange rate information, we can perform conversions
+
+        # Forwards
+        convert_money(Money(100, 'USD'), 'AUD')
+
+        # Backwards
+        convert_money(Money(100, 'AUD'), 'USD')
+
+        # Convert between non base currencies
+        convert_money(Money(100, 'CAD'), 'NZD')
+
+        # Convert to a symbol which is not covered
+        with self.assertRaises(MissingRate):
+            convert_money(Money(100, 'GBP'), 'ZWL')