Fix asset filter

This commit is contained in:
2023-02-14 07:20:47 +00:00
parent 0a89d96b86
commit b6952767d5
2 changed files with 32 additions and 9 deletions

View File

@@ -14,6 +14,7 @@ class AssetfilterTestCase(TestCase):
user=self.user,
name="Group1",
description="Test group",
when_no_data=6,
)
def test_get_allowed_prohibited(self):
@@ -87,6 +88,18 @@ class AssetfilterTestCase(TestCase):
self.assertTrue(assetfilter.get_allowed(self.group, "EUR", "USD", "long"))
def test_get_allowed_no_data_prohibited(self):
self.group.when_no_data = 7
self.group.save()
self.asset_rule = AssetRule.objects.create(
user=self.user,
group=self.group,
asset="EUR",
status=0,
)
self.asset_rule.save()
self.assertFalse(assetfilter.get_allowed(self.group, "EUR", "USD", "long"))
def test_get_allowed_no_match(self):
self.asset_rule = AssetRule.objects.create(
user=self.user,
@@ -118,6 +131,8 @@ class AssetfilterTestCase(TestCase):
)
self.asset_rule.save()
self.assertTrue(assetfilter.get_allowed(self.group, "EUR", "USD", "long"))
def test_get_allowed_always_allow(self):
self.asset_rule = AssetRule.objects.create(
user=self.user,