def test_find_job_trackers_in_folder(self): config_dir = ConfigDirectory() script_mock = MagicMock() script_mock.return_value = ['parent_directory/foo', 'parent_directory/foo.conf', 'parent_directory/foo_pipeline', 'parent_directory/foo_pipeline.conf'] config_dir.get_files_in_directory = script_mock trackers = config_dir.find_job_trackers_in_folder('parent_directory') expected_trackers = ['parent_directory/foo_pipeline.conf'] self.assertEqual(trackers, expected_trackers)
def test_get_files_in_directory(self, os_mock): config_dir = ConfigDirectory() os_mock.path.join = os.path.join os_mock.listdir.return_value = ['file_foo', 'dir_bar', 'link_baz'] os_mock.path.isfile.side_effect = self.isfile files = config_dir.get_files_in_directory('parent_directory') expected_files = ['parent_directory/file_foo', 'parent_directory/link_baz'] self.assertItemsEqual(files, expected_files) os_mock.listdir.assert_called_with('parent_directory')