1 #!/usr/bin/env python3 2 3 import os 4 import unittest 5 6 from sourcedr.codesearch import CodeSearch 7 from sourcedr.review_db import ReviewDB 8 9 10 TESTDATA_DIR = os.path.join(os.path.dirname(__file__), 'testdata') 11 ANDROID_DIR = os.path.join(TESTDATA_DIR, 'android_src') 12 13 14 class ReviewDBTest(unittest.TestCase): 15 def setUp(self): 16 self.csearch_index_path = 'csearchindex' 17 self.review_db_path = ReviewDB.DEFAULT_NAME 18 19 20 def tearDown(self): 21 os.remove(self.csearch_index_path) 22 os.remove(self.review_db_path) 23 24 25 def test_preprocess(self): 26 codesearch = CodeSearch(ANDROID_DIR, self.csearch_index_path) 27 codesearch.build_index() 28 review_db = ReviewDB(ReviewDB.DEFAULT_NAME, codesearch) 29 review_db.find(patterns=['dlopen'], is_regexs=[False]) 30 self.assertTrue(os.path.exists(ReviewDB.DEFAULT_NAME)) 31 32 33 if __name__ == '__main__': 34 unittest.main() 35