diff --git a/tests/unit/remote/test_gdrive.py b/tests/unit/remote/test_gdrive.py deleted file mode 100644 index 90b2936ec..000000000 --- a/tests/unit/remote/test_gdrive.py +++ /dev/null @@ -1,51 +0,0 @@ -import pytest - -from dvc.fs import GDriveAuthError, GDriveFileSystem - -USER_CREDS_TOKEN_REFRESH_ERROR = '{"access_token": "", "client_id": "", "client_secret": "", "refresh_token": "", "token_expiry": "", "token_uri": "https://oauth2.googleapis.com/token", "user_agent": null, "revoke_uri": "https://oauth2.googleapis.com/revoke", "id_token": null, "id_token_jwt": null, "token_response": {"access_token": "", "expires_in": 3600, "scope": "https://www.googleapis.com/auth/drive.appdata https://www.googleapis.com/auth/drive", "token_type": "Bearer"}, "scopes": ["https://www.googleapis.com/auth/drive", "https://www.googleapis.com/auth/drive.appdata"], "token_info_uri": "https://oauth2.googleapis.com/tokeninfo", "invalid": true, "_class": "OAuth2Credentials", "_module": "oauth2client.client"}' # noqa: E501 - -USER_CREDS_MISSED_KEY_ERROR = "{}" - - -class TestRemoteGDrive: - CONFIG = { - "url": "gdrive://root/data", - "gdrive_client_id": "client", - "gdrive_client_secret": "secret", - } - - def test_init(self, dvc): - fs = GDriveFileSystem( - gdrive_credentials_tmp_dir=dvc.tmp_dir, **self.CONFIG - ) - assert fs.url == self.CONFIG["url"] - - def test_drive(self, dvc, monkeypatch): - fs = GDriveFileSystem( - gdrive_credentials_tmp_dir=dvc.tmp_dir, **self.CONFIG - ) - monkeypatch.setenv( - GDriveFileSystem.GDRIVE_CREDENTIALS_DATA, - USER_CREDS_TOKEN_REFRESH_ERROR, - ) - with pytest.raises(GDriveAuthError): - assert fs.fs - - monkeypatch.setenv(GDriveFileSystem.GDRIVE_CREDENTIALS_DATA, "") - fs = GDriveFileSystem( - gdrive_credentials_tmp_dir=dvc.tmp_dir, **self.CONFIG - ) - monkeypatch.setenv( - GDriveFileSystem.GDRIVE_CREDENTIALS_DATA, - USER_CREDS_MISSED_KEY_ERROR, - ) - with pytest.raises(GDriveAuthError): - assert fs.fs - - def test_service_account_using_env_var(self, dvc, monkeypatch): - monkeypatch.setenv(GDriveFileSystem.GDRIVE_CREDENTIALS_DATA, "foo") - GDriveFileSystem( - gdrive_credentials_tmp_dir=dvc.tmp_dir, - gdrive_use_service_account=True, - **self.CONFIG - )