xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-05 13:20:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add unit tests for dynamic context handling in extend_patch function

Browse Source
This commit is contained in:
mrT23
2024-08-19 10:33:47 +03:00
parent fc40ca9196
commit 8fd8d298e7
1 changed files with 30 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tests/unittest/test_extend_patch.py
View File

@ -2,9 +2,13 @@ import pytest
from pr_agent.algo.git_patch_processing import extend_patch from pr_agent.algo.git_patch_processing import extend_patch
from pr_agent.algo.pr_processing import pr_generate_extended_diff from pr_agent.algo.pr_processing import pr_generate_extended_diff
from pr_agent.algo.token_handler import TokenHandler from pr_agent.algo.token_handler import TokenHandler
from pr_agent.config_loader import get_settings
class TestExtendPatch: class TestExtendPatch:
def setUp(self):
get_settings().config.allow_dynamic_context = False
# Tests that the function works correctly with valid input # Tests that the function works correctly with valid input
def test_happy_path(self): def test_happy_path(self):
original_file_str = 'line1\nline2\nline3\nline4\nline5' original_file_str = 'line1\nline2\nline3\nline4\nline5'
@ -61,8 +65,34 @@ class TestExtendPatch:
patch_extra_lines_before=num_lines, patch_extra_lines_after=num_lines) patch_extra_lines_before=num_lines, patch_extra_lines_after=num_lines)
assert actual_output == expected_output assert actual_output == expected_output
def test_dynamic_context(self):
get_settings().config.max_extra_lines_before_dynamic_context = 10
original_file_str = "def foo():"
for i in range(9):
original_file_str += f"\n line({i})"
patch_str ="@@ -11,1 +11,1 @@ def foo():\n- line(9)\n+ new_line(9)"
num_lines=1
get_settings().config.allow_dynamic_context = True
actual_output = extend_patch(original_file_str, patch_str,
patch_extra_lines_before=num_lines, patch_extra_lines_after=num_lines)
expected_output='\n@@ -1,10 +1,10 @@ \n def foo():\n line(0)\n line(1)\n line(2)\n line(3)\n line(4)\n line(5)\n line(6)\n line(7)\n line(8)\n- line(9)\n+ new_line(9)'
assert actual_output == expected_output
get_settings().config.allow_dynamic_context = False
actual_output2 = extend_patch(original_file_str, patch_str,
patch_extra_lines_before=num_lines, patch_extra_lines_after=num_lines)
expected_output_no_dynamic_context = '\n@@ -10,1 +10,1 @@ def foo():\n line(8)\n- line(9)\n+ new_line(9)'
assert actual_output2 == expected_output_no_dynamic_context
class TestExtendedPatchMoreLines: class TestExtendedPatchMoreLines:
def setUp(self):
get_settings().config.allow_dynamic_context = False
class File: class File:
def __init__(self, base_file, patch, filename): def __init__(self, base_file, patch, filename):
self.base_file = base_file self.base_file = base_file