class ApiTestCase(ResourceTestCase):
fixtures = ['app/detective/fixtures/default_topics.json',
'app/detective/fixtures/search_terms.json',]
def setUp(self):
super(ApiTestCase, self).setUp()
# Use custom api client
self.api_client = TestApiClient()
self.salt = SaltMixin.salt
self.super_username = '******'
self.super_password = '******'
self.super_email = '*****@*****.**'
self.contrib_username = '******'
self.contrib_password = '******'
self.contrib_email = '*****@*****.**'
self.lambda_username = '******'
self.lambda_password = '******'
self.lambda_email = '*****@*****.**'
contributors = Group.objects.get(name='energy_contributor')
# Create the new user users
super_user = User.objects.create(
username=self.super_username,
email=self.super_email,
is_staff=True,
is_superuser = True
)
super_user.set_password(self.super_password)
super_user.save()
self.super_user = super_user
contrib_user = User.objects.create(
username=self.contrib_username,
email=self.contrib_email,
is_active=True
)
contrib_user.groups.add(contributors)
contrib_user.set_password(self.contrib_password)
contrib_user.save()
self.contrib_user = contrib_user
lambda_user = User.objects.create(
username=self.lambda_username,
email=self.lambda_email,
)
lambda_user.set_password(self.lambda_password)
lambda_user.save()
self.lambda_user = lambda_user
# Create related objects
self.jpp = Organization(name=u"Journalism++")
self.jpp._author = [super_user.pk]
self.jpp.founded = datetime(2011, 4, 3)
self.jpp.website_url = 'http://jplusplus.com'
self.jpp.save()
self.jg = Organization(name=u"Journalism Grant")
self.jg._author = [super_user.pk]
self.jg.save()
self.fra = Country(name=u"France", isoa3=u"FRA")
self.fra.save()
self.pr = Person(name=u"Pierre Roméra")
self.pr.based_in.add(self.fra)
self.pr.activity_in_organization.add(self.jpp)
self.pr.save()
self.pb = Person(name=u"Pierre Bellon")
self.pb.based_in.add(self.fra)
self.pb.activity_in_organization.add(self.jpp)
self.pb.save()
# Creates Christmas topic
ontology = File(open(settings.DATA_ROOT + "/ontology-v5.7.owl"))
self.christmas = Topic(slug=u"christmas", title="It's christmas!", ontology_as_owl=ontology, author=super_user)
self.christmas.save()
# Creates Thanksgiving topic
self.thanksgiving = Topic(slug=u"thanksgiving", title="It's thanksgiving!", ontology_as_owl=ontology, author=super_user)
self.thanksgiving.save()
self.post_data_simple = {
"name": "Lorem ispum TEST",
"twitter_handle": "loremipsum"
}
self.post_data_related = {
"name": "Lorem ispum TEST RELATED",
"owner": [
{ "id": self.jpp.id },
{ "id": self.jg.id }
],
"activity_in_country": [
{ "id": self.fra.id }
]
}
self.rdf_jpp = {
"label": u"Person that has activity in Journalism++",
"object": {
"id": self.jpp.id,
"model": u"Organization",
"name": u"Journalism++"
},
"predicate": {
"label": u"has activity in",
"name": u"activity_in_organization",
"subject": u"Person"
},
"subject": {
"label": u"Person",
"name": u"Person"
}
}
def cleanModel(self, model_instance):
if model_instance:
model_instance.delete()
def tearDown(self):
# Clean & delete generated data
# individuals
self.cleanModel(self.jpp) # organization
self.cleanModel(self.jg) # organization
self.cleanModel(self.fra) # country
self.cleanModel(self.pr) # people
self.cleanModel(self.pb) # people
# Simply flush the database
management.call_command('flush', verbosity=0, interactive=False)
# Utility functions (Auth, operation etc.)
def login(self, username, password):
return self.api_client.client.login(username=username, password=password)
def get_super_credentials(self):
return self.login(self.super_username, self.super_password)
def get_contrib_credentials(self):
return self.login(self.contrib_username, self.contrib_password)
def get_lambda_credentials(self):
return self.login(self.lambda_username, self.lambda_password)
def signup_user(self, user_dict):
""" Utility method to signup through API """
return self.api_client.post('/api/common/v1/user/signup/', format='json', data=user_dict)
def patch_individual(self, scope=None, model_name=None, model_id=None,
patch_data=None, auth=None, skip_auth=False):
if not skip_auth and not auth:
auth = self.get_super_credentials()
url = '/api/%s/v1/%s/%d/patch/' % (scope, model_name, model_id)
return self.api_client.post(url, format='json', data=patch_data, authentication=auth)
def check_permissions(self, permissions=None, user=None):
user_permissions = list(user.get_all_permissions())
self.assertEqual(len(user_permissions), len(permissions))
for perm in user_permissions:
self.assertTrue(perm in permissions)
# All test functions
def test_user_signup_succeed(self):
"""
Test with proper data to signup user
Expected: HTTT 201 (Created)
"""
user_dict = dict(username=u"newuser", password=u"newuser", email=u"*****@*****.**")
resp = self.signup_user(user_dict)
self.assertHttpCreated(resp)
def test_user_signup_empty_data(self):
"""
Test with empty data to signup user
Expected: HTTP 400 (BadRequest)
"""
user_dict = dict(username=u"", password=u"", email=u"")
resp = self.signup_user(user_dict)
self.assertHttpBadRequest(resp)
def test_user_signup_no_data(self):
resp = self.api_client.post('/api/common/v1/user/signup/', format='json')
self.assertHttpBadRequest(resp)
def test_user_signup_existing_user(self):
user_dict = dict(username=self.super_username, password=self.super_password, email=self.super_email)
resp = self.signup_user(user_dict)
self.assertHttpForbidden(resp)
def test_user_activate_succeed(self):
user_dict = dict(username='******', password='******', email='*****@*****.**')
self.assertHttpCreated(self.signup_user(user_dict))
innactive_user = User.objects.get(email=user_dict.get('email'))
activation_profile = RegistrationProfile.objects.get(user=innactive_user)
activation_key = activation_profile.activation_key
resp_activate = self.api_client.get('/api/common/v1/user/activate/?token=%s' % activation_key)
self.assertHttpOK(resp_activate)
user = User.objects.get(email=user_dict.get('email'))
self.assertTrue(user.is_active)
def test_user_activate_fake_token(self):
resp = self.api_client.get('/api/common/v1/user/activate/?token=FAKE')
self.assertHttpForbidden(resp)
def test_user_activate_no_token(self):
resp = self.api_client.get('/api/common/v1/user/activate/')
self.assertHttpBadRequest(resp)
def test_user_activate_empty_token(self):
resp = self.api_client.get('/api/common/v1/user/activate/?token')
self.assertHttpBadRequest(resp)
def test_user_contrib_login_succeed(self):
auth = dict(username=self.contrib_username, password=self.contrib_password)
resp = self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
self.assertValidJSON(resp.content)
data = json.loads(resp.content)
self.assertTrue(data["success"])
self.check_permissions(permissions=data.get("permissions"), user=self.contrib_user)
def test_user_login_succeed(self):
auth = dict(username=self.super_username, password=self.super_password)
resp = self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["success"], True)
def test_user_login_failed(self):
auth = dict(username=self.super_username, password=u"awrongpassword")
resp = self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["success"], False)
def test_user_logout_succeed(self):
# First login
auth = dict(username=self.super_username, password=self.super_password)
self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
# Then logout
resp = self.api_client.get('/api/common/v1/user/logout/', format='json')
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["success"], True)
def test_user_logout_failed(self):
resp = self.api_client.get('/api/common/v1/user/logout/', format='json')
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["success"], False)
def test_user_permissions_is_logged(self):
auth = dict(username=self.contrib_username, password=self.contrib_password)
self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
resp = self.api_client.get('/api/common/v1/user/permissions/', format='json')
self.assertValidJSON(resp.content)
data = json.loads(resp.content)
self.check_permissions(permissions=data.get("permissions"), user=self.contrib_user)
def test_user_permissions_isnt_logged(self):
resp = self.api_client.get('/api/common/v1/user/permissions/', format='json')
self.assertHttpUnauthorized(resp)
def test_user_status_isnt_logged(self):
resp = self.api_client.get('/api/common/v1/user/status/', format='json')
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["is_logged"], False)
def test_user_status_is_logged(self):
# Log in
auth = dict(username=self.super_username, password=self.super_password)
resp = self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
self.assertValidJSON(resp.content)
data = json.loads(resp.content)
self.assertTrue(data['success'])
resp = self.api_client.get('/api/common/v1/user/status/', format='json')
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["is_logged"], True)
def test_contrib_user_status_is_logged(self):
# Log in
auth = dict(username=self.contrib_username, password=self.contrib_password, remember_me=True)
resp = self.api_client.post('/api/common/v1/user/login/', format='json', data=auth)
self.assertValidJSON(resp.content)
data = json.loads(resp.content)
self.assertTrue(data['success'])
resp = self.api_client.get('/api/common/v1/user/status/', format='json')
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual(data["is_logged"], True)
def test_reset_password_success(self):
email = dict(email=self.super_email)
resp = self.api_client.post('/api/common/v1/user/reset_password/', format='json', data=email)
self.assertValidJSON(resp.content)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertTrue(data['success'])
def test_reset_password_wrong_email(self):
email = dict(email="*****@*****.**")
resp = self.api_client.post('/api/common/v1/user/reset_password/', format='json', data=email)
self.assertEqual(resp.status_code in [302, 404], True)
def test_reset_password_no_data(self):
resp = self.api_client.post('/api/common/v1/user/reset_password/', format='json')
self.assertHttpBadRequest(resp)
def test_reset_password_empty_email(self):
resp = self.api_client.post('/api/common/v1/user/reset_password/', format='json', data=dict(email=''))
self.assertHttpBadRequest(resp)
def test_reset_password_confirm_succes(self):
"""
Test to successfuly reset a password with a new one.
Expected:
HTTP 200 - OK
"""
token = signing.dumps(self.super_user.pk, salt=self.salt)
password = "******"
auth = dict(password=password, token=token)
resp = self.api_client.post(
'/api/common/v1/user/reset_password_confirm/',
format='json',
data=auth
)
self.assertValidJSON(resp.content)
data = json.loads(resp.content)
self.assertTrue(data['success'])
# we query users to get the latest user object (updated with password)
user = User.objects.get(email=self.super_user.email)
self.assertTrue(user.check_password(password))
def test_reset_password_confirm_no_data(self):
"""
Test on reset_password_confirm API endpoint without any data.
Expected response:
HTTP 400 (BadRequest).
Explanation:
Every request on /reset_password_confirm/ must have a JSON data payload.
{
password: ... // the password to reset"
token: ... // the reset password token (received by emai)
}
"""
resp = self.api_client.post('/api/common/v1/user/reset_password_confirm/', format='json')
self.assertHttpBadRequest(resp)
self.assertIsNotNone(resp.content)
def test_reset_password_confirm_empty_data(self):
"""
Test on reset_password_confirm API endpoint with empty data:
{
password: ""
token: ""
}
Expected result:
HTTP 400 (BadRequest)
Explanation:
A reset_password_confirm request must have a password and should be
authenticated with a token.
"""
auth = dict(password='', token='')
resp = self.api_client.post('/api/common/v1/user/reset_password_confirm/', format='json', data=auth)
self.assertHttpBadRequest(resp)
def test_reset_password_confirm_fake_token(self):
"""
Test on reset_password_confirm API endpoint with empty data:
{
password: ""
token: ""
}
Expected result:
HTTP 403 (Forbidden)
Explanation:
A reset_password_confirm request should be authenticated with a valid
token.
"""
fake_token = 'f4k:t0k3N'
auth = dict(password='******', token=fake_token)
resp = self.api_client.post(
'/api/common/v1/user/reset_password_confirm/',
format='json',
data=auth
)
self.assertHttpForbidden(resp)
def test_get_list_json(self):
resp = self.api_client.get('/api/energy/v1/energyproject/?limit=20', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Number of element on the first page
count = min(20, EnergyProject.objects.count())
self.assertEqual( len(self.deserialize(resp)['objects']), count)
def test_post_list_unauthenticated(self):
self.assertHttpUnauthorized(self.api_client.post('/api/energy/v1/energyproject/', format='json', data=self.post_data_simple))
def test_post_list_staff(self):
# Check how many are there first.
count = EnergyProject.objects.count()
self.assertHttpCreated(
self.api_client.post('/api/energy/v1/energyproject/',
format='json',
data=self.post_data_simple,
authentication=self.get_super_credentials()
)
)
# Verify a new one has been added.
self.assertEqual(EnergyProject.objects.count(), count+1)
def test_post_list_contributor(self):
# Check how many are there first.
count = EnergyProject.objects.count()
self.assertHttpCreated(
self.api_client.post('/api/energy/v1/energyproject/',
format='json',
data=self.post_data_simple,
authentication=self.get_contrib_credentials()
)
)
# Verify a new one has been added.
self.assertEqual(EnergyProject.objects.count(), count+1)
def test_post_list_lambda(self):
self.assertHttpUnauthorized(
self.api_client.post('/api/energy/v1/energyproject/',
format='json',
data=self.post_data_simple,
authentication=self.get_lambda_credentials()
)
)
def test_post_list_related(self):
# Check how many are there first.
count = EnergyProject.objects.count()
# Record API response to extract data
resp = self.api_client.post('/api/energy/v1/energyproject/',
format='json',
data=self.post_data_related,
authentication=self.get_super_credentials()
)
# Vertify the request status
self.assertHttpCreated(resp)
# Verify a new one has been added.
self.assertEqual(EnergyProject.objects.count(), count+1)
# Are the data readable?
self.assertValidJSON(resp.content)
# Parse data to verify relationship
data = json.loads(resp.content)
self.assertEqual(len(data["owner"]), len(self.post_data_related["owner"]))
self.assertEqual(len(data["activity_in_country"]), len(self.post_data_related["activity_in_country"]))
def test_mine(self):
resp = self.api_client.get('/api/energy/v1/organization/mine/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual( len(data["objects"]), 2)
def test_mine_empty(self):
resp = self.api_client.get('/api/energy/v1/organization/mine/', format='json')
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
self.assertEqual( len(data["objects"]), 0)
def test_search_organization(self):
resp = self.api_client.get('/api/energy/v1/organization/search/?q=Journalism', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
# At least 2 results
self.assertGreater( len(data.items()), 1 )
def test_search_organization_wrong_page(self):
resp = self.api_client.get('/api/energy/v1/organization/search/?q=Roméra&page=10000', format='json', authentication=self.get_super_credentials())
self.assertEqual(resp.status_code in [302, 404], True)
def test_cypher_detail(self):
resp = self.api_client.get('/api/common/v1/cypher/111/', format='json', authentication=self.get_super_credentials())
self.assertTrue(resp.status_code in [302, 404])
def test_cypher_unauthenticated(self):
self.assertHttpUnauthorized(self.api_client.get('/api/common/v1/cypher/?q=START%20n=node%28*%29RETURN%20n;', format='json'))
def test_cypher_unauthorized(self):
# Ensure the user isn't authorized to process cypher request
self.super_user.is_staff = True
self.super_user.is_superuser = False
self.super_user.save()
self.assertHttpUnauthorized(self.api_client.get('/api/common/v1/cypher/?q=START%20n=node%28*%29RETURN%20n;', format='json', authentication=self.get_super_credentials()))
def test_cypher_authorized(self):
# Ensure the user IS authorized to process cypher request
self.super_user.is_superuser = True
self.super_user.save()
self.assertValidJSONResponse(self.api_client.get('/api/common/v1/cypher/?q=START%20n=node%28*%29RETURN%20n;', format='json', authentication=self.get_super_credentials()))
def test_summary_list(self):
resp = self.api_client.get('/api/common/v1/summary/', format='json')
self.assertEqual(resp.status_code in [302, 404], True)
def test_summary_mine_success(self):
resp = self.api_client.get('/api/energy/v1/summary/mine/', authentication=self.get_super_credentials(), format='json')
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
objects = data['objects']
self.assertIsNotNone(find(lambda x: x['label'] == self.jpp.name, objects))
self.assertIsNotNone(find(lambda x: x['label'] == self.jg.name, objects))
def test_summary_mine_unauthenticated(self):
resp = self.api_client.get('/api/energy/v1/summary/mine/', format='json')
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
objects = data['objects']
self.assertEqual(len(objects), 0)
def test_countries_summary(self):
resp = self.api_client.get('/api/energy/v1/summary/countries/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
# Only France is present
self.assertGreater(len(data), 0)
# We added 1 relation to France
self.assertEqual("count" in data["FRA"], True)
def test_forms_summary(self):
resp = self.api_client.get('/api/energy/v1/summary/forms/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
# As many descriptors as models
self.assertEqual( 11, len(data.items()) )
def test_types_summary(self):
resp = self.api_client.get('/api/energy/v1/summary/types/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
def test_search_summary(self):
resp = self.api_client.get('/api/energy/v1/summary/search/?q=Journalism', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
# Parse data to check the number of result
data = json.loads(resp.content)
# At least 2 results
self.assertGreater( len(data.items()), 1 )
def test_search_summary_wrong_offset(self):
resp = self.api_client.get('/api/energy/v1/summary/search/?q=Journalism&offset=-1', format='json', authentication=self.get_super_credentials())
self.assertEqual(resp.status_code in [302, 404], True)
def test_summary_human_search(self):
query = "Person activity in Journalism"
resp = self.api_client.get('/api/energy/v1/summary/human/?q=%s' % query, format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
data = json.loads(resp.content)
self.assertGreater(len(data['objects']), 1)
def test_rdf_search(self):
# RDF object for persons that have activity in J++, we need to urlencode
# the JSON string to avoid '+' loss
rdf_str = urllib.quote(json.dumps(self.rdf_jpp))
url = '/api/energy/v1/summary/rdf_search/?limit=20&offset=0&q=%s' % rdf_str
resp = self.api_client.get(url, format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
data = json.loads(resp.content)
objects = data['objects']
self.assertIsNotNone(find(lambda x: x['name'] == self.pr.name, objects))
self.assertIsNotNone(find(lambda x: x['name'] == self.pb.name, objects))
def test_patch_individual_date_staff(self):
"""
Test a patch request on an invidividual's date attribute.
Request: /api/energy/v1/organization/
Expected: HTTP 200 (OK)
"""
# date are subject to special process with patch method.
new_date = datetime(2011, 4, 1, 0, 0, 0, 0)
data = {
'founded': new_date.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
}
args = {
'scope' : 'energy',
'model_id' : self.jpp.id,
'model_name' : 'organization',
'patch_data' : data
}
resp = self.patch_individual(**args)
self.assertHttpOK(resp)
self.assertValidJSONResponse(resp)
updated_jpp = Organization.objects.get(name=self.jpp.name)
self.assertEqual(timezone.make_naive(updated_jpp.founded), new_date)
def test_patch_individual_website_staff(self):
jpp_url = 'http://jplusplus.org'
data = {
'website_url': jpp_url,
}
args = {
'scope' : 'energy',
'model_id' : self.jpp.id,
'model_name' : 'organization',
'patch_data' : data
}
resp = self.patch_individual(**args)
self.assertHttpOK(resp)
self.assertValidJSONResponse(resp)
updated_jpp = Organization.objects.get(name=self.jpp.name)
self.assertEqual(updated_jpp.website_url, jpp_url)
def test_patch_individual_website_unauthenticated(self):
jpp_url = 'http://jplusplus.org'
data = {
'website_url': jpp_url,
}
args = {
'scope' : 'energy',
'model_id' : self.jpp.id,
'model_name' : 'organization',
'patch_data' : data,
'skip_auth' : True,
}
resp = self.patch_individual(**args)
self.assertHttpUnauthorized(resp)
def test_patch_individual_website_contributor(self):
jpp_url = 'http://www.jplusplus.org'
data = {
'website_url': jpp_url,
}
args = {
'scope' : 'energy',
'model_id' : self.jpp.id,
'model_name' : 'organization',
'patch_data' : data,
'auth' : self.get_contrib_credentials(),
}
resp = self.patch_individual(**args)
self.assertHttpOK(resp)
self.assertValidJSONResponse(resp)
updated_jpp = Organization.objects.filter(name=self.jpp.name)[0]
self.assertEqual(updated_jpp.website_url, jpp_url)
def test_patch_individual_website_lambda(self):
jpp_url = 'http://bam.jplusplus.org'
data = {
'website_url': jpp_url,
}
args = {
'scope' : 'energy',
'model_id' : self.jpp.id,
'model_name' : 'organization',
'patch_data' : data,
'auth' : self.get_lambda_credentials(),
}
resp = self.patch_individual(**args)
self.assertHttpUnauthorized(resp)
def test_patch_individual_not_found(self):
jpp_url = 'http://jplusplus.org'
data = {
'website_url': jpp_url,
}
args = {
'scope' : 'energy',
'model_id' : 1337,
'model_name' : 'organization',
'patch_data' : data,
}
resp = self.patch_individual(**args)
self.assertEqual(resp.status_code in [302, 404], True)
def test_topic_endpoint_exists(self):
resp = self.api_client.get('/api/common/v1/topic/?slug=christmas', follow=True, format='json')
# Parse data to check the number of result
data = json.loads(resp.content)
# 1 result
self.assertEqual( len( data["objects"] ), 1 )
def test_topic_api_exists(self):
resp = self.api_client.get('/api/christmas/v1/', format='json')
self.assertValidJSONResponse(resp)
def test_topic_has_person(self):
resp = self.api_client.get('/api/christmas/v1/', format='json')
self.assertValidJSONResponse(resp)
def test_topic_multiple_api(self):
# API 1
resp = self.api_client.get('/api/christmas/v1/', format='json')
self.assertValidJSONResponse(resp)
# API 2
resp = self.api_client.get('/api/thanksgiving/v1/', format='json')
self.assertValidJSONResponse(resp)
def test_topic_has_summary_syntax_from_ontology(self):
resp = self.api_client.get('/api/christmas/v1/summary/syntax/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
def test_topic_has_summary_syntax_from_file(self):
resp = self.api_client.get('/api/energy/v1/summary/syntax/', format='json', authentication=self.get_super_credentials())
self.assertValidJSONResponse(resp)
class ApiTestCase(ResourceTestCase):
fixtures = ['app/detective/fixtures/default_topics.json',
'app/detective/fixtures/default_skeletons.json',
'app/detective/fixtures/tests_topics.json',
'app/detective/fixtures/tests_pillen.json',
'app/detective/fixtures/search_terms.json',]
def setUp(self):
super(ApiTestCase, self).setUp()
# Use custom api client
self.api_client = TestApiClient()
self.salt = SaltMixin.salt
self.super_username = '******'
self.super_password = '******'
self.super_email = '*****@*****.**'
self.contrib_username = '******'
self.contrib_password = '******'
self.contrib_email = '*****@*****.**'
self.lambda_username = '******'
self.lambda_password = '******'
self.lambda_email = '*****@*****.**'
contributors = Group.objects.get(name='energy_contributor')
test_contributors = Topic.objects.get(slug='test-topic').get_contributor_group()
# Create the new user users
super_user = User.objects.create(
username=self.super_username,
email=self.super_email,
is_staff=True,
is_superuser = True
)
super_user.set_password(self.super_password)
super_user.save()
self.super_user = super_user
contrib_user = User.objects.create(
username=self.contrib_username,
email=self.contrib_email,
is_active=True
)
# put the user in the best plan
profile = contrib_user.detectiveprofileuser
profile.plan = PLANS_CHOICES[-1][0]
profile.save()
contrib_user.groups.add(contributors)
contrib_user.groups.add(test_contributors)
contrib_user.set_password(self.contrib_password)
contrib_user.save()
self.contrib_user = contrib_user
lambda_user = User.objects.create(
username=self.lambda_username,
email=self.lambda_email,
)
lambda_user.set_password(self.lambda_password)
lambda_user.save()
self.lambda_user = lambda_user
# Create related objects
self.jpp = Organization(name=u"Journalism++")
self.jpp._author = [super_user.pk]
self.jpp.founded = datetime(2011, 4, 3)
self.jpp.website_url = 'http://jplusplus.com'
self.jpp.save()
self.jg = Organization(name=u"Journalism Grant")
self.jg._author = [super_user.pk]
self.jg.save()
self.fra = Country(name=u"France", isoa3=u"FRA")
self.fra.save()
self.pr = Person(name=u"Pierre Roméra")
self.pr.based_in.add(self.fra)
self.pr.activity_in_organization.add(self.jpp)
self.pr.save()
self.pb = Person(name=u"Pierre Bellon")
self.pb.based_in.add(self.fra)
self.pb.activity_in_organization.add(self.jpp)
self.pb.save()
# Creates Christmas topic
ontology = File(open(settings.DATA_ROOT + "/ontology-v5.7.owl"))
self.christmas = Topic(slug=u"christmas", title="It's christmas!", ontology_as_owl=ontology, author=super_user)
self.christmas.save()
# Creates Thanksgiving topic
self.thanksgiving = Topic(slug=u"thanksgiving", title="It's thanksgiving!", ontology_as_owl=ontology, author=super_user)
self.thanksgiving.save()
self.post_data_simple = {
"name": "Lorem ispum TEST",
"twitter_handle": "loremipsum"
}
self.post_data_related = {
"name": "Lorem ispum TEST RELATED",
"owner": [
{ "id": self.jpp.id },
{ "id": self.jg.id }
],
"activity_in_country": [
{ "id": self.fra.id }
]
}
self.rdf_jpp = {
"label": u"Person that has activity in Journalism++",
"object": {
"id": self.jpp.id,
"model": u"Organization",
"name": u"Journalism++"
},
"predicate": {
"label": u"has activity in",
"name": u"activity_in_organization",
"subject": u"Person"
},
"subject": {
"label": u"Person",
"name": u"Person"
}
}
def cleanModel(self, model_instance):
if model_instance:
model_instance.delete()
def tearDown(self):
# Clean & delete generated data
# individuals
self.cleanModel(self.jpp) # organization
self.cleanModel(self.jg) # organization
self.cleanModel(self.fra) # country
self.cleanModel(self.pr) # people
self.cleanModel(self.pb) # people
# Simply flush the database
management.call_command('flush', verbosity=0, interactive=False)
def create_user(self, username='******', password=None, email=None, plan=None):
if not email:
email = "*****@*****.**" % username
if not plan:
plan = PLANS_CHOICES[-1][0]
if not password:
password = username
user = User.objects.create(
username=username,
email=email,
is_active=True
)
# put the user in the best plan
profile = user.detectiveprofileuser
profile.plan = plan
profile.save()
user.set_password(password)
user.save()
return user
# Utility functions (Auth, operation etc.)
def login(self, username, password):
return self.api_client.client.login(username=username, password=password)
def logout(self):
return self.api_client.client.logout()
def get_super_credentials(self):
return self.login(self.super_username, self.super_password)
def get_contrib_credentials(self):
return self.login(self.contrib_username, self.contrib_password)
def get_lambda_credentials(self):
return self.login(self.lambda_username, self.lambda_password)
def signup_user(self, user_dict):
""" Utility method to signup through API """
return self.api_client.post('/api/detective/common/v1/user/signup/', format='json', data=user_dict)
def patch_individual(self, scope, model_name, model_id,
patch_data=None, auth=None, skip_auth=False):
if not skip_auth and not auth:
auth = self.get_super_credentials()
url = '/api/%s/v1/%s/%d/patch/' % (scope, model_name, model_id)
return self.api_client.post(url, format='json', data=patch_data, authentication=auth)
def check_permissions(self, permissions=None, user=None):
user_permissions = list(user.get_all_permissions())
self.assertEqual(len(user_permissions), len(permissions))
for perm in user_permissions:
self.assertTrue(perm in permissions)
def topic_to_dict(self, topic):
return {
'description': topic.description,
'title': topic.title,
'slug': topic.slug,
'ontology_as_json': topic.ontology_as_json,
'ontology_as_owl': topic.ontology_as_owl,
'ontology_as_mod': topic.ontology_as_mod,
'about': topic.about,
'background': topic.background,
'public': topic.public,
'featured': topic.featured,
'author': topic.author
}