def ensure_proxy_checks_if_package_is_already_cached(temp_dir):
temp_dir.create_directory("packages")
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
proxy_package_index._package_index = mock()
when(proxy_package_index._package_index).contains(
any_value(), any_value()).thenReturn(True)
proxy_package_index.get_package_content("pyassert", "0.2.5")
verify(proxy_package_index._package_index).contains("pyassert", "0.2.5")
def ensure_list_available_package_names_delegates_to_cached_index_when_failing_to_download_index(temp_dir):
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
temp_dir.touch("packages", "spam-0.1.2.tar.gz")
temp_dir.touch("packages", "eggs-0.1.2.tar.gz")
cached_environment = os.environ
os.environ = {}
when(pypiproxy.packageindex.urllib2).urlopen(
any_value()).thenRaise(URLError("Failed!"))
actual_list = proxy_package_index.list_available_package_names()
assert_that(actual_list).is_equal_to(['eggs', 'spam'])
os.environ = cached_environment
def ensure_list_available_package_names_retrieves_index_from_pypi(temp_dir):
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
package_stream = StringIO("""<!doctype html><html><body>
<a href='alpha'>alpha</a><br/>
<a href='beta'>beta</a><br/>
<a href='gamma'>gamma</a><br/>
</body></html>""")
cached_environment = os.environ
os.environ = {}
when(pypiproxy.packageindex.urllib2).urlopen(
any_value()).thenReturn(package_stream)
actual_list = proxy_package_index.list_available_package_names()
assert_that(actual_list).is_equal_to(['alpha', 'beta', 'gamma'])
verify(pypiproxy.packageindex.urllib2).urlopen(
"http://pypi.python.org/simple/")
os.environ = cached_environment
def ensure_list_versions_delegates_to_cached_versions_when_to_download_versions_from_pypi(temp_dir):
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
temp_dir.touch("packages", "spam-0.1.2.tar.gz")
temp_dir.touch("packages", "spam-2.3.4.tar.gz")
temp_dir.touch("packages", "spam-2.3.4.egg")
temp_dir.touch("packages", "spam-1.2.3.tar.gz")
temp_dir.touch("packages", "eggs-0.1.2.tar.gz")
temp_dir.touch("packages", "eggs-0.1.2.egg")
cached_environment = os.environ
os.environ = {}
when(pypiproxy.packageindex.urllib2).urlopen(
any_value()).thenRaise(URLError("Failed!"))
actual_list = proxy_package_index.list_versions("spam")
assert_that(actual_list).is_equal_to(['0.1.2', '1.2.3', '2.3.4'])
verify(pypiproxy.packageindex.urllib2).urlopen(
"http://pypi.python.org/simple/spam/")
os.environ = cached_environment
def ensure_proxy_gets_package_content_from_pypi_if_it_is_not_cached(temp_dir):
temp_dir.create_directory("packages")
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
proxy_package_index._package_index = mock()
package_content = mock()
when(proxy_package_index)._fetch_url(
any_value(), raw=any_value()).thenReturn(package_content)
when(proxy_package_index._package_index).contains(
any_value(), any_value()).thenReturn(False)
when(proxy_package_index._package_index).get_package_content(
any_value(), any_value()).thenReturn(package_content)
actual_package = proxy_package_index.get_package_content(
"pyassert", "0.2.5")
assert_that(actual_package).is_equal_to(package_content)
verify(proxy_package_index)._fetch_url(
"http://pypi.python.org/packages/source/p/pyassert/pyassert-0.2.5.tar.gz", raw=True)
verify(proxy_package_index._package_index).add_package(
"pyassert", "0.2.5", package_content)
verify(proxy_package_index._package_index).get_package_content(
"pyassert", "0.2.5")
def ensure_list_versions_retrieves_versions_from_pypi_with_md5_hash_in_href(temp_dir):
proxy_package_index = ProxyPackageIndex(
"cached", temp_dir.join("packages"), "http://pypi.python.org")
package_stream = StringIO("""<!doctype html><html><body>
<a href='package-0.1.2.tar.gz#md5=foobar'>package-0.1.2.tar.gz</a><br/>
<a href='package-1.2.3.tar.gz#md5=foobar'>package-1.2.3.tar.gz</a><br/>
<a href='package-1.2.3.egg#md5=foobar'>package-1.2.3.egg</a><br/>
<a href='package-2.3.4.egg#md5=foobar'>package-2.3.4.egg</a><br/>
<a href='package-2.3.4.tar.gz#md5=foobar'>package-2.3.4.tar.gz</a><br/>
<a href="package-3.01.tar.gz#md5=foobar" rel="download">3.01 download_url</a><br/>
<a href="package" rel="homepage">3.02 home_page</a><br/>
</body></html>""")
cached_environment = os.environ
os.environ = {}
when(pypiproxy.packageindex.urllib2).urlopen(
any_value()).thenReturn(package_stream)
actual_list = proxy_package_index.list_versions("package")
assert_that(actual_list).is_equal_to(['0.1.2', '1.2.3', '2.3.4', '3.01'])
verify(pypiproxy.packageindex.urllib2).urlopen(
"http://pypi.python.org/simple/package/")
os.environ = cached_environment