def test_delete_sql_whenExists_shouldDeleteFromStore(self): file_name = os.path.join(TEST_FILES_PATH, "test_input.txt") self.assertTrue(os.path.isfile(file_name)) store = command_store_lib.SqlCommandStore(':memory:') command_store_lib.read_history_file(store, file_name, "doesntmatter", None, False) self.assertTrue(store.has_command_by_name("vim somefile.txt")) self.assertIsNotNone(store.delete_command('vim somefile.txt')) self.assertFalse(store.has_command_by_name("vim somefile.txt"))
def test_search_commands(self): file_name = os.path.join(TEST_FILES_PATH, "test_input.txt") store = command_store_lib.CommandStore() command_store_lib.read_history_file(store, file_name, "doesntmatter", None, False) matches = store.search_commands(["add"]) self.assertIsNotNone(matches) matches = store.search_commands(["add"], True) self.assertTrue(len(matches) == 0) matches = store.search_commands(["subl"], True) self.assertTrue(len(matches) == 1)
def test_readFile(self): file_name = os.path.join(TEST_FILES_PATH, "test_input.txt") store = command_store_lib.CommandStore() command_store_lib.read_history_file(store, file_name, "doesntmatter", None, False) self.assertTrue(store.has_command_by_name("vim somefile.txt")) self.assertTrue(store.has_command_by_name("rm somefile.txt")) self.assertTrue(store.has_command_by_name("whereis script")) self.assertTrue(store.has_command_by_name("vim /usr/bin/script")) self.assertFalse(store.has_command_by_name("vim somefil")) self.assertEqual( 2, store.get_command_by_name("rm somefile.txt").get_count_seen())
def generate_store_from_args(history_file_path, save_directory, use_json, use_sql): store_file_path = com_lib.get_file_path(save_directory, use_json, use_sql) commands_file_path = os.path.join(save_directory, com_lib.FILE_STORE_NAME) ignore_rule_file = os.path.join(save_directory, IGNORE_RULE_FILE_NAME) if not os.path.isfile(ignore_rule_file): ignore_rule_file = None else: print('Using ignore rules from ' + ignore_rule_file) store_type = com_lib.get_store_type(use_json, use_sql) store = com_lib.load_command_store(store_file_path, store_type) com_lib.read_history_file(store, history_file_path, commands_file_path, ignore_rule_file) print('Reading ' + history_file_path) if com_lib.save_command_store(store, store_file_path, use_json): print('Writing file out to ' + store_file_path)
def test_ignoreRule_whenFileDoestExist_shouldNotCrash(self): file_name = os.path.join(TEST_FILES_PATH, "test_input.txt") store = command_store_lib.CommandStore() command_store_lib.read_history_file(store, file_name, "doesntmatter", "test_files/fileNotthere.txt", False)