Python download_package Examples

Example #1

 def test_download_fallback_cache(self, fetch_dist):
     """Downloading missing package caches result from fallback"""
     db = self.request.db = MagicMock()
     locator = self.request.locator = MagicMock()
     self.request.registry.fallback = "cache"
     self.request.fallback_simple = "https://pypi.org/simple"
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), MagicMock())
     context = MagicMock()
     context.filename = "package.tar.gz"
     url = "https://pypi.org/simple/%s" % context.filename
     dist = make_dist(url=url)
     locator.get_releases.return_value = [dist]
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(
         self.request,
         dist["url"],
         dist["name"],
         dist["version"],
         dist["summary"],
         dist["requires_python"],
     )
     self.assertEqual(ret.body, fetch_dist()[1])
     ret.headers.update.assert_any_call([("Cache-Control",
                                          "public, max-age=0")])

Example #2

File: test_api.py Project: jwfh/pypicloud

 def test_download_fallback_cache_max_age(self, fetch_dist):
     """ Downloading missing package caches result from fallback """
     db = self.request.db = MagicMock()
     locator = self.request.locator = MagicMock()
     self.request.registry.fallback = "cache"
     self.request.fallback_simple = "https://pypi.org/simple"
     self.request.access.can_update_cache.return_value = True
     self.request.registry.package_max_age = 30
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), MagicMock())
     context = MagicMock()
     context.filename = "package.tar.gz"
     dist = MagicMock()
     url = "https://pypi.org/simple/%s" % context.filename
     locator.get_project.return_value = {
         "0.1": dist,
         "urls": {
             "0.1": set([url])
         }
     }
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(self.request, dist.name, url)
     self.assertEqual(ret.body, fetch_dist()[1])
     ret.headers.update.assert_any_call([("Cache-Control",
                                          "public, max-age=30")])

Example #3

File: test_api.py Project: stylight/pypicloud

 def test_download(self):
     """ Downloading package returns download response from db """
     db = self.request.db = MagicMock()
     context = MagicMock()
     ret = api.download_package(context, self.request)
     db.fetch.assert_called_with(context.filename)
     db.download_response.assert_called_with(db.fetch())
     self.assertEqual(ret, db.download_response())

Example #4

File: test_api.py Project: johtso/pypicloud

 def test_download_fallback_no_cache(self):
     """ Downloading missing package on non-'cache' fallback returns 404 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'none'
     db.fetch.return_value = None
     context = MagicMock()
     ret = api.download_package(context, self.request)
     self.assertEqual(ret.status_code, 404)

Example #5

File: test_api.py Project: stylight/pypicloud

 def test_download_fallback_no_cache(self):
     """ Downloading missing package on non-'cache' fallback returns 404 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'none'
     db.fetch.return_value = None
     context = MagicMock()
     ret = api.download_package(context, self.request)
     self.assertEqual(ret.status_code, 404)

Example #6

File: test_api.py Project: johtso/pypicloud

 def test_download(self):
     """ Downloading package returns download response from db """
     db = self.request.db = MagicMock()
     context = MagicMock()
     ret = api.download_package(context, self.request)
     db.fetch.assert_called_with(context.filename)
     db.download_response.assert_called_with(db.fetch())
     self.assertEqual(ret, db.download_response())

Example #7

File: test_api.py Project: stylight/pypicloud

 def test_download_fallback_cache_no_perm(self):
     """ Downloading missing package without cache perm returns 403 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.access.can_update_cache.return_value = False
     db.fetch.return_value = None
     context = MagicMock()
     ret = api.download_package(context, self.request)
     self.assertEqual(ret, self.request.forbid())

Example #8

File: test_api.py Project: johtso/pypicloud

 def test_download_fallback_cache_no_perm(self):
     """ Downloading missing package without cache perm returns 403 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.access.can_update_cache.return_value = False
     db.fetch.return_value = None
     context = MagicMock()
     ret = api.download_package(context, self.request)
     self.assertEqual(ret, self.request.forbid())

Example #9

 def test_download_fallback_cache_missing(self, locator):
     """ If fallback url is missing dist, return 404 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     context = MagicMock()
     locator().get_project.return_value = {context.filename: None}
     ret = api.download_package(context, self.request)
     self.assertEqual(ret.status_code, 404)

Example #10

 def test_download_with_stream_files(self):
     """Downloading package returns download response from db with max age"""
     db = self.request.db = MagicMock()
     self.request.registry.stream_files = True
     self.request.registry.package_max_age = 30
     context = MagicMock()
     ret = api.download_package(context, self.request)
     db.fetch.assert_called_with(context.filename)
     db.storage.open.assert_called_once_with(db.fetch())
     db.download_response.assert_not_called()
     ret.headers.update.assert_any_call([("Cache-Control",
                                          "public, max-age=30")])

Example #11

File: test_api.py Project: johtso/pypicloud

 def test_download_fallback_cache_missing(self, locator):
     """ If fallback url is missing dist, return 404 """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     context = MagicMock()
     locator().get_project.return_value = {
         context.filename: None
     }
     ret = api.download_package(context, self.request)
     self.assertEqual(ret.status_code, 404)

Example #12

 def test_download_with_stream_files(self):
     """Downloading package returns download response from db with max age"""
     db = self.request.db = MagicMock()
     data = MagicMock()
     db.storage.open.return_value.__enter__.return_value = data
     data.read.return_value = b""
     self.request.registry.stream_files = True
     self.request.registry.package_max_age = 30
     context = MagicMock()
     ret = api.download_package(context, self.request)
     db.fetch.assert_called_with(context.filename)
     db.storage.open.assert_called_once_with(db.fetch())
     db.download_response.assert_not_called()
     self.assertDictContainsSubset({"Cache-Control": "public, max-age=30"},
                                   ret.headers)

Example #13

File: test_api.py Project: johtso/pypicloud

 def test_download_fallback_cache(self, locator, fetch_dist):
     """ Downloading missing package caches result from fallback """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), MagicMock())
     context = MagicMock()
     dist = MagicMock(spec=Distribution)
     locator().get_project.return_value = {
         context.filename: dist,
     }
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(self.request, dist)
     self.assertEqual(ret.body, fetch_dist()[1])

Example #14

 def test_download_fallback_cache(self, locator, fetch_dist):
     """ Downloading missing package caches result from fallback """
     db = self.request.db = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     context = MagicMock()
     dist = MagicMock(spec=Distribution)
     locator().get_project.return_value = {
         context.filename: dist,
     }
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(self.request, dist)
     db.download_response.assert_called_with(fetch_dist())
     self.assertEqual(ret, db.download_response())

Example #15

 def test_download_fallback_cache_max_age(self, fetch_dist):
     """Downloading missing package caches result from fallback"""
     db = self.request.db = MagicMock()
     locator = self.request.locator = MagicMock()
     self.request.registry.fallback = "cache"
     self.request.fallback_simple = "https://pypi.org/simple"
     self.request.access.can_update_cache.return_value = True
     self.request.registry.package_max_age = 30
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), BytesIO(b"fds"))
     context = MagicMock()
     context.filename = "package.tar.gz"
     url = "https://pypi.org/simple/%s" % context.filename
     dist = make_dist(url=url)
     locator.get_releases.return_value = [dist]
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_once()
     self.assertEqual(ret.body, b"fds")
     self.assertDictContainsSubset({"Cache-Control": "public, max-age=30"},
                                   ret.headers)

Example #16

File: test_api.py Project: stylight/pypicloud

 def test_download_fallback_cache(self, fetch_dist):
     """ Downloading missing package caches result from fallback """
     db = self.request.db = MagicMock()
     locator = self.request.locator = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), MagicMock())
     context = MagicMock()
     context.filename = 'package.tar.gz'
     dist = MagicMock()
     url = 'http://pypi.com/simple/%s' % context.filename
     locator.get_project.return_value = {
         '0.1': dist,
         'urls': {
             '0.1': set([url])
         }
     }
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(self.request, dist.name, url)
     self.assertEqual(ret.body, fetch_dist()[1])

Example #17

File: test_api.py Project: Hexadite/pypicloud-hexadite

 def test_download_fallback_cache(self, fetch_dist):
     """ Downloading missing package caches result from fallback """
     db = self.request.db = MagicMock()
     locator = self.request.locator = MagicMock()
     self.request.registry.fallback = 'cache'
     self.request.registry.fallback_url = 'http://pypi.com'
     self.request.access.can_update_cache.return_value = True
     db.fetch.return_value = None
     fetch_dist.return_value = (MagicMock(), MagicMock())
     context = MagicMock()
     context.filename = 'package.tar.gz'
     dist = MagicMock()
     url = 'http://pypi.com/simple/%s' % context.filename
     locator.get_project.return_value = {
         '0.1': dist,
         'urls': {
             '0.1': set([url])
         }
     }
     ret = api.download_package(context, self.request)
     fetch_dist.assert_called_with(self.request, dist.name, url)
     self.assertEqual(ret.body, fetch_dist()[1])