Add tests for ez_setup.py

2 files changed, 20 insertions, 0 deletions

diff --git a/conftest.py b/conftest.py
new file mode 100644
index 00000000..6e8557d0
--- /dev/null
+++ b/conftest.py
@@ -0,0 +1,10 @@
+import pytest
+
+
+@pytest.yield_fixture
+def tmpdir_as_cwd(tmpdir):
+    try:
+        orig = tmpdir.chdir()
+        yield tmpdir
+    finally:
+        orig.chdir()
diff --git a/test_ez_setup.py b/test_ez_setup.py
new file mode 100644
index 00000000..98e7d773
--- /dev/null
+++ b/test_ez_setup.py
@@ -0,0 +1,10 @@
+import ez_setup
+
+
+def test_download(tmpdir_as_cwd):
+    ez_setup.download_setuptools()
+    res, = tmpdir_as_cwd.listdir()
+    assert res.basename.startswith('setuptools-')
+    assert res.basename.endswith('.zip')
+    # file should be bigger than 64k
+    assert res.size() > 2**16