Added environment check.

1 files changed, 32 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index f0891d6..86156ed 100644
--- a/setup.py
+++ b/setup.py
@@ -2,9 +2,34 @@
 
 import distutils.core
 import logutils
+import os
 from os.path import join, dirname, abspath
 import re
 
+def missing_files():
+    result = []
+    if os.name == 'nt':
+
+        def found_file(fn):
+            if os.path.exists(fn):
+                return True
+            for d in os.environ['PATH'].split(os.pathsep):
+                p = os.path.join(d, fn)
+                if os.path.exists(p):
+                    return True
+            return False
+
+        FILES = ('cat.exe', 'echo.exe', 'tee.exe', 'false.exe', 'true.exe',
+                 'sleep.exe', 'touch.exe')
+        if 'APPVEYOR' not in os.environ:
+            FILES = ('libiconv2.dll', 'libintl3.dll') + FILES
+
+        missing = []
+        for fn in FILES:
+            if not found_file(fn):
+                result.append(fn)
+    return result
+
 
 def description():
     f = open(join(dirname(__file__), 'README.rst'))
@@ -20,6 +45,13 @@ class TestCommand(distutils.core.Command):
     user_options = []
 
     def run(self):
+        missing = missing_files()
+
+        if missing:
+            missing = ', '.join(missing)
+            raise ValueError('Can\'t find one or more files needed for '
+                             'tests: %s' % missing)
+
         import sys
         import unittest