Fix logic in clean

This commit is contained in:
zandercymatics 2024-08-21 11:43:40 -06:00
parent 5c07775277
commit 32ed84a5e1
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 8 additions and 10 deletions

View file

@ -1459,15 +1459,9 @@ class TestUser(TestCase):
portfolio_permission.portfolio = None
portfolio_permission.roles = [UserPortfolioRoleChoices.ORGANIZATION_ADMIN]
# Create a new UserPortfolioPermission instance without a portfolio
invalid_permission = UserPortfolioPermission(
user=self.user,
roles=[UserPortfolioRoleChoices.ORGANIZATION_ADMIN],
portfolio=None # This should trigger the validation error
)
# Test if the ValidationError is raised with the correct message
with self.assertRaises(ValidationError) as cm:
invalid_permission.clean()
portfolio_permission.clean()
self.assertEqual(
cm.exception.message, "When portfolio roles or additional permissions are assigned, portfolio is required."