Merge pull request #1 from andrewkroh/merge_773

Add Topbeat system test for process username

topbeat/tests/system/test_procs.py

@@ -1,7 +1,8 @@
 from topbeat import BaseTest
 
+import getpass
 import re
-
+import os
 
 """
 Contains tests for per process statistics.
@@ -31,6 +32,7 @@ def test_procs(self):
         assert re.match("(?i).*topbeat.test(.exe)? -e -c", output["proc.cmdline"])
         assert isinstance(output["proc.state"], basestring)
         assert isinstance(output["proc.cpu.start_time"], basestring)
+        self.check_username(output["proc.username"])
 
         for key in [
             "proc.pid",
@@ -49,3 +51,14 @@ def test_procs(self):
             "proc.mem.rss_p",
         ]:
             assert type(output[key]) in [int, float]
+
+    def check_username(self, observed, expected = None):
+        if expected == None:
+            expected = getpass.getuser()
+
+        if os.name == 'nt':
+            parts = observed.split("\\", 2)
+            assert len(parts) == 2, "Expected proc.username to be of form DOMAIN\username, but was %s" % observed
+            observed = parts[1]
+
+        assert expected == observed, "proc.username = %s, but expected %s" % (observed, expected)