kopia lustrzana https://gitlab.com/marnanel/chapeau
bowler's tests gain BowlerClient, a subclass of django.test.Client
which will set the Accept param correctly on get requests. Used throughout test_views instead of d.t.Client.merge-requests/2/merge
rodzic
7b37822d8d
commit
2e794e2a8a
|
@ -35,6 +35,18 @@ CONTEXT_URL = "https://www.w3.org/ns/activitystreams"
|
||||||
|
|
||||||
logger = logging.getLogger(name='kepi')
|
logger = logging.getLogger(name='kepi')
|
||||||
|
|
||||||
|
MIME_TYPE = 'application/activity+json'
|
||||||
|
|
||||||
|
class BowlerClient(django.test.Client):
|
||||||
|
|
||||||
|
def get(self, *args, **kwargs):
|
||||||
|
|
||||||
|
result = super().get(*args, **kwargs,
|
||||||
|
HTTP_ACCEPT = MIME_TYPE,
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
class DummyMessage(object):
|
class DummyMessage(object):
|
||||||
|
|
||||||
fields = None
|
fields = None
|
||||||
|
@ -276,7 +288,7 @@ def post_test_message(
|
||||||
):
|
):
|
||||||
|
|
||||||
if client is None:
|
if client is None:
|
||||||
client = django.test.Client()
|
client = BowlerClient()
|
||||||
|
|
||||||
body, headers = test_message_body_and_headers(
|
body, headers = test_message_body_and_headers(
|
||||||
secret = secret,
|
secret = secret,
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
from django.test import TestCase, Client
|
from django.test import TestCase
|
||||||
from . import *
|
from . import *
|
||||||
from kepi.trilby_api.tests import create_local_person, create_local_status
|
from kepi.trilby_api.tests import create_local_person, create_local_status
|
||||||
import logging
|
import logging
|
||||||
|
@ -7,6 +7,7 @@ import json
|
||||||
logger = logging.getLogger(name='kepi')
|
logger = logging.getLogger(name='kepi')
|
||||||
|
|
||||||
ALICE_SUMMARY = 'Remember Alice? It\'s a song about Alice.'
|
ALICE_SUMMARY = 'Remember Alice? It\'s a song about Alice.'
|
||||||
|
ALICE_SUMMARY_HTML = f'<p>{ALICE_SUMMARY}</p>'
|
||||||
|
|
||||||
def _response_to_dict(response):
|
def _response_to_dict(response):
|
||||||
|
|
||||||
|
@ -32,7 +33,7 @@ class Tests(TestCase):
|
||||||
note = ALICE_SUMMARY,
|
note = ALICE_SUMMARY,
|
||||||
)
|
)
|
||||||
|
|
||||||
c = Client()
|
c = BowlerClient()
|
||||||
response = c.get('/users/alice')
|
response = c.get('/users/alice')
|
||||||
self.assertEqual(response.status_code, 200)
|
self.assertEqual(response.status_code, 200)
|
||||||
result = _response_to_dict(response)
|
result = _response_to_dict(response)
|
||||||
|
@ -74,7 +75,7 @@ class Tests(TestCase):
|
||||||
'BfTLbe/QMcssQ5rHv9oAMy/hWHGyaES3vbxzqT2qMxI5bIJRpOJfDlTpAY5AVqrn\n'
|
'BfTLbe/QMcssQ5rHv9oAMy/hWHGyaES3vbxzqT2qMxI5bIJRpOJfDlTpAY5AVqrn\n'
|
||||||
'8sYx/1XA9YJOKFkQIQIDAQAB\n'
|
'8sYx/1XA9YJOKFkQIQIDAQAB\n'
|
||||||
'-----END PUBLIC KEY-----'},
|
'-----END PUBLIC KEY-----'},
|
||||||
'summary': ALICE_SUMMARY,
|
'summary': ALICE_SUMMARY_HTML,
|
||||||
'tag': [],
|
'tag': [],
|
||||||
'type': 'Person',
|
'type': 'Person',
|
||||||
'url': 'https://testserver/users/alice',
|
'url': 'https://testserver/users/alice',
|
||||||
|
@ -90,7 +91,7 @@ class Tests(TestCase):
|
||||||
posted_by = alice,
|
posted_by = alice,
|
||||||
)
|
)
|
||||||
|
|
||||||
c = Client()
|
c = BowlerClient()
|
||||||
response = c.get('/users/alice/featured')
|
response = c.get('/users/alice/featured')
|
||||||
self.assertEqual(response.status_code, 200)
|
self.assertEqual(response.status_code, 200)
|
||||||
result = _response_to_dict(response)
|
result = _response_to_dict(response)
|
||||||
|
@ -130,7 +131,7 @@ class TestTombstone(TestCase):
|
||||||
|
|
||||||
queen_anne = create_local_person('queen_anne')
|
queen_anne = create_local_person('queen_anne')
|
||||||
|
|
||||||
c = Client()
|
c = BowlerClient()
|
||||||
response = c.get('/users/queen_anne')
|
response = c.get('/users/queen_anne')
|
||||||
|
|
||||||
self.assertEqual(response.status_code, 200)
|
self.assertEqual(response.status_code, 200)
|
||||||
|
|
Ładowanie…
Reference in New Issue