336 lines
17 KiB
Python
336 lines
17 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from django.test import TestCase, Client
|
|
|
|
from authentication.models import ExtendedUser
|
|
from mail.models import Email, EmailAttachment
|
|
from tickets.models import IssueThread, StateChange, Comment
|
|
from django.contrib.auth.models import Permission
|
|
from knox.models import AuthToken
|
|
|
|
from base64 import b64encode
|
|
|
|
|
|
class IssueApiTest(TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.user = ExtendedUser.objects.create_user('testuser', 'test', 'test')
|
|
self.user.user_permissions.add(*Permission.objects.all())
|
|
self.user.save()
|
|
self.token = AuthToken.objects.create(user=self.user)
|
|
self.client = Client(headers={'Authorization': 'Token ' + self.token[1]})
|
|
|
|
def test_issues_empty(self):
|
|
response = self.client.get('/api/2/tickets/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|
|
|
|
def test_issues(self):
|
|
now = datetime.now()
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
mail1 = Email.objects.create(
|
|
subject='test',
|
|
body='test',
|
|
sender='test',
|
|
recipient='test',
|
|
issue_thread=issue,
|
|
timestamp=now,
|
|
)
|
|
mail2 = Email.objects.create(
|
|
subject='test',
|
|
body='test',
|
|
sender='test',
|
|
recipient='test',
|
|
issue_thread=issue,
|
|
in_reply_to=mail1.reference,
|
|
timestamp=now + timedelta(seconds=2),
|
|
)
|
|
comment = Comment.objects.create(
|
|
issue_thread=issue,
|
|
comment="test",
|
|
timestamp=now + timedelta(seconds=3),
|
|
)
|
|
self.assertEqual('pending_new', issue.state)
|
|
self.assertEqual('test issue', issue.name)
|
|
self.assertEqual(None, issue.assigned_to)
|
|
self.assertEqual(36, len(issue.uuid))
|
|
response = self.client.get('/api/2/tickets/')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.json()), 1)
|
|
self.assertEqual(response.json()[0]['id'], issue.id)
|
|
self.assertEqual(response.json()[0]['name'], "test issue")
|
|
self.assertEqual(response.json()[0]['state'], "pending_new")
|
|
self.assertEqual(response.json()[0]['assigned_to'], None)
|
|
self.assertEqual(response.json()[0]['uuid'], issue.uuid)
|
|
self.assertEqual(response.json()[0]['last_activity'], comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
|
self.assertEqual(len(response.json()[0]['timeline']), 4)
|
|
self.assertEqual(response.json()[0]['timeline'][0]['type'], 'state')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['type'], 'mail')
|
|
self.assertEqual(response.json()[0]['timeline'][2]['type'], 'mail')
|
|
self.assertEqual(response.json()[0]['timeline'][3]['type'], 'comment')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['id'], mail1.id)
|
|
self.assertEqual(response.json()[0]['timeline'][2]['id'], mail2.id)
|
|
self.assertEqual(response.json()[0]['timeline'][3]['id'], comment.id)
|
|
self.assertEqual(response.json()[0]['timeline'][0]['state'], 'pending_new')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['sender'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['recipient'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['subject'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['body'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][1]['timestamp'],
|
|
mail1.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
|
self.assertEqual(response.json()[0]['timeline'][2]['sender'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][2]['recipient'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][2]['subject'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][2]['body'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][2]['timestamp'],
|
|
mail2.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
|
self.assertEqual(response.json()[0]['timeline'][3]['comment'], 'test')
|
|
self.assertEqual(response.json()[0]['timeline'][3]['timestamp'],
|
|
comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
|
|
|
|
def test_issues_incomplete_timeline(self):
|
|
now = datetime.now()
|
|
issue1 = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
issue2 = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
issue3 = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
mail1 = Email.objects.create(
|
|
subject='test',
|
|
body='test',
|
|
sender='test',
|
|
recipient='test',
|
|
issue_thread=issue2,
|
|
timestamp=now + timedelta(seconds=2),
|
|
)
|
|
comment = Comment.objects.create(
|
|
issue_thread=issue3,
|
|
comment="test",
|
|
timestamp=now + timedelta(seconds=3),
|
|
)
|
|
|
|
response = self.client.get('/api/2/tickets/')
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertEqual(3, len(response.json()))
|
|
self.assertEqual(issue1.id, response.json()[0]['id'])
|
|
self.assertEqual(issue2.id, response.json()[1]['id'])
|
|
self.assertEqual(issue3.id, response.json()[2]['id'])
|
|
self.assertEqual(issue1.state_changes.first().timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[0]['last_activity'])
|
|
self.assertEqual(mail1.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[1]['last_activity'])
|
|
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[2]['last_activity'])
|
|
self.assertEqual(1, len(response.json()[0]['timeline']))
|
|
self.assertEqual(2, len(response.json()[1]['timeline']))
|
|
self.assertEqual(2, len(response.json()[2]['timeline']))
|
|
self.assertEqual('state', response.json()[0]['timeline'][0]['type'])
|
|
self.assertEqual('state', response.json()[1]['timeline'][0]['type'])
|
|
self.assertEqual('mail', response.json()[1]['timeline'][1]['type'])
|
|
self.assertEqual('state', response.json()[2]['timeline'][0]['type'])
|
|
self.assertEqual('comment', response.json()[2]['timeline'][1]['type'])
|
|
self.assertEqual('pending_new', response.json()[0]['timeline'][0]['state'])
|
|
self.assertEqual('pending_new', response.json()[1]['timeline'][0]['state'])
|
|
self.assertEqual('test', response.json()[1]['timeline'][1]['sender'])
|
|
self.assertEqual('test', response.json()[1]['timeline'][1]['recipient'])
|
|
self.assertEqual('test', response.json()[1]['timeline'][1]['subject'])
|
|
self.assertEqual('test', response.json()[1]['timeline'][1]['body'])
|
|
self.assertEqual(mail1.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[1]['timeline'][1]['timestamp'])
|
|
self.assertEqual('pending_new', response.json()[2]['timeline'][0]['state'])
|
|
self.assertEqual('test', response.json()[2]['timeline'][1]['comment'])
|
|
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[2]['timeline'][1]['timestamp'])
|
|
|
|
def test_issues_with_files(self):
|
|
from django.core.files.base import ContentFile
|
|
from hashlib import sha256
|
|
now = datetime.now()
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
mail1 = Email.objects.create(
|
|
subject='test',
|
|
body='test',
|
|
sender='test',
|
|
recipient='test',
|
|
issue_thread=issue,
|
|
timestamp=now,
|
|
)
|
|
mail2 = Email.objects.create(
|
|
subject='test',
|
|
body='test',
|
|
sender='test',
|
|
recipient='test',
|
|
issue_thread=issue,
|
|
in_reply_to=mail1.reference,
|
|
timestamp=now + timedelta(seconds=2),
|
|
)
|
|
comment = Comment.objects.create(
|
|
issue_thread=issue,
|
|
comment="test",
|
|
timestamp=now + timedelta(seconds=3),
|
|
)
|
|
file1 = EmailAttachment.objects.create(
|
|
name='file1', mime_type='text/plain', file=ContentFile(b"foo1", "f1"),
|
|
hash=sha256(b"foo1").hexdigest(), email=mail1
|
|
)
|
|
file2 = EmailAttachment.objects.create(
|
|
name='file2', mime_type='text/plain', file=ContentFile(b"foo2", "f2"),
|
|
hash=sha256(b"foo2").hexdigest(), email=mail1
|
|
)
|
|
|
|
response = self.client.get('/api/2/tickets/')
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertEqual(1, len(response.json()))
|
|
self.assertEqual(issue.id, response.json()[0]['id'])
|
|
self.assertEqual('pending_new', response.json()[0]['state'])
|
|
self.assertEqual('test issue', response.json()[0]['name'])
|
|
self.assertEqual(None, response.json()[0]['assigned_to'])
|
|
self.assertEqual(36, len(response.json()[0]['uuid']))
|
|
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[0]['last_activity'])
|
|
self.assertEqual(4, len(response.json()[0]['timeline']))
|
|
self.assertEqual('state', response.json()[0]['timeline'][0]['type'])
|
|
self.assertEqual('mail', response.json()[0]['timeline'][1]['type'])
|
|
self.assertEqual('mail', response.json()[0]['timeline'][2]['type'])
|
|
self.assertEqual('comment', response.json()[0]['timeline'][3]['type'])
|
|
self.assertEqual(mail1.id, response.json()[0]['timeline'][1]['id'])
|
|
self.assertEqual(mail2.id, response.json()[0]['timeline'][2]['id'])
|
|
self.assertEqual(comment.id, response.json()[0]['timeline'][3]['id'])
|
|
self.assertEqual('pending_new', response.json()[0]['timeline'][0]['state'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][1]['sender'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][1]['recipient'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][1]['subject'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][1]['body'])
|
|
self.assertEqual(mail1.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[0]['timeline'][1]['timestamp'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][2]['sender'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][2]['recipient'])
|
|
|
|
self.assertEqual('test', response.json()[0]['timeline'][2]['subject'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][2]['body'])
|
|
self.assertEqual(mail2.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[0]['timeline'][2]['timestamp'])
|
|
self.assertEqual('test', response.json()[0]['timeline'][3]['comment'])
|
|
self.assertEqual(comment.timestamp.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
|
|
response.json()[0]['timeline'][3]['timestamp'])
|
|
self.assertEqual(2, len(response.json()[0]['timeline'][1]['attachments']))
|
|
self.assertEqual(0, len(response.json()[0]['timeline'][2]['attachments']))
|
|
self.assertEqual('file1', response.json()[0]['timeline'][1]['attachments'][0]['name'])
|
|
self.assertEqual('file2', response.json()[0]['timeline'][1]['attachments'][1]['name'])
|
|
self.assertEqual('text/plain', response.json()[0]['timeline'][1]['attachments'][0]['mime_type'])
|
|
self.assertEqual('text/plain', response.json()[0]['timeline'][1]['attachments'][1]['mime_type'])
|
|
self.assertEqual(file1.hash, response.json()[0]['timeline'][1]['attachments'][0]['hash'])
|
|
self.assertEqual(file2.hash, response.json()[0]['timeline'][1]['attachments'][1]['hash'])
|
|
|
|
def test_manual_creation(self):
|
|
response = self.client.post('/api/2/evt/tickets/manual/',
|
|
{'name': 'test issue', 'sender': 'test', 'recipient': 'test', 'body': 'test'},
|
|
content_type='application/json')
|
|
self.assertEqual(response.status_code, 201)
|
|
self.assertEqual(response.json()['state'], 'pending_new')
|
|
self.assertEqual(response.json()['name'], 'test issue')
|
|
self.assertEqual(response.json()['assigned_to'], None)
|
|
timeline = response.json()['timeline']
|
|
self.assertEqual(len(timeline), 2)
|
|
self.assertEqual(timeline[0]['type'], 'state')
|
|
self.assertEqual(timeline[0]['state'], 'pending_new')
|
|
self.assertEqual(timeline[1]['type'], 'mail')
|
|
self.assertEqual(timeline[1]['sender'], 'test')
|
|
self.assertEqual(timeline[1]['recipient'], 'test')
|
|
self.assertEqual(timeline[1]['subject'], 'test issue')
|
|
self.assertEqual(timeline[1]['body'], 'test')
|
|
|
|
# def test_post_comment(self):
|
|
# issue = IssueThread.objects.create(
|
|
# name="test issue",
|
|
# )
|
|
# response = self.client.post('/api/2/comments/', {'comment': 'test', 'issue_thread': issue.id})
|
|
# self.assertEqual(response.status_code, 201)
|
|
# self.assertEqual(response.json()['comment'], 'test')
|
|
# self.assertEqual(response.json()['issue_thread'], issue.id)
|
|
# self.assertEqual(response.json()['timestamp'], response.json()['timestamp'])
|
|
|
|
def test_post_comment_altenative(self):
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
response = self.client.post(f'/api/2/tickets/{issue.id}/comment/', {'comment': 'test'})
|
|
self.assertEqual(response.status_code, 201)
|
|
self.assertEqual(response.json()['comment'], 'test')
|
|
self.assertEqual(response.json()['issue_thread'], issue.id)
|
|
self.assertEqual(response.json()['timestamp'], response.json()['timestamp'])
|
|
|
|
def test_post_alt_comment_empty(self):
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
response = self.client.post(f'/api/2/tickets/{issue.id}/comment/', {'comment': ''})
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_state_change(self):
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
response = self.client.patch(f'/api/2/tickets/{issue.id}/', {'state': 'pending_open'},
|
|
content_type='application/json')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json()['state'], 'pending_open')
|
|
self.assertEqual(response.json()['name'], 'test issue')
|
|
self.assertEqual(response.json()['assigned_to'], None)
|
|
timeline = response.json()['timeline']
|
|
self.assertEqual(len(timeline), 2)
|
|
self.assertEqual(timeline[0]['type'], 'state')
|
|
self.assertEqual(timeline[0]['state'], 'pending_new')
|
|
self.assertEqual(timeline[1]['type'], 'state')
|
|
self.assertEqual(timeline[1]['state'], 'pending_open')
|
|
|
|
def test_state_change_invalid_state(self):
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
response = self.client.patch(f'/api/2/tickets/{issue.id}/', {'state': 'invalid'},
|
|
content_type='application/json')
|
|
self.assertEqual(400, response.status_code)
|
|
|
|
def test_assign_user(self):
|
|
issue = IssueThread.objects.create(
|
|
name="test issue",
|
|
)
|
|
response = self.client.patch(f'/api/2/tickets/{issue.id}/', {'assigned_to': self.user.username},
|
|
content_type='application/json')
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertEqual('pending_new', response.json()['state'])
|
|
self.assertEqual('test issue', response.json()['name'])
|
|
self.assertEqual(self.user.username, response.json()['assigned_to'])
|
|
timeline = response.json()['timeline']
|
|
self.assertEqual(2, len(timeline))
|
|
self.assertEqual('state', timeline[0]['type'])
|
|
self.assertEqual('pending_new', timeline[0]['state'])
|
|
self.assertEqual('assignment', timeline[1]['type'])
|
|
self.assertEqual(self.user.username, timeline[1]['assigned_to'])
|
|
|
|
|
|
class IssueSearchTest(TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.user = ExtendedUser.objects.create_user('testuser', 'test', 'test')
|
|
self.user.user_permissions.add(*Permission.objects.all())
|
|
self.user.save()
|
|
self.token = AuthToken.objects.create(user=self.user)
|
|
self.client = Client(headers={'Authorization': 'Token ' + self.token[1]})
|
|
|
|
def test_search(self):
|
|
search_query = b64encode(b'abc').decode('utf-8')
|
|
response = self.client.get(f'/api/2/evt/tickets/{search_query}/')
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertEqual([], response.json())
|