Pass tokens

tests/test_run.py

@@ -1,4 +1,5 @@
 import json
+import os
 from pathlib import Path
 import subprocess
 from typing import Any, Dict
@@ -8,6 +9,7 @@
 from sweagent.agent.agents import Agent, AgentArguments, AgentHook
 from sweagent.agent.models import ModelArguments
 from sweagent.environment.swe_env import EnvironmentArguments, SWEEnv
+from unittest.mock import Mock
 
 def test_run_cli_help():
     args = [
@@ -23,7 +25,8 @@ def test_run_cli_help():
 def open_pr_hook_init_for_sop():
     hook = OpenPRHook()
     hook._token = ""
-    hook._env = None
+    hook._env = Mock()
+    hook._env._github_token = os.environ.get("GITHUB_TOKEN", "")
     hook._data_path = "https://github.com/klieret/swe-agent-test-repo/issues/1"
     hook._open_pr = True
     hook._skip_if_commits_reference_issue = True

tests/test_utils.py

@@ -1,8 +1,12 @@
 import hashlib
+import os
 import subprocess
 import pytest
 from sweagent.environment.utils import InvalidGithubURL, format_trajectory_markdown, _MARKDOWN_TRAJECTORY_EMOJI_MAPPING, get_instances, is_github_repo_url, remove_triple_backticks, parse_gh_repo_url, parse_gh_issue_url, is_github_issue_url, get_associated_commit_urls
 
+
+_TOKEN = {"token": os.environ.get("GITHUB_TOKEN", "")}
+
 def test_format_trajectory_markdown(test_trajectory):
     formatted = format_trajectory_markdown(test_trajectory["trajectory"])
     assert formatted.startswith("<details>")
@@ -92,6 +96,7 @@ def test_get_instance_gh_issue_local_repo(tmp_path):
     instance = get_instances(
         file_path="https://github.com/klieret/swe-agent-test-repo/issues/1",
         repo_path=str(tmp_path / "swe-agent-test-repo"),
+        **_TOKEN
     )[0]
     compare_with = {
         'repo': str(tmp_path.resolve() / "swe-agent-test-repo"),
@@ -129,6 +134,7 @@ def test_get_instance_gh_issue_gh_repo(tmp_path):
     instance = get_instances(
         file_path="https://github.com/klieret/swe-agent-test-repo/issues/1",
         repo_path="https://github.com/princeton-nlp/SWE-agent",
+        **_TOKEN
     )[0]
     compare_with = {
         'repo': "princeton-nlp/SWE-agent",
@@ -146,6 +152,7 @@ def test_get_instance_text_issue_gh_repo(tmp_path):
     instance = get_instances(
         file_path="text://this is a test",
         repo_path="https://github.com/princeton-nlp/SWE-agent",
+        **_TOKEN
     )[0]
     compare_with = {
         'repo': "princeton-nlp/SWE-agent",
@@ -162,4 +169,4 @@ def test_load_instances(test_data_path, caplog):
     test_data_sources = test_data_path / "data_sources"
     examples = list(test_data_sources.iterdir())
     for example in examples:
-        get_instances(file_path=str(example))
+        get_instances(file_path=str(example), **_TOKEN)