Skip to content

Commit

Permalink
refactor(bot): refactor bot.py logic into TrestleBot class
Browse files Browse the repository at this point in the history 
Signed-off-by: Jennifer Power <[email protected]>
  • Loading branch information
@jpower432
jpower432 committed Nov 7, 2023
1 parent e93a4e1 commit ec7abc0
Show file tree
Hide file tree
Showing 3 changed files with 284 additions and 213 deletions.
135 changes: 83 additions & 52 deletions tests/trestlebot/test_bot.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
from git import GitCommandError
from git.repo import Repo

import trestlebot.bot as bot
from trestlebot.bot import RepoException, TrestleBot
from trestlebot.provider import GitProvider, GitProviderException
from trestlebot.tasks.base_task import TaskBase, TaskException

Expand Down Expand Up @@ -60,6 +60,12 @@ def test_stage_files(
f.write("test,")

# Stage the files
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)
bot._stage_files(repo, file_patterns)

# Verify that files are staged
Expand All @@ -80,10 +86,14 @@ def test_local_commit(tmp_repo: Tuple[str, Repo]) -> None:
repo.index.add(test_file_path)

# Commit the test file
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)
commit_sha = bot._local_commit(
repo,
commit_user="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
)
assert commit_sha != ""
Expand All @@ -110,10 +120,16 @@ def test_local_commit_with_committer(tmp_repo: Tuple[str, Repo]) -> None:
repo.index.add(test_file_path)

# Commit the test file
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
author_name="Test Commit User",
author_email="[email protected]",
)
commit_sha = bot._local_commit(
repo,
commit_user="Test Commit User",
commit_email="[email protected]",
commit_message="Test commit message",
)

Expand Down Expand Up @@ -141,14 +157,18 @@ def test_local_commit_with_author(tmp_repo: Tuple[str, Repo]) -> None:
repo.index.add(test_file_path)

# Commit the test file
commit_sha = bot._local_commit(
repo,
commit_user="Test User",
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
)
commit_sha = bot._local_commit(
repo,
commit_message="Test commit message",
)
assert commit_sha != ""

# Verify that the commit is made
Expand Down Expand Up @@ -176,14 +196,16 @@ def test_run(tmp_repo: Tuple[str, Repo]) -> None:
mock_push.return_value = "Mocked result"

# Test running the bot
commit_sha, pr_number = bot.run(
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
)
commit_sha, pr_number = bot.run(
commit_message="Test commit message",
patterns=["*.txt"],
dry_run=False,
)
Expand Down Expand Up @@ -214,14 +236,14 @@ def test_run_dry_run(tmp_repo: Tuple[str, Repo]) -> None:
mock_push.return_value = "Mocked result"

# Test running the bot
commit_sha, pr_number = bot.run(
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)
commit_sha, pr_number = bot.run(
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
dry_run=True,
)
Expand All @@ -236,14 +258,14 @@ def test_empty_commit(tmp_repo: Tuple[str, Repo]) -> None:
repo_path, repo = tmp_repo

# Test running the bot
commit_sha, pr_number = bot.run(
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)
commit_sha, pr_number = bot.run(
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
dry_run=True,
)
Expand All @@ -260,18 +282,19 @@ def test_run_check_only(tmp_repo: Tuple[str, Repo]) -> None:
with open(test_file_path, "w") as f:
f.write("Test content")

bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)

with pytest.raises(
bot.RepoException,
RepoException,
match="Check only mode is enabled and diff detected. Manual intervention on main is required.",
):
_, _ = bot.run(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
dry_run=True,
check_only=True,
Expand Down Expand Up @@ -306,18 +329,19 @@ def test_run_with_exception(

repo.create_remote("origin", url="git.test.com/test/repo.git")

bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)

with patch("git.remote.Remote.push") as mock_push:
mock_push.side_effect = side_effect

with pytest.raises(bot.RepoException, match=msg):
with pytest.raises(RepoException, match=msg):
_ = bot.run(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
dry_run=False,
)
Expand All @@ -337,15 +361,16 @@ def test_run_with_failed_pre_task(tmp_repo: Tuple[str, Repo]) -> None:

repo.create_remote("origin", url="git.test.com/test/repo.git")

with pytest.raises(bot.RepoException, match="Bot pre-tasks failed: example"):
bot = TrestleBot(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
)

with pytest.raises(RepoException, match="Bot pre-tasks failed: example"):
_ = bot.run(
working_dir=repo_path,
branch="main",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
dry_run=True,
pre_tasks=[mock],
Expand All @@ -367,21 +392,24 @@ def test_run_with_provider(tmp_repo: Tuple[str, Repo]) -> None:

repo.create_remote("origin", url="git.test.com/test/repo.git")

bot = TrestleBot(
working_dir=repo_path,
branch="test",
commit_name="Test User",
commit_email="[email protected]",
author_name="The Author",
author_email="[email protected]",
target_branch="main",
)

with patch("git.remote.Remote.push") as mock_push:
mock_push.return_value = "Mocked result"

# Test running the bot
commit_sha, pr_number = bot.run(
working_dir=repo_path,
branch="test",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
git_provider=mock,
target_branch="main",
dry_run=False,
)
assert commit_sha != ""
Expand Down Expand Up @@ -423,21 +451,24 @@ def test_run_with_provider_with_custom_pr_title(tmp_repo: Tuple[str, Repo]) -> N

repo.create_remote("origin", url="git.test.com/test/repo.git")

bot = TrestleBot(
working_dir=repo_path,
branch="test",
commit_name="Test User",
commit_email="[email protected]",
author_name="The Author",
author_email="[email protected]",
target_branch="main",
)

with patch("git.remote.Remote.push") as mock_push:
mock_push.return_value = "Mocked result"

# Test running the bot
commit_sha = bot.run(
working_dir=repo_path,
branch="test",
commit_name="Test User",
commit_email="[email protected]",
commit_message="Test commit message",
author_name="The Author",
author_email="[email protected]",
patterns=["*.txt"],
git_provider=mock,
target_branch="main",
pull_request_title="Test",
dry_run=False,
)
Expand Down
Loading

0 comments on commit ec7abc0

Please sign in to comment.