From c04aa203de0c02e2ec07b8ae028f991905f6483d Mon Sep 17 00:00:00 2001
From: Johannes Bechberger <me@mostlynerdless.de>
Date: Sun, 30 Oct 2016 16:59:10 +0100
Subject: [PATCH] Fix installation and Windows issues

---
 mjtest/__init__.py      |  1 +
 mjtest/cli.py           |  6 ++++--
 mjtest/environment.py   |  5 ++---
 mjtest/test/__init__.py |  1 -
 mjtest/util/shell.py    | 21 ++++++++++++++++++---
 mjtest/util/utils.py    |  2 +-
 6 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/mjtest/__init__.py b/mjtest/__init__.py
index af5ffa1..e11b829 100644
--- a/mjtest/__init__.py
+++ b/mjtest/__init__.py
@@ -1 +1,2 @@
+
 VERSION = "0.42"
\ No newline at end of file
diff --git a/mjtest/cli.py b/mjtest/cli.py
index 7b9ff58..e525721 100644
--- a/mjtest/cli.py
+++ b/mjtest/cli.py
@@ -2,9 +2,11 @@ import logging
 import os
 from pprint import pprint
 import sys
-import mjtest.util.utils
+from os.path import dirname
+sys.path.append(dirname(__file__))
+import util.utils
 import argparse
-from mjtest.environment import TestMode, Environment, TEST_MODES
+from environment import TestMode, Environment, TEST_MODES
 from mjtest.test.tests import TestSuite
 
 # adapted from http://stackoverflow.com/a/8527629
diff --git a/mjtest/environment.py b/mjtest/environment.py
index ab21eba..9c06243 100644
--- a/mjtest/environment.py
+++ b/mjtest/environment.py
@@ -55,16 +55,15 @@ class Environment:
         self.parallel = parallel
         self.timeout = timeout
 
-        if tmp_dir:
+        if report_dir:
             self.report_dir = os.path.abspath(os.path.expandvars(report_dir))
             if not os.path.exists(report_dir):
                 os.mkdir(self.report_dir)
         else:
             self.report_dir = os.path.join(get_mjtest_basedir(), "reports")
-
             if not os.path.exists(self.report_dir):
                 os.mkdir(self.report_dir)
-        self.report_dir = os.path.join(self.report_dir, datetime.now().strftime("%d.%m.%y:%X"))
+        self.report_dir = os.path.join(self.report_dir, datetime.now().strftime("%d-%m-%y_%H-%M-%S"))
         os.mkdir(self.report_dir)
         logging.basicConfig(level=self.LOG_LEVELS[log_level])
 
diff --git a/mjtest/test/__init__.py b/mjtest/test/__init__.py
index 81c1436..e69de29 100644
--- a/mjtest/test/__init__.py
+++ b/mjtest/test/__init__.py
@@ -1 +0,0 @@
-__author__ = 'parttimenerd'
diff --git a/mjtest/util/shell.py b/mjtest/util/shell.py
index 90cb9f2..a65ddaa 100644
--- a/mjtest/util/shell.py
+++ b/mjtest/util/shell.py
@@ -5,7 +5,13 @@ Convenience function
 Alternative to subprocess and os.system
 """
 import subprocess
-import resource
+has_resource_module = True
+try:
+	import resource
+except ImportError:
+	has_resource_module = False
+	pass
+	
 import sys
 import signal
 import threading
@@ -24,6 +30,8 @@ class SigKill(Exception):
 
 
 def _lower_rlimit(res, limit):
+    if not has_resource_module:
+        return
     (soft, hard) = resource.getrlimit(res)
     if soft > limit or soft == resource.RLIM_INFINITY:
         soft = limit
@@ -51,6 +59,8 @@ class _Execute(object):
             rlimit['RLIMIT_FSIZE'] = 512 * MB
 
     def _set_rlimit(self):
+        if not has_resource_module:
+            return
         if self.timeout > 0.0:
             _lower_rlimit(resource.RLIMIT_CPU, self.timeout)
         for k,v in self.rlimit.items():
@@ -58,11 +68,14 @@ class _Execute(object):
 
     def _run_process(self):
         try:
+            prexec_args = {}
+            if has_resource_module:
+                prexec_args["preexec_fn"] = self._set_rlimit
             self.proc = subprocess.Popen(self.cmd,
                                          stdout=subprocess.PIPE,
                                          stderr=subprocess.PIPE,
-                                         preexec_fn=self._set_rlimit,
-                                         env=self.env)
+                                         env=self.env,
+                                         **prexec_args)
             x = self.proc.communicate()
             self.out, self.err = x
             self.returncode = self.proc.returncode
@@ -107,6 +120,8 @@ def execute(cmd, env=None, timeout=0, rlimit=None, propagate_sigint=True):
         rlimit = dict()
     if cmd is str:
         cmd = filter(lambda x: x, cmd.split(' '))
+    else:
+        cmd = list(filter(lambda x: x, cmd[0].split(' '))) + cmd[1:]
     exc = _Execute(cmd, timeout, env, rlimit)
     (out, err, returncode) = exc.run()
     if returncode == -signal.SIGINT:
diff --git a/mjtest/util/utils.py b/mjtest/util/utils.py
index 24a4e5c..09059a0 100644
--- a/mjtest/util/utils.py
+++ b/mjtest/util/utils.py
@@ -5,7 +5,7 @@ import sys
 
 
 def get_mjtest_basedir() -> str:
-    return path.dirname(path.dirname(path.dirname(path.abspath(__file__))))
+    return path.dirname(path.dirname(path.dirname(path.realpath(__file__))))
 
 
 """ Colored logging handler that is used for the root logger """
-- 
GitLab