-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test(itim): Ticket Linked Item API field checks
- Loading branch information
Showing
1 changed file
with
225 additions
and
0 deletions.
There are no files selected for viewing
225 changes: 225 additions & 0 deletions
225
app/core/tests/unit/ticket_linked_item/test_ticket_linked_item_viewset.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,225 @@ | ||
from django.contrib.auth.models import User | ||
from django.contrib.contenttypes.models import ContentType | ||
from django.shortcuts import reverse | ||
from django.test import Client, TestCase | ||
|
||
from access.models import Organization, Team, TeamUsers, Permission | ||
|
||
from api.tests.abstract.api_permissions_viewset import ( | ||
APIPermissionAdd, | ||
APIPermissionChange, | ||
APIPermissionDelete, | ||
APIPermissionView | ||
) | ||
|
||
from core.models.ticket.ticket_linked_items import Ticket, TicketLinkedItem | ||
|
||
from itam.models.device import Device | ||
|
||
from settings.models.user_settings import UserSettings | ||
|
||
|
||
|
||
class TicketLinkedItemPermissionsAPI( | ||
APIPermissionAdd, | ||
APIPermissionDelete, | ||
APIPermissionView, | ||
TestCase | ||
): | ||
""" Test Cases common to ALL ticket types """ | ||
|
||
model = TicketLinkedItem | ||
|
||
app_namespace = 'v2' | ||
|
||
delete_data = {} | ||
|
||
ticket_type: str = 'request' | ||
|
||
ticket_type_enum = Ticket.TicketType.REQUEST | ||
|
||
url_name = '_api_v2_ticket_linked_item' | ||
|
||
|
||
@classmethod | ||
def setUpTestData(self): | ||
"""Setup Test | ||
1. Create an organization for user and item | ||
. create an organization that is different to item | ||
2. Create a team | ||
3. create teams with each permission: view, add, change, delete | ||
4. create a user per team | ||
""" | ||
|
||
|
||
|
||
organization = Organization.objects.create(name='test_org') | ||
|
||
self.organization = organization | ||
|
||
different_organization = Organization.objects.create(name='test_different_organization') | ||
|
||
|
||
view_permissions = Permission.objects.get( | ||
codename = 'view_' + self.model._meta.model_name, | ||
content_type = ContentType.objects.get( | ||
app_label = self.model._meta.app_label, | ||
model = self.model._meta.model_name, | ||
) | ||
) | ||
|
||
view_team = Team.objects.create( | ||
team_name = 'view_team', | ||
organization = organization, | ||
) | ||
|
||
view_team.permissions.set([view_permissions]) | ||
|
||
|
||
|
||
add_permissions = Permission.objects.get( | ||
codename = 'add_' + self.model._meta.model_name, | ||
content_type = ContentType.objects.get( | ||
app_label = self.model._meta.app_label, | ||
model = self.model._meta.model_name, | ||
) | ||
) | ||
|
||
add_team = Team.objects.create( | ||
team_name = 'add_team', | ||
organization = organization, | ||
) | ||
|
||
add_team.permissions.set([add_permissions]) | ||
|
||
|
||
|
||
change_permissions = Permission.objects.get( | ||
codename = 'change_' + self.model._meta.model_name, | ||
content_type = ContentType.objects.get( | ||
app_label = self.model._meta.app_label, | ||
model = self.model._meta.model_name, | ||
) | ||
) | ||
|
||
change_team = Team.objects.create( | ||
team_name = 'change_team', | ||
organization = organization, | ||
) | ||
|
||
change_team.permissions.set([change_permissions]) | ||
|
||
|
||
|
||
delete_permissions = Permission.objects.get( | ||
codename = 'delete_' + self.model._meta.model_name, | ||
content_type = ContentType.objects.get( | ||
app_label = self.model._meta.app_label, | ||
model = self.model._meta.model_name, | ||
) | ||
) | ||
|
||
delete_team = Team.objects.create( | ||
team_name = 'delete_team', | ||
organization = organization, | ||
) | ||
|
||
delete_team.permissions.set([delete_permissions]) | ||
|
||
|
||
self.no_permissions_user = User.objects.create_user(username="test_no_permissions", password="password") | ||
|
||
|
||
self.view_user = User.objects.create_user(username="test_user_view", password="password") | ||
teamuser = TeamUsers.objects.create( | ||
team = view_team, | ||
user = self.view_user | ||
) | ||
|
||
|
||
self.ticket = Ticket.objects.create( | ||
organization = self.organization, | ||
title = 'one', | ||
description = 'some text for body', | ||
opened_by = self.view_user, | ||
ticket_type = self.ticket_type_enum, | ||
status = Ticket.TicketStatus.All.NEW | ||
) | ||
|
||
self.device = Device.objects.create( | ||
organization = self.organization, | ||
name = 'one', | ||
) | ||
|
||
self.device_two = Device.objects.create( | ||
organization = self.organization, | ||
name = 'two', | ||
) | ||
|
||
self.item = self.model.objects.create( | ||
organization = self.organization, | ||
item = self.device.id, | ||
item_type = TicketLinkedItem.Modules.DEVICE, | ||
ticket = self.ticket, | ||
) | ||
|
||
|
||
self.url_kwargs = {'ticket_id': self.ticket.id} | ||
|
||
self.url_view_kwargs = {'ticket_id': self.ticket.id, 'pk': self.item.id} | ||
|
||
self.add_data = { | ||
'organization': self.organization.id, | ||
'ticket': self.ticket.id, | ||
'item': self.device_two.id, | ||
'item_type': int(TicketLinkedItem.Modules.DEVICE), | ||
} | ||
|
||
|
||
self.add_user = User.objects.create_user(username="test_user_add", password="password") | ||
|
||
user_settings = UserSettings.objects.get(user=self.add_user) | ||
|
||
user_settings.default_organization = self.organization | ||
|
||
user_settings.save() | ||
|
||
|
||
teamuser = TeamUsers.objects.create( | ||
team = add_team, | ||
user = self.add_user | ||
) | ||
|
||
self.change_user = User.objects.create_user(username="test_user_change", password="password") | ||
teamuser = TeamUsers.objects.create( | ||
team = change_team, | ||
user = self.change_user | ||
) | ||
|
||
self.delete_user = User.objects.create_user(username="test_user_delete", password="password") | ||
teamuser = TeamUsers.objects.create( | ||
team = delete_team, | ||
user = self.delete_user | ||
) | ||
|
||
|
||
self.different_organization_user = User.objects.create_user(username="test_different_organization_user", password="password") | ||
|
||
|
||
different_organization_team = Team.objects.create( | ||
team_name = 'different_organization_team', | ||
organization = different_organization, | ||
) | ||
|
||
different_organization_team.permissions.set([ | ||
view_permissions, | ||
add_permissions, | ||
change_permissions, | ||
delete_permissions, | ||
]) | ||
|
||
TeamUsers.objects.create( | ||
team = different_organization_team, | ||
user = self.different_organization_user | ||
) |