diff options
-rw-r--r-- | src/leap/testing/basetest.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/leap/testing/basetest.py b/src/leap/testing/basetest.py index ccf1a76f..1ea26363 100644 --- a/src/leap/testing/basetest.py +++ b/src/leap/testing/basetest.py @@ -26,12 +26,22 @@ class BaseLeapTest(unittest.TestCase): os.environ["PATH"] = cls.old_path shutil.rmtree(cls.tempdir) + # you have to override these methods + # this way we ensure we did not put anything + # here that you can forget to call. + def setUp(self): raise NotImplementedError("abstract base class") def tearDown(self): raise NotImplementedError("abstract base class") + # + # helper methods + # + + def get_tempfile(self, filename): + return os.path.join(self.tempdir, filename) if __name__ == "__main__": unittest.main() |