refactor load debugtalk.py module

This commit is contained in:
debugtalk
2018-08-08 18:23:42 +08:00
parent 33120fb183
commit a33384bb37
2 changed files with 59 additions and 52 deletions

View File

@@ -144,23 +144,39 @@ class TestFileLoader(unittest.TestCase):
class TestModuleLoader(unittest.TestCase):
def test_locate_debugtalk_py(self):
self.assertEqual(loader.locate_debugtalk_py(os.getcwd()), "tests.debugtalk")
with self.assertRaises(exceptions.FileNotFound):
loader.locate_debugtalk_py(os.getcwd())
start_dir_path = os.path.join(os.getcwd(), "tests")
with self.assertRaises(exceptions.FileNotFound):
loader.locate_debugtalk_py("")
start_path = os.path.join(os.getcwd(), "tests")
self.assertEqual(
loader.locate_debugtalk_py(start_dir_path),
"debugtalk"
loader.locate_debugtalk_py(start_path),
"tests.debugtalk"
)
start_dir_path = os.path.join(os.getcwd(), "not_exist")
self.assertEqual(
loader.locate_debugtalk_py(start_dir_path),
None
loader.locate_debugtalk_py("tests/"),
"tests.debugtalk"
)
self.assertEqual(
loader.locate_debugtalk_py("tests"),
"tests.debugtalk"
)
self.assertEqual(
loader.locate_debugtalk_py("tests/base.py"),
"tests.debugtalk"
)
self.assertEqual(
loader.locate_debugtalk_py("tests/data/test.env"),
"tests.debugtalk"
)
def test_load_debugtalk_module(self):
imported_module_items = loader.load_debugtalk_module("tests.debugtalk")
print(imported_module_items)
imported_module_items = loader.load_debugtalk_module()
self.assertEqual(imported_module_items, {})
imported_module_items = loader.load_debugtalk_module("tests")
self.assertEqual(
imported_module_items["variables"]["SECRET_KEY"],
"DebugTalk"
@@ -171,9 +187,6 @@ class TestModuleLoader(unittest.TestCase):
self.assertTrue(is_status_code_200(200))
self.assertFalse(is_status_code_200(500))
with self.assertRaises(exceptions.ParamsError):
loader.load_debugtalk_module("debugtalk")
class TestSuiteLoader(unittest.TestCase):