diff --git a/CHANGELOG.md b/CHANGELOG.md index cab46c1..eafba89 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -4,6 +4,11 @@ All notable changes to this project will be documented in this file. The format is based on [Keep a Changelog](https://keepachangelog.com), and this project adheres to [Semantic Versioning](https://semver.org). + +## [3.7.0] - 2023-10-18 +### Added +- Added support to pass in a path when calling get_files(), resolves issue #34 + ## [3.6.0] - 2023-10-06 ### Fix - Fetch repo content from target branch diff --git a/gordian/repo.py b/gordian/repo.py index ae72642..0c53a0b 100644 --- a/gordian/repo.py +++ b/gordian/repo.py @@ -77,9 +77,9 @@ def get_objects(self, filename, klass=None): return PlainTextFile(file, self) - def get_files(self): + def get_files(self, path=''): if not self.files: - contents = self._get_repo_contents('') + contents = self._get_repo_contents(path) while contents: file = contents.pop(0) diff --git a/tests/test_repo.py b/tests/test_repo.py index 4398ec0..a341ea1 100644 --- a/tests/test_repo.py +++ b/tests/test_repo.py @@ -191,6 +191,16 @@ def test_delete_file(self, mock_git): repo._source_repo.delete_file.assert_called_once() self.assertTrue(repo.dirty) + def test_get_files_with_path(self): + self.repo._set_target_branch('target') + self.repo.files = [] + self.repo._source_repo = MagicMock() + repository_file = MagicMock(path='test/afile.txt', type='not_dir') + self.repo._source_repo.get_contents.side_effect = [[MagicMock(path='directory', type='dir')],[repository_file]] + self.repo.get_files('test') + self.repo._source_repo.get_contents.assert_has_calls([call('test', 'target'), call('directory', 'target')]) + self.assertEquals(self.repo.files, [repository_file]) + def test__get_github_client(self): repo = Repo('test_repo', branch='', github=self.mock_git)