Add missing unit tests for plays API endpoint

Change-Id: I409788ff4cce8f69a833b193a9d08b92449b64d3
This commit is contained in:
David Moreau Simard 2018-06-20 18:27:20 -04:00
parent 64df7943a4
commit bd8675bfe1
No known key found for this signature in database
GPG Key ID: 33A07694CBB71ECC
2 changed files with 83 additions and 0 deletions

View File

@ -28,3 +28,12 @@ class PlaybookFactory(factory.DjangoModelFactory):
completed = True
parameters = b'x\x9c\xabVJ\xcb\xcfW\xb2RPJJ,R\xaa\x05\x00 \x98\x04T'
file = factory.SubFactory(FileFactory)
class PlayFactory(factory.DjangoModelFactory):
class Meta:
model = models.Play
name = 'test play'
completed = True
playbook = factory.SubFactory(PlaybookFactory)

View File

@ -0,0 +1,74 @@
import datetime
from django.utils import timezone
from rest_framework.test import APITestCase
from ara.api import models, serializers
from ara.api.tests import factories
class PlayTestCase(APITestCase):
def test_play_factory(self):
play = factories.PlayFactory(name='play factory')
self.assertEqual(play.name, 'play factory')
def test_play_serializer(self):
playbook = factories.PlaybookFactory()
serializer = serializers.PlaySerializer(data={
'name': 'serializer',
'completed': True,
'playbook': playbook.id
})
serializer.is_valid()
play = serializer.save()
play.refresh_from_db()
self.assertEqual(play.name, 'serializer')
def test_get_no_plays(self):
request = self.client.get('/api/v1/plays/')
self.assertEqual(0, len(request.data['results']))
def test_get_plays(self):
play = factories.PlayFactory()
request = self.client.get('/api/v1/plays/')
self.assertEqual(1, len(request.data['results']))
self.assertEqual(play.name, request.data['results'][0]['name'])
def test_delete_play(self):
play = factories.PlayFactory()
self.assertEqual(1, models.Play.objects.all().count())
request = self.client.delete('/api/v1/plays/%s/' % play.id)
self.assertEqual(204, request.status_code)
self.assertEqual(0, models.Play.objects.all().count())
def test_create_play(self):
playbook = factories.PlaybookFactory()
self.assertEqual(0, models.Play.objects.count())
request = self.client.post('/api/v1/plays/', {
'name': 'create',
'completed': False,
'playbook': playbook.id
})
self.assertEqual(201, request.status_code)
self.assertEqual(1, models.Play.objects.count())
def test_partial_update_play(self):
play = factories.PlayFactory()
self.assertNotEqual('update', play.name)
request = self.client.patch('/api/v1/plays/%s/' % play.id, {
'name': 'update',
})
self.assertEqual(200, request.status_code)
play_updated = models.Play.objects.get(id=play.id)
self.assertEqual('update', play_updated.name)
def test_get_play(self):
play = factories.PlayFactory()
request = self.client.get('/api/v1/plays/%s/' % play.id)
self.assertEqual(play.name, request.data['name'])
def test_get_play_duration(self):
started = timezone.now()
ended = started + datetime.timedelta(hours=1)
play = factories.PlayFactory(started=started, ended=ended)
request = self.client.get('/api/v1/plays/%s/' % play.id)
self.assertEqual(request.data['duration'], datetime.timedelta(0, 3600))