fa6e5338c2
This commit adds fields that are related to the requesting user's interaction with the thread (e.g. following).
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""
|
|
Discussion API test utilities
|
|
"""
|
|
import json
|
|
|
|
import httpretty
|
|
|
|
|
|
class CommentsServiceMockMixin(object):
|
|
"""Mixin with utility methods for mocking the comments service"""
|
|
def register_get_threads_response(self, threads, page, num_pages):
|
|
"""Register a mock response for GET on the CS thread list endpoint"""
|
|
httpretty.register_uri(
|
|
httpretty.GET,
|
|
"http://localhost:4567/api/v1/threads",
|
|
body=json.dumps({
|
|
"collection": threads,
|
|
"page": page,
|
|
"num_pages": num_pages,
|
|
}),
|
|
status=200
|
|
)
|
|
|
|
def register_get_user_response(self, user, subscribed_thread_ids=None, upvoted_ids=None):
|
|
"""Register a mock response for GET on the CS user instance endpoint"""
|
|
httpretty.register_uri(
|
|
httpretty.GET,
|
|
"http://localhost:4567/api/v1/users/{id}".format(id=user.id),
|
|
body=json.dumps({
|
|
"id": str(user.id),
|
|
"subscribed_thread_ids": subscribed_thread_ids or [],
|
|
"upvoted_ids": upvoted_ids or [],
|
|
}),
|
|
status=200
|
|
)
|
|
|
|
def assert_last_query_params(self, expected_params):
|
|
"""
|
|
Assert that the last mock request had the expected query parameters
|
|
"""
|
|
actual_params = dict(httpretty.last_request().querystring)
|
|
actual_params.pop("request_id") # request_id is random
|
|
self.assertEqual(actual_params, expected_params)
|