Пример #1
0
 def test_uses_context_secrets(self, monkeypatch):
     github = MagicMock()
     monkeypatch.setattr("prefect.utilities.git.Github", github)
     with set_temporary_config({"cloud.use_local_secrets": True}):
         with prefect.context(secrets=dict(
                 GITHUB_ACCESS_TOKEN="ACCESS_TOKEN")):
             get_github_client()
     assert github.call_args[0][0] == "ACCESS_TOKEN"
Пример #2
0
    def test_creds_default_to_environment(self, monkeypatch):
        github = MagicMock()
        monkeypatch.setattr("prefect.utilities.git.Github", github)
        get_github_client()
        assert github.call_args[0][0] is None

        monkeypatch.setenv("GITHUB_ACCESS_TOKEN", "TOKEN")
        get_github_client()
        assert github.call_args[0][0] == "TOKEN"
Пример #3
0
 def test_prefers_passed_credentials_over_secrets(self, monkeypatch):
     github = MagicMock()
     monkeypatch.setattr("prefect.utilities.git.Github", github)
     desired_credentials = {"GITHUB_ACCESS_TOKEN": "PROVIDED_KEY"}
     with set_temporary_config({"cloud.use_local_secrets": True}):
         with prefect.context(secrets=dict(
                 GITHUB_ACCESS_TOKEN="ACCESS_TOKEN")):
             get_github_client(credentials=desired_credentials)
     assert github.call_args[0][0] == "PROVIDED_KEY"
Пример #4
0
    def _github_client(self):  # type: ignore
        from prefect.utilities.git import get_github_client

        return get_github_client()