From 968bcb4580903a66ffa35dbb4266f815853eb8f7 Mon Sep 17 00:00:00 2001
From: Paul Moore <p.f.moore@gmail.com>
Date: Tue, 24 Apr 2018 16:00:14 +0100
Subject: [PATCH] Add a very basic test suite

---
 dev-requirements.txt  |  3 +++
 tests/test_install.py | 51 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 dev-requirements.txt
 create mode 100644 tests/test_install.py

diff --git a/dev-requirements.txt b/dev-requirements.txt
new file mode 100644
index 00000000..9a9fc0da
--- /dev/null
+++ b/dev-requirements.txt
@@ -0,0 +1,3 @@
+pytest
+virtualenv
+scripttest
diff --git a/tests/test_install.py b/tests/test_install.py
new file mode 100644
index 00000000..fcafa6c8
--- /dev/null
+++ b/tests/test_install.py
@@ -0,0 +1,51 @@
+import os
+import sys
+import scripttest
+
+# Locate get-pip.py relative to this test file
+GET_PIP = os.path.join(os.path.dirname(os.path.dirname(__file__)),
+                       'get-pip.py')
+
+# Handle platform specific differences
+if sys.platform.startswith('win'):
+    BIN_DIR = 'Scripts'
+    def exe_base(path):
+        """Command name of an executable"""
+        return os.path.splitext(os.path.basename(path))[0]
+else:
+    BIN_DIR = 'bin'
+    def exe_base(path):
+        """Command name of an executable"""
+        return os.path.basename(path)
+
+
+def test_install(tmpdir):
+    """Simple test that installing pip using get-pip works"""
+
+    # Create a scripttest environment
+    env = scripttest.TestFileEnvironment(tmpdir / 'env')
+
+    # Create a virtual environment with no pip present.
+    # TODO: add the following cases
+    #   1. environment name with spaces
+    #   2. environment with some version of pip already present
+
+    ve_name = 've'
+    env.run('virtualenv', '--no-pip', '--no-setuptools', '--no-wheel',
+            ve_name)
+    ve_python = os.path.join(ve_name, BIN_DIR, 'python')
+
+    # Run get-pip.py from this repository
+    result = env.run(ve_python, GET_PIP)
+
+    # Check if the pip command was created
+    created_pip = None
+    for name, file_obj in result.files_created.items():
+        if exe_base(name) == 'pip' and file_obj.file:
+            created_pip = file_obj
+            break
+    assert created_pip
+
+    # Confirm that both the pip exe and python -m pip work
+    env.run(created_pip.full, '--version')
+    env.run(ve_python, '-m', 'pip', '--version')