add unit test and throw exception on timeout

Signed-off-by: zliu3 <[email protected]>

gordian/repo.py

@@ -137,6 +137,7 @@ def _get_repo_contents(self, path):
             logger.info(f'Error fetching repo contents: {e}')
         except TimeoutError as e:
             logger.info(f'Error fetching repo contents: {e}')
+            raise e
 
     @retry(GithubException, tries=3, delay=1, backoff=2)
     def _make_branch(self):

tests/test_repo.py

@@ -206,3 +206,12 @@ def test__get_github_client(self):
 
         self.assertIsNotNone(repo.get_github_client())
         self.assertEqual(repo.get_github_client(), self.mock_git)
+
+    def test_get_repo_contents(self):
+        self.repo._set_target_branch('target')
+        self.repo.files = []
+        self.repo._source_repo = MagicMock()
+        self.repo._source_repo.get_contents.side_effect = TimeoutError('Read Timeout')
+        with pytest.raises(Exception) as context:
+            self.repo._get_repo_contents(path='test/afile.txt')
+        assert "Read Timeout" in str(context.value)