Add missing unit tests for plays API endpoint
Change-Id: I409788ff4cce8f69a833b193a9d08b92449b64d3
This commit is contained in:
parent
64df7943a4
commit
bd8675bfe1
|
@ -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)
|
||||
|
|
|
@ -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))
|
Loading…
Reference in New Issue