def test_static(self, mock_open): # request matches STATIC_URL, request handled url = 'file:///static/styles.css' with mock.patch('weasyprint.default_url_fetcher') as url_fetcher: data = django_url_fetcher(url) url_fetcher.assert_not_called() self.assertEqual( sorted(data.keys()), ['encoding', 'file_obj', 'filename', 'mime_type'], ) self.assertEqual(data['filename'], 'styles.css') self.assertEqual(data['mime_type'], 'text/css') self.assertEqual(data['encoding'], None)
def test_media(self, mock_storage): # request matches MEDIA_URL, request handled url = 'file:///media/image.jpg' with mock.patch('weasyprint.default_url_fetcher') as url_fetcher: data = django_url_fetcher(url) url_fetcher.assert_not_called() self.assertEqual( sorted(data.keys()), ['encoding', 'file_obj', 'filename', 'mime_type'], ) self.assertEqual(data['filename'], 'image.jpg') self.assertEqual(data['mime_type'], 'image/jpeg') self.assertEqual(data['encoding'], None)
def test_default(self): # MEDIA_URL='' and STATIC_URL=None, all requests passed though url = 'http://s3.amazon.test/images/image.jpg' with mock.patch('weasyprint.default_url_fetcher') as url_fetcher: django_url_fetcher(url) url_fetcher.assert_called_once_with(url) url = 'file:///media/image.jpg' with mock.patch('weasyprint.default_url_fetcher') as url_fetcher: django_url_fetcher(url) url_fetcher.assert_called_once_with(url) url = 'file:///static/styles.css' with mock.patch('weasyprint.default_url_fetcher') as url_fetcher: django_url_fetcher(url) url_fetcher.assert_called_once_with(url)