diff options
Diffstat (limited to 'src/leap/testing')
-rw-r--r-- | src/leap/testing/basetest.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/src/leap/testing/basetest.py b/src/leap/testing/basetest.py index 1ea26363..a55b0525 100644 --- a/src/leap/testing/basetest.py +++ b/src/leap/testing/basetest.py @@ -1,4 +1,5 @@ import os +import platform import shutil import tempfile @@ -7,6 +8,10 @@ try: except ImportError: import unittest +from leap.base.config import get_username, get_groupname + +_system = platform.system() + class BaseLeapTest(unittest.TestCase): @@ -43,5 +48,15 @@ class BaseLeapTest(unittest.TestCase): def get_tempfile(self, filename): return os.path.join(self.tempdir, filename) -if __name__ == "__main__": - unittest.main() + def get_username(self): + return get_username() + + def get_groupname(self): + return get_groupname() + + def _missing_test_for_plat(self, do_raise=False): + if do_raise: + raise NotImplementedError( + "This test is not implemented " + "for the running platform: %s" % + _system) |