def setUp(self):
super(BulletinTestCase, self).setUp()
self.test_user_util = TestUserUtility()
self.user = self.test_user_util.user
self.location = Location(name_en='test location', loc_type='Village')
self.location.save()
self.actor = Actor(fullname_en='Test Actor',
fullname_ar='Test name ar',
nickname_en='nick name',
nickname_ar='nick name')
self.actor.save()
self.role = ActorRole(role_status='Detained', actor_id=self.actor.pk)
self.role.save()
self.sourceType = SourceType(source_type='test source type',
description='test source description')
self.sourceType.save()
self.source = Source(reliability_score=0,
name_en='test source',
source_type_id=self.sourceType.pk)
self.source.save()
self.source = Source(reliability_score=0,
name_en='test source 2',
source_type_id=self.sourceType.pk)
self.source.save()
self.label = Label(name_en='test label')
self.label.save()
self.comment = Comment(assigned_user_id=self.user.pk,
status_id=3,
comments_en='test comment')
self.comment.save()
self.media = Media(media_type='Video',
name_en='test media',
media_file='')
self.media.save()
fixture = AutoFixture(Bulletin, generate_m2m={1, 5})
fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
def setUp(self):
super(SourceTestCase, self).setUp()
self.sourceType = SourceType(source_type="test type",
description="test description")
self.sourceType.save()
self.user = User(username='******', password='******', email='*****@*****.**')
self.user.save()
fixture = AutoFixture(Source)
sources = fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
def setUp(self):
super(SourceTestCase, self).setUp()
self.sourceType = SourceType(
source_type="test type",
description="test description"
)
self.sourceType.save()
self.user = User(username='******', password='******', email='*****@*****.**')
self.user.save()
fixture = AutoFixture(Source)
sources = fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
class SourceTestCase(ResourceTestCase):
def setUp(self):
super(SourceTestCase, self).setUp()
self.sourceType = SourceType(source_type="test type",
description="test description")
self.sourceType.save()
self.user = User(username='******', password='******', email='*****@*****.**')
self.user.save()
fixture = AutoFixture(Source)
sources = fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
def tearDown(self):
Source.objects.all().delete()
SourceType.objects.all().delete()
def test_source_get(self):
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
response = self.api_client.get(url)
self.assertEqual(response.status_code, 200)
unauth_url = '/api/v1/source/?format=json'
response = self.api_client.get(unauth_url)
self.assertEqual(response.status_code, 401)
def test_source_post(self):
post_data = {
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type':
"/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
response = self.api_client.post(url, data=post_data)
self.assertEqual(response.status_code, 201)
def test_source_put(self):
source = Source.objects.all()[0]
url = '/api/v1/source/{0}/?format=json{1}'.format(
source.id, self.auth_string)
put_data = {
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type':
"/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
def test_source_patch(self):
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
patch_data = {
'objects': [{
'name_en':
"Test Source",
'name_ar':
"Test Source Arabic",
'comments_en':
"comments en",
'comments_ar':
"comments Arabic",
'reliability_score':
1,
'source_type':
"/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}, {
'name_en':
"Test Source",
'name_ar':
"Test Source Arabic",
'comments_en':
"comments en",
'comments_ar':
"comments Arabic",
'reliability_score':
1,
'source_type':
"/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}]
}
response = self.api_client.patch(url, data=patch_data)
self.assertEqual(response.status_code, 202)
class SourceTestCase(ResourceTestCase):
def setUp(self):
super(SourceTestCase, self).setUp()
self.sourceType = SourceType(
source_type="test type",
description="test description"
)
self.sourceType.save()
self.user = User(username='******', password='******', email='*****@*****.**')
self.user.save()
fixture = AutoFixture(Source)
sources = fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
def tearDown(self):
Source.objects.all().delete()
SourceType.objects.all().delete()
def test_source_get(self):
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
response = self.api_client.get(url)
self.assertEqual(response.status_code, 200)
unauth_url = '/api/v1/source/?format=json'
response = self.api_client.get(unauth_url)
self.assertEqual(response.status_code, 401)
def test_source_post(self):
post_data = {
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type': "/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
response = self.api_client.post(url, data=post_data)
self.assertEqual(response.status_code, 201)
def test_source_put(self):
source = Source.objects.all()[0]
url = '/api/v1/source/{0}/?format=json{1}'.format(source.id, self.auth_string)
put_data = {
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type': "/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
def test_source_patch(self):
url = '/api/v1/source/?format=json{}'.format(self.auth_string)
patch_data = {
'objects': [
{
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type': "/api/vi/sourceType/{0}/".format(self.sourceType.pk),
},
{
'name_en': "Test Source",
'name_ar': "Test Source Arabic",
'comments_en': "comments en",
'comments_ar': "comments Arabic",
'reliability_score': 1,
'source_type': "/api/vi/sourceType/{0}/".format(self.sourceType.pk),
}
]
}
response = self.api_client.patch(url, data=patch_data)
self.assertEqual(response.status_code, 202)
def setUp(self):
super(BulletinTestCase, self).setUp()
self.test_user_util = TestUserUtility()
self.user = self.test_user_util.user
self.location = Location(name_en='test location', loc_type='Village')
self.location.save()
self.actor = Actor(
fullname_en='Test Actor',
fullname_ar='Test name ar',
nickname_en='nick name',
nickname_ar='nick name'
)
self.actor.save()
self.role = ActorRole(role_status='Detained', actor_id=self.actor.pk)
self.role.save()
self.sourceType = SourceType(
source_type='test source type',
description='test source description'
)
self.sourceType.save()
self.source = Source(
reliability_score=0,
name_en='test source',
source_type_id=self.sourceType.pk
)
self.source.save()
self.source = Source(
reliability_score=0,
name_en='test source 2',
source_type_id=self.sourceType.pk
)
self.source.save()
self.label = Label(name_en='test label')
self.label.save()
self.comment = Comment(
assigned_user_id=self.user.pk,
status_id=3,
comments_en='test comment'
)
self.comment.save()
self.media = Media(
media_type='Video',
name_en='test media',
media_file=''
)
self.media.save()
fixture = AutoFixture(Bulletin, generate_m2m={1, 5})
fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
class BulletinTestCase(ResourceTestCase):
fixtures = ['status_update', ]
def setUp(self):
super(BulletinTestCase, self).setUp()
self.test_user_util = TestUserUtility()
self.user = self.test_user_util.user
self.location = Location(name_en='test location', loc_type='Village')
self.location.save()
self.actor = Actor(
fullname_en='Test Actor',
fullname_ar='Test name ar',
nickname_en='nick name',
nickname_ar='nick name'
)
self.actor.save()
self.role = ActorRole(role_status='Detained', actor_id=self.actor.pk)
self.role.save()
self.sourceType = SourceType(
source_type='test source type',
description='test source description'
)
self.sourceType.save()
self.source = Source(
reliability_score=0,
name_en='test source',
source_type_id=self.sourceType.pk
)
self.source.save()
self.source = Source(
reliability_score=0,
name_en='test source 2',
source_type_id=self.sourceType.pk
)
self.source.save()
self.label = Label(name_en='test label')
self.label.save()
self.comment = Comment(
assigned_user_id=self.user.pk,
status_id=3,
comments_en='test comment'
)
self.comment.save()
self.media = Media(
media_type='Video',
name_en='test media',
media_file=''
)
self.media.save()
fixture = AutoFixture(Bulletin, generate_m2m={1, 5})
fixture.create(10)
try:
self.api_key = ApiKey.objects.get(user=self.user)
except ApiKey.DoesNotExist:
self.api_key = ApiKey.objects.create(user=self.user)
self.auth_string = '&username={0}&api_key={1}'.format(
self.user.username, self.api_key.key)
def tearDown(self):
User.objects.all().delete()
Actor.objects.all().delete()
ActorRole.objects.all().delete()
Location.objects.all().delete()
TimeInfo.objects.all().delete()
Media.objects.all().delete()
Comment.objects.all().delete()
StatusUpdate.objects.all().delete()
Bulletin.objects.all().delete()
def test_bulletin_get(self):
url = '/api/v1/bulletin/?format=json{}'.format(self.auth_string)
response = self.api_client.get(url)
self.assertEqual(response.status_code, 200)
unauth_url = '/api/v1/bulletin/?format=json'
response = self.api_client.get(unauth_url)
self.assertEqual(response.status_code, 401)
def test_bulletin_post(self):
post_data = {
'title_en': "Test Bulletin",
'description_ar': "description Arabic",
'confidence_score': 73,
'sources': ['/api/v1/source/1/', ],
'bulletin_imported_comments': ['/api/v1/comment/1/', ],
'assigned_user': '******',
'actors_role': [],
'times': [],
'medias': [],
'locations': [],
'labels': [],
'ref_bulletins': [],
'comment': 'new bulletin',
}
url = '/api/v1/bulletin/?format=json{}'.format(self.auth_string)
response = self.api_client.post(url, data=post_data)
self.assertEqual(response.status_code, 201)
new_bulletin_dict = json.loads(response.content)
new_bulletin = Bulletin(id=new_bulletin_dict['id'])
bulletin_comments = new_bulletin.bulletin_comments.all()
self.assertEqual(len(bulletin_comments), 1)
vs = VersionStatus.objects.filter(
user_id=self.user.id).order_by('version_timestamp')
self.assertEqual(len(vs), 1)
def test_bulletin_put(self):
self.test_user_util.add_user_to_group('data-analyst')
b = Bulletin.objects.all()[1]
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
b.id,
self.auth_string
)
put_data = create_put_data(5)
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
self.check_dehydrated_data(response)
self.assertEqual(retrieve_last_comment_status(response), 'Updated')
vs = VersionStatus.objects.filter(
user_id=self.user.id).order_by('version_timestamp')
self.assertEqual(len(vs), 1)
def test_data_entry_put(self):
self.test_user_util.add_user_to_group('data-entry')
b = Bulletin.objects.all()[0]
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
b.id,
self.auth_string
)
put_data = create_put_data(3)
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
self.assertEqual(retrieve_last_comment_status(response), 'Updated')
b.assigned_user = None
b.save()
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 403)
def test_senior_data_analyst_put(self):
self.test_user_util.add_user_to_group('senior-data-analyst')
b = Bulletin.objects.all()[0]
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
b.id,
self.auth_string
)
put_data = create_put_data(4)
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
self.assertEqual(retrieve_last_comment_status(response), 'Reviewed')
def test_chief_data_analyst_put(self):
self.test_user_util.add_user_to_group('chief-data-analyst')
b = Bulletin.objects.all()[0]
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
b.id,
self.auth_string
)
put_data = create_put_data(5)
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 202)
self.assertEqual(retrieve_last_comment_status(response), 'Finalized')
def test_finalized_is_not_updated(self):
precreated_bulletin = Bulletin.objects.all()[0]
comment = Comment(
assigned_user_id=1,
comments_en='comment',
status_id=5
)
comment.save()
precreated_bulletin.bulletin_comments.add(comment)
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
precreated_bulletin.id, self.auth_string)
put_data = create_put_data(4)
response = self.api_client.put(url, data=put_data)
self.assertEqual(response.status_code, 403)
def test_assigned_user_perm_enforced(self):
self.test_user_util.add_user_to_group('data-analyst')
fixture = AutoFixture(User)
fixture.create(1)
precreated_bulletin = Bulletin.objects.all()[0]
precreated_bulletin.assigned_user = User.objects.get(id=2)
precreated_bulletin.save()
url = '/api/v1/bulletin/{0}/?format=json{1}'.format(
precreated_bulletin.id, self.auth_string)
put_data = create_put_data(4)
response = self.api_client.put(url, data=put_data)
assigned_user_id = retrieve_user_id(response)
self.assertEqual(assigned_user_id, 2)
self.test_user_util.add_user_to_group('chief-data-analyst')
response = self.api_client.put(url, data=put_data)
assigned_user_id = retrieve_user_id(response)
self.assertEqual(assigned_user_id, 1)
def check_dehydrated_data(self, response):
"""
Test that returned data contains required dehydrated fields.
New fields should be added to this method as they are added to the
API to ensure that consistency between front and back end.
"""
dehydrate_keys = [
'bulletin_comments',
'bulletin_imported_comments',
'bulletin_locations',
'bulletin_labels',
'bulletin_times',
'bulletin_sources',
'most_recent_status_bulletin',
'count_actors',
'actor_roles_status',
'actors',
'actors_role'
]
content = json.loads(response.content)
self.assertEqual(
all(
test in content for test in dehydrate_keys
),
True
)
