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)
