diff --git a/main.py b/main.py index f30f588908ffee1e331c11a62cdc0edd359612a5..f55b3e8d7ad8a35eff15d5c02533054e84881fe0 100644 --- a/main.py +++ b/main.py @@ -102,6 +102,8 @@ if __name__ == '__main__': parser.add_argument('-repo', action = 'store_true', help='Whether to analyze repo packages.') parser.add_argument('-local', action = 'store_true', help='Whether to analyze local installed packages.') parser.add_argument('-console_log', action = 'store_true', help='Output log to console.') + rpmquery = parser.add_argument_group('rpmquery') + rpmquery.add_argument('-q_rpminiso', type=str, help='Query input rpm layer in iso.') args = parser.parse_args() if len(sys.argv) == 1: @@ -121,5 +123,7 @@ if __name__ == '__main__': kyClassifier.process_repo() if args.local: kyClassifier.process_local() + if args.q_rpminiso: + pass logger.info("Kyclassifier end!") diff --git a/src/utils/exceptions.py b/src/utils/exceptions.py index 745472d2ed9cb002c394ed8631c4e9fa269d3502..dcfc6fc6615e70c0ed4308afe6014c8b7de34709 100644 --- a/src/utils/exceptions.py +++ b/src/utils/exceptions.py @@ -31,3 +31,12 @@ class ReportGenerateError(Exception): def __str__(self): return f"ReportGenerateError: {self.message}" + +class RpmCheckError(Exception): + """Rpm check exception + """ + def __init__(self, message): + self.message = message + + def __str__(self): + return f"RpmCheckError: {self.message}" \ No newline at end of file diff --git a/test/test_algclassify.py b/test/test_algclassify.py index ca9d9d6d1c8a97c7126533db0938b30c9de98901..35152891107e10fa6b351d3743f80ab3298d39a5 100644 --- a/test/test_algclassify.py +++ b/test/test_algclassify.py @@ -157,7 +157,16 @@ class TestAlgClassify(unittest.TestCase): self.assertEqual(result['aaa-devel'], ['其它'], "_merge_pkg2category_dict test failed!") def test_run(self): - pass + """ + Test class AlgClassify method run() + Returns: + dict + """ + if not os.path.exists(self.data_f): + self.skipTest("Test File not exists, test skiped!") + else: + result = AlgClassify.run(self.data_obj,self.data_f) + self.assertIsInstance(result,dict,"run test failed!") if __name__ == "__main__": unittest.main(verbosity=2) \ No newline at end of file