示例#1
0
    def to_representation(self, obj):
        data = super().to_representation(obj)
        request = self.context.get('request', None)
        # when 'wrap-outgoing-parameter' is on url is a flat string already
        is_flat_url = request and is_gate_active(request,
                                                 'wrap-outgoing-parameter')

        url = data.get('url')
        if not url:
            if obj.endpoint == Shelf.Endpoints.SEARCH:
                query_string = '&'.join(
                    f'{key}={value}'
                    for key, value in obj.get_param_dict().items())
                fallback = absolutify(
                    f'{reverse("search.search")}?{query_string}')
            elif obj.endpoint == Shelf.Endpoints.RANDOM_TAG:
                tag_page_url = reverse('tags.detail',
                                       kwargs={'tag_name': obj.tag})
                query_string = '&'.join(
                    f'{key}={value}'
                    for key, value in obj.get_param_dict().items()
                    if key != 'tag')
                fallback = absolutify(f'{tag_page_url}?{query_string}')
            elif obj.endpoint == Shelf.Endpoints.COLLECTIONS:
                fallback = absolutify(
                    reverse(
                        'collections.detail',
                        kwargs={
                            'user_id': str(settings.TASK_USER_ID),
                            'slug': obj.criteria,
                        },
                    ))
            else:
                # shouldn't happen
                fallback = None
            url = ({
                'url': fallback,
                'outgoing': fallback
            } if not is_flat_url else fallback)
        # text = data.get('text')

        if is_flat_url:
            return {
                **data,
                'url': url,
            }
        else:
            return {
                **data,
                'url': (url or {}).get('url'),
                'outgoing': (url or {}).get('outgoing'),
            }
示例#2
0
    def test_basic(self):
        file_upload = FileUpload(version='123', automated_signing=True)
        data = FileUploadSerializer(
            instance=file_upload, context={'request': self.request}
        ).data
        assert data == {
            'uuid': file_upload.uuid.hex,
            'channel': 'unlisted',
            'processed': False,
            'submitted': False,
            'url': settings.EXTERNAL_SITE_URL
            + reverse(
                'v5:addon-upload-detail',
                args=[file_upload.uuid.hex],
            ),
            'valid': False,
            'validation': None,
            'version': '123',
        }

        file_upload.update(
            automated_signing=False,
            validation=json.dumps([{'something': 'happened'}]),
        )
        data = FileUploadSerializer(
            instance=file_upload, context={'request': self.request}
        ).data
        assert data['channel'] == 'listed'
        assert data['validation'] == [{'something': 'happened'}]
示例#3
0
    def test_footer_url(self):
        wrap_outgoing_settings_override = {
            'DRF_API_GATES': {
                api_settings.DEFAULT_VERSION: ('wrap-outgoing-parameter',)
            }
        }
        # search
        self.search_rec_ext.update(footer_pathname='')
        data = self.serialize(self.search_rec_ext)
        url = (
            'http://testserver'
            + reverse('search.search')
            + '?promoted=recommended&sort=random&type=extension'
        )
        assert data['footer'] == {
            'url': url,
            'outgoing': url,
            'text': {'en-US': 'See more recommended extensions'},
        }
        with override_settings(**wrap_outgoing_settings_override):
            data = self.serialize(self.search_rec_ext)
            assert data['footer'] == {
                'url': url,
                'text': {'en-US': 'See more recommended extensions'},
            }
        # collections
        self.collections_shelf.update(footer_pathname='')
        data = self.serialize(self.collections_shelf)
        url = 'http://testserver' + reverse(
            'collections.detail',
            kwargs={
                'user_id': str(settings.TASK_USER_ID),
                'slug': 'privacy-matters',
            },
        )
        assert data['footer'] == {
            'url': url,
            'outgoing': url,
            'text': {'en-US': 'See more enhanced privacy extensions'},
        }
        with override_settings(**wrap_outgoing_settings_override):
            data = self.serialize(self.collections_shelf)
            assert data['footer'] == {
                'url': url,
                'text': {'en-US': 'See more enhanced privacy extensions'},
            }

        # random-tag
        self.tag_shelf.update(footer_pathname='')
        data = self.serialize(self.tag_shelf)
        url = 'http://testserver' + reverse('search.search') + '?tag=foo'
        assert data['footer'] == {
            'url': url,
            'outgoing': url,
            'text': {'en-US': 'See similar add-ons?'},
        }
        with override_settings(**wrap_outgoing_settings_override):
            data = self.serialize(self.tag_shelf)
            assert data['footer'] == {
                'url': url,
                'text': {'en-US': 'See similar add-ons?'},
            }