def test__classes(self):
filename = os.path.join(root_path(),
'applications/app1/modules/module_a.py')
module = Module(filename, root_path())
module.parse()
expect = ['ClassA', 'ClassAA']
self.assertEqual(module.classes(), expect)
return
def test__import_all_line(self):
filename = os.path.join(root_path(),
'applications/app1/modules/module_a.py')
module = Module(filename, root_path())
module.parse()
expect = 'from applications.app1.modules.module_a import *'
self.assertEqual(module.import_all_line(), expect)
# Test parse with script with sub directory
filename = os.path.join(root_path(),
'applications/app2/modules/adir/module_d.py'
)
module = Module(filename, root_path())
module.parse()
expect = 'from applications.app2.modules.adir.module_d import *'
self.assertEqual(module.import_all_line(), expect)
return
def test__parse(self):
filename = os.path.join(root_path(),
'applications/app1/modules/module_a.py')
module = Module(filename, root_path())
# Test parse with script without sub directory
module.parse()
self.assertEqual(module.application, 'app1')
self.assertEqual(module.sub_directory, '')
self.assertEqual(module.module, 'module_a')
# Test parse with script with sub directory
filename = os.path.join(root_path(),
'applications/app2/modules/adir/module_d.py'
)
module = Module(filename, root_path())
module.parse()
self.assertEqual(module.application, 'app2')
self.assertEqual(module.sub_directory, 'adir')
self.assertEqual(module.module, 'module_d')
return
