def test_custom_query_parameter(self):
request = Request(APIRequestFactory().get('/thirds/', {'pony': 'extra'}))
serializer = ThirdSerializer(context={'request': request})
adapter = ExpanderAdapter(serializer)
parser = ExpanderParser(adapter)
parser.expansion_key = 'pony'
expander = parser.parse()
self.assertEqual(1, self.child_count(expander))
self.assertEqual(0, self.child_count(expander, 'extra'))
def parse_expand(self, expand=None, settings=dict()):
"""
Helper method for running the parser.
"""
params = dict()
if expand is not None:
params['expand'] = expand
request = Request(APIRequestFactory().get('/thirds/', params))
serializer = ThirdSerializer(context={'request': request})
adapter = ExpanderAdapter(serializer)
parser = ExpanderParser(adapter)
for key, value in six.iteritems(settings):
setattr(parser, key, value)
return parser.parse()
