fix for linux and unit test

sdpython · Aug 25, 2015 · 3315c6c · 3315c6c
1 parent cb23e5b
commit 3315c6c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/_unittests/ut_serverdoc/test_ping.py b/_unittests/ut_serverdoc/test_ping.py
@@ -25,14 +25,22 @@
 
 
 class TestPing(unittest.TestCase):
+
+    def get_machine(self):
+        if sys.platform.startswith("win"):
+            return os.environ["COMPUTERNAME"]
+        else:
+            r = os.environ.get("HOSTNAME", None)
+            if r is None:
+                raise Exception(str(os.environ))
 
     def test_ping(self):
         fLOG(
             __file__,
             self._testMethodName,
             OutputPrint=__name__ == "__main__")
 
-        machine = os.environ["COMPUTERNAME"]
+        machine = self.get_machine()
         out = ping_machine(machine, fLOG=fLOG)
         fLOG(out)
         assert len(out) > 0
@@ -43,7 +51,7 @@ def test_regular_ping(self):
             self._testMethodName,
             OutputPrint=__name__ == "__main__")
 
-        machine = os.environ["COMPUTERNAME"]
+        machine = self.get_machine()
         out = regular_ping_machine(machine, delay=0.1, nb_max=3, fLOG=fLOG)
         fLOG(out)
         assert len(out) > 0