Fix two tests

This commit is contained in:
zandercymatics 2024-08-21 10:15:49 -06:00
parent cbec3f2ed7
commit 5c07775277
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 28 additions and 16 deletions

View file

@ -1459,9 +1459,15 @@ 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:
portfolio_permission.clean()
invalid_permission.clean()
self.assertEqual(
cm.exception.message, "When portfolio roles or additional permissions are assigned, portfolio is required."