def main():
try:
opts, args = getopt.getopt(sys.argv[1:], "hd:e:", ["help", "basedir", "extension"])
except getopt.GetoptError as err:
print(str(err))
usage()
sys.exit(2)
extension = '*.java'
base_dir = os.path.dirname(os.path.abspath(__file__))
for o, a in opts:
if o in ("-h", "--help"):
usage()
sys.exit()
elif o in ("-d", "--basedir"):
base_dir = a
elif o in ("-e", "--extension"):
extension = a
else:
assert False, "unhandled option"
files = FileBrowser.get_files(basedir=base_dir, extension=extension)
parser = CodeParser()
for f in files:
parser.get_requirements_and_test_cases(f)
writer = MatrixWriter(parser.req_test_mapping)
writer.write()
def test_get_files(self):
files = FileBrowser.get_files(extension='*.java')
self.assertEqual(3, len(files))
