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()