From 04191ea1e6874fd9809f80f5567b3b9df911c8da Mon Sep 17 00:00:00 2001 From: Lucain Date: Tue, 19 Sep 2023 13:25:12 +0200 Subject: [PATCH] Fix gated repo tests (#26257) * Fix gated repo tests * Apply suggestions from code review --- tests/utils/test_hub_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/utils/test_hub_utils.py b/tests/utils/test_hub_utils.py index 89e41e4a07..5ff1ed679e 100644 --- a/tests/utils/test_hub_utils.py +++ b/tests/utils/test_hub_utils.py @@ -128,14 +128,14 @@ class GetFromCacheTests(unittest.TestCase): self.assertIsNone(get_file_from_repo(tmp_dir, "b.txt")) - @unittest.skip("Test is broken, fix me Wauplain!") def test_get_file_gated_repo(self): """Test download file from a gated repo fails with correct message when not authenticated.""" with self.assertRaisesRegex(EnvironmentError, "You are trying to access a gated repo."): - cached_file(GATED_REPO, README_FILE, use_auth_token=False) + # All files except README.md are protected on a gated repo. + cached_file(GATED_REPO, "gated_file.txt", use_auth_token=False) - @unittest.skip("Test is broken, fix me Wauplain!") def test_has_file_gated_repo(self): """Test check file existence from a gated repo fails with correct message when not authenticated.""" with self.assertRaisesRegex(EnvironmentError, "is a gated repository"): - has_file(GATED_REPO, README_FILE, use_auth_token=False) + # All files except README.md are protected on a gated repo. + has_file(GATED_REPO, "gated_file.txt", use_auth_token=False)