def test_model_create_activity(self, new_activity_info, new_user_info):
new_user = User.objects.create_user(
first_name=new_user_info['first_name'],
last_name=new_user_info['last_name'],
email=new_user_info['email'],
country=new_user_info['country'],
mobile_number=new_user_info['mobile_number'],
goal=new_user_info['goal'],
password=new_user_info['password'])
category = handle_activity_category(
new_activity_info['category']['name'])
new_activity = Activity.objects.create(
name=new_activity_info['name'],
description=new_activity_info['description'],
start_time=new_activity_info['start_time'],
end_time=new_activity_info['end_time'],
productive=new_activity_info['productive'],
user=new_user,
category=category)
assert new_activity.id
assert new_activity.name == new_activity_info['name']
assert new_activity.description == new_activity_info['description']
assert new_activity.start_time == new_activity_info['start_time']
assert new_activity.end_time == new_activity_info['end_time']
assert new_activity.productive == new_activity_info['productive']
assert new_activity.user == new_user
assert new_activity.category == category
def update(self, activity_instance, data):
category = handle_activity_category(
data.pop('category', {}).get('name'))
activity_instance.name = data.get('name')
activity_instance.description = data.get('description')
activity_instance.start_time = data.get('start_time')
activity_instance.end_time = data.get('end_time')
activity_instance.productive = data.get('productive')
activity_instance.category = category
activity_instance.save()
return activity_instance
def create(self, data):
category = handle_activity_category(
data.pop('category', {}).get('name'))
new_activity = Activity.objects.create(
name=data.get('name'),
description=data.get('description'),
start_time=data.get('start_time'),
end_time=data.get('end_time'),
productive=data.get('productive') or False,
user=self.context['request'].user,
category=category)
return new_activity
def test_user_view(self, new_user_info, new_activity_info):
view = ActivityViewSet.as_view({'get': 'list'})
request = factory.get(ACTIVITIES_URI)
response = view(request)
assert response.status_code == 401
u1 = User.objects.create_user(
first_name=new_user_info['first_name'],
last_name=new_user_info['last_name'],
email=new_user_info['email'],
country=new_user_info['country'],
mobile_number=new_user_info['mobile_number'],
goal=new_user_info['goal'],
password=new_user_info['password'])
category = handle_activity_category(
new_activity_info['category']['name'])
new_activity = Activity.objects.create(
name=new_activity_info['name'],
description=new_activity_info['description'],
start_time=new_activity_info['start_time'],
end_time=new_activity_info['end_time'],
productive=new_activity_info['productive'],
user=u1,
category=category)
view = ActivityViewSet.as_view({'get': 'list'})
request = factory.get(ACTIVITIES_URI,
HTTP_AUTHORIZATION=get_jwt_header(
new_user_info['email'],
new_user_info['password']))
response = view(request)
assert response.status_code == 200
for activity in response.data:
assert activity.get('user_details',
{}).get('email') == new_user_info['email']
def test_view_activity_successful_crud(self, new_user_info,
new_activity_info):
# Create user
User.objects.create_user(first_name=new_user_info['first_name'],
last_name=new_user_info['last_name'],
email=new_user_info['email'],
country=new_user_info['country'],
mobile_number=new_user_info['mobile_number'],
goal=new_user_info['goal'],
password=new_user_info['password'])
# Make datetimes strings and create new category
start_time = new_activity_info['start_time']
end_time = new_activity_info['end_time']
# Stringify datetimes
new_activity_info['start_time'] = str(start_time)
new_activity_info['end_time'] = str(end_time)
new_activity_info['category']['name'] = handle_activity_category(
new_activity_info['category']['name']).name
# Post activity
view = ActivityViewSet.as_view({'post': 'create'})
request = factory.post(ACTIVITIES_URI,
data=json.dumps(new_activity_info),
HTTP_AUTHORIZATION=get_jwt_header(
new_user_info['email'],
new_user_info['password']),
content_type='application/json')
response = view(request)
activity_id = response.data.get('id')
# Assert newly created activity exists
assert response.status_code == 201
assert activity_id
assert response.data.get('name') == new_activity_info['name']
assert response.data.get(
'description') == new_activity_info['description']
assert response.data.get('start_time') == str(
new_activity_info['start_time'].replace(' ', 'T').split('+')[0] +
'Z')
assert response.data.get('end_time') == str(
new_activity_info['end_time'].replace(' ', 'T').split('+')[0] +
'Z')
assert response.data.get('user_details', {}).get('id')
assert response.data.get('user_details',
{}).get('email') == new_user_info['email']
assert response.data.get(
'category',
{}).get('name') == new_activity_info['category']['name']
# Assert user who created the activity can retrieve it
good_retrieve_view = ActivityViewSet.as_view({'get': 'retrieve'})
good_retrieve_request = factory.get(
ACTIVITIES_URI + '/{}'.format(activity_id),
HTTP_AUTHORIZATION=get_jwt_header(new_user_info['email'],
new_user_info['password']))
good_retrieve_response = good_retrieve_view(good_retrieve_request,
pk=activity_id)
assert good_retrieve_response.status_code == 200
assert good_retrieve_response.data.get('id') == activity_id
# Set edited attributes
edited_activity_info = new_activity_info
edited_activity_info['end_time'] = str(end_time + timedelta(hours=1))
edited_activity_info['category']['name'] = 'Hooligans'
good_put_view = ActivityViewSet.as_view({'put': 'update'})
good_put_request = factory.put(
ACTIVITIES_URI + '/{}'.format(activity_id),
data=json.dumps(edited_activity_info),
content_type='application/json',
HTTP_AUTHORIZATION=get_jwt_header(new_user_info['email'],
new_user_info['password']))
good_put_response = good_put_view(good_put_request, pk=activity_id)
# Assert user who created the activity can edited it
# and contains the newly edited info
assert good_put_response.status_code == 200
assert good_put_response.data.get('id') == activity_id
assert good_put_response.data.get(
'name') == edited_activity_info['name']
assert good_put_response.data.get(
'description') == edited_activity_info['description']
assert good_put_response.data.get(
'start_time') == edited_activity_info['start_time'].replace(
' ', 'T').split('+')[0] + 'Z'
assert good_put_response.data.get(
'end_time') == edited_activity_info['end_time'].replace(
' ', 'T').split('+')[0] + 'Z'
assert good_put_response.data.get(
'user_details', {}).get('email') == new_user_info['email']
assert good_put_response.data.get('category',
{}).get('name') == 'Hooligans'
assert Category.objects.get(name=new_activity_info['category']['name'])
assert Category.objects.get(
name=edited_activity_info['category']['name']).name == 'Hooligans'
# Assert user who created the activity can delete it
good_delete_view = ActivityViewSet.as_view({'delete': 'destroy'})
good_delete_request = factory.delete(
ACTIVITIES_URI + '/{}'.format(activity_id),
HTTP_AUTHORIZATION=get_jwt_header(new_user_info['email'],
new_user_info['password']))
good_delete_response = good_delete_view(good_delete_request,
pk=activity_id)
assert good_delete_response.status_code == 204
def test_view_activity_permissions(self, new_user_info, new_activity_info):
# Create first user
User.objects.create_user(first_name=new_user_info['first_name'],
last_name=new_user_info['last_name'],
email=new_user_info['email'],
country=new_user_info['country'],
mobile_number=new_user_info['mobile_number'],
goal=new_user_info['goal'],
password=new_user_info['password'])
# Make datetimes strings and create new category
new_activity_info['start_time'] = str(new_activity_info['start_time'])
new_activity_info['end_time'] = str(new_activity_info['end_time'])
new_activity_info['category']['name'] = handle_activity_category(
new_activity_info['category']['name']).name
view = ActivityViewSet.as_view({'post': 'create'})
request = factory.post(ACTIVITIES_URI,
data=json.dumps(new_activity_info),
HTTP_AUTHORIZATION=get_jwt_header(
new_user_info['email'],
new_user_info['password']),
content_type='application/json')
response = view(request)
activity_id = response.data.get('id')
assert response.status_code == 201
assert activity_id
assert response.data.get('name') == new_activity_info['name']
assert response.data.get(
'description') == new_activity_info['description']
assert response.data.get('start_time') == str(
new_activity_info['start_time'].replace(' ', 'T').split('+')[0] +
'Z')
assert response.data.get('end_time') == str(
new_activity_info['end_time'].replace(' ', 'T').split('+')[0] +
'Z')
assert response.data.get('user_details', {}).get('id')
assert response.data.get('user_details',
{}).get('email') == new_user_info['email']
assert response.data.get(
'category',
{}).get('name') == new_activity_info['category']['name']
onui = other_new_user_info()
User.objects.create_user(first_name=onui['first_name'],
last_name=onui['last_name'],
email=onui['email'],
country=onui['country'],
mobile_number=onui['mobile_number'],
goal=onui['goal'],
password=onui['password'])
# u2 can't GET created activity
bad_retrieve_view = ActivityViewSet.as_view({'get': 'retrieve'})
bad_retrieve_request = factory.get(
ACTIVITIES_URI + '/{}'.format(activity_id),
HTTP_AUTHORIZATION=get_jwt_header(onui['email'], onui['password']))
bad_retrieve_response = view(bad_retrieve_request, pk=activity_id)
assert bad_retrieve_response.status_code == 405
# u2 can't PUT created activity
edited_activity_info = new_activity_info
edited_activity_info['name'] = 'slkdjfalsdf'
edited_activity_info['description'] = 'slkdjfalsdf'
bad_put_view = ActivityViewSet.as_view({'put': 'update'})
bad_put_request = factory.put(
ACTIVITIES_URI + '/{}'.format(activity_id),
data=json.dumps(edited_activity_info),
content_type='application/json',
HTTP_AUTHORIZATION=get_jwt_header(onui['email'], onui['password']))
response = view(bad_put_request, pk=activity_id)
assert response.status_code == 405
# u2 can't DELETE created activity
bad_delete_view = ActivityViewSet.as_view({'delete': 'destroy'})
bad_delete_request = factory.delete(
ACTIVITIES_URI + '/{}'.format(activity_id),
HTTP_AUTHORIZATION=get_jwt_header(onui['email'], onui['password']))
bad_delete_response = view(bad_delete_request, pk=activity_id)
assert response.status_code == 405