Write tests for asset filter

core/tests/trading/test_assetfilter.py

@@ -0,0 +1,28 @@
+from django.test import TestCase
+
+from core.trading.assetfilter import get_allowed
+from core.models import AssetGroup, User
+
+class AssetfilterTestCase(TestCase):
+    def setUp(self):
+        self.user = User.objects.create_user(
+            username="testuser",
+            email="test@example.com",
+        )
+        self.group = AssetGroup.objects.create(
+            user=self.user,
+            name="Group1",
+            description="Test group",
+        )
+
+
+    def test_get_allowed(self):
+        """
+        Test that the asset filter works.
+        """
+        self.group.allowed = {"EUR_USD": True, "EUR_GBP": False}
+        self.assertTrue(get_allowed(self.group, "EUR_USD", "buy"))
+        self.assertFalse(get_allowed(self.group, "EUR_GBP", "sell"))
+
+        # Default true
+        self.assertTrue(get_allowed(self.group, "nonexistent", "sell"))