def test_get_all_files(self): """Test get all files.""" self.settings['casefiles_dir'] = self.get_mocks_dir() + '/casefiles' load_casefiles = loaders.get_file_system_loader(self.settings) casefiles = load_casefiles() casefile_names = sorted([x.name() for x in casefiles]) expected_casefile_names = sorted(['uname', 'kelly']) self.assertEqual(casefile_names, expected_casefile_names)
def test_get_specific_file(self): """Test get a specifi file.""" self.settings['casefiles_dir'] = self.get_mocks_dir() + '/casefiles' load_casefiles = loaders.get_file_system_loader(self.settings) casefiles = load_casefiles(casefile_name='kelly') casefile_names = sorted([x.name() for x in casefiles]) expected_casefile_names = sorted(['kelly']) self.assertEqual(casefile_names, expected_casefile_names)