Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import json
- from django.test import TestCase
- from rest_framework.test import APITestCase
- from django.contrib.auth.models import User
- import api.models as models
- import api.drf_views as views
- ...
- class SituationReportTypeTest(APITestCase):
- fixtures = ['DisasterTypes']
- def setUp(self):
- type1 = models.DocumentType.objects.create(type='Strange')
- type2 = models.DocumentType.objects.create(type='Lyric')
- type3 = models.DocumentType.objects.create(type='Epic')
- dtype1 = models.DisasterType.objects.get(pk=1)
- dtype2 = models.DisasterType.objects.get(pk=2)
- dtype3 = models.DisasterType.objects.get(pk=4)
- event1 = models.Event.objects.create(name='disaster1', summary='test disaster1', dtype=dtype1)
- event2 = models.Event.objects.create(name='disaster2', summary='test disaster2', dtype=dtype2)
- event3 = models.Event.objects.create(name='disaster3', summary='test disaster3', dtype=dtype3)
- report1 = models.SituationReport.objects.create(name='test1', event=event1, type=type1)
- report2 = models.SituationReport.objects.create(name='test2', event=event2, type=type2)
- report3 = models.SituationReport.objects.create(name='test3', event=event3, type=type3)
- def test_sit_rep_types(self):
- type1 = models.DocumentType.objects.get(type='Strange')
- type2 = models.DocumentType.objects.get(type='Lyric')
- type3 = models.DocumentType.objects.get(type='Epic')
- report1 = models.SituationReport.objects.get(name='test1')
- report2 = models.SituationReport.objects.get(name='test2')
- report3 = models.SituationReport.objects.get(name='test3')
- self.assertIsNotNone(type1)
- self.assertIsNotNone(type2)
- self.assertIsNotNone(type3)
- self.assertIsNotNone(report1)
- self.assertIsNotNone(report2)
- self.assertIsNotNone(report3)
- print(report3)
- body = {
- 'limit': 100,
- 'event': 2,
- }
- headers = {'CONTENT_TYPE': 'application/json'}
- # response = self.client.post('/api/v2/situation_report', body, format='json', headers=headers)
- response = self.client.get('/api/v2/situation_report/?limit=100&event=1')
- print(response.content)
- # response.content:
- self.assertEqual(response.status_code, 200)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement