1- from unittest .mock import patch , mock_open
1+ from unittest .mock import mock_open , patch
2+
23import pytest
34
45from ai_feedback .__main__ import _load_content_with_fallback
@@ -11,66 +12,46 @@ def test_load_content_predefined_name_success(self):
1112 """Test loading content using a predefined name."""
1213 content = "Test prompt content"
1314 predefined_values = ["test_prompt" , "another_prompt" ]
14-
15+
1516 with patch ("builtins.open" , mock_open (read_data = content )):
1617 with patch ("os.path.join" ) as mock_join :
1718 mock_join .return_value = "/fake/path/user/test_prompt.md"
18-
19- result = _load_content_with_fallback (
20- "test_prompt" ,
21- predefined_values ,
22- "user" ,
23- "prompt"
24- )
25-
19+
20+ result = _load_content_with_fallback ("test_prompt" , predefined_values , "user" , "prompt" )
21+
2622 assert result == content
2723 mock_join .assert_called_once ()
2824
2925 def test_load_content_custom_file_path_success (self ):
3026 """Test loading content using a custom file path."""
3127 content = "Custom file content"
3228 predefined_values = ["predefined_prompt" ]
33-
29+
3430 with patch ("builtins.open" , mock_open (read_data = content )):
35- result = _load_content_with_fallback (
36- "/custom/path/file.md" ,
37- predefined_values ,
38- "user" ,
39- "prompt"
40- )
41-
31+ result = _load_content_with_fallback ("/custom/path/file.md" , predefined_values , "user" , "prompt" )
32+
4233 assert result == content
4334
4435 def test_load_content_predefined_name_not_found (self , capsys ):
4536 """Test error when predefined file is not found."""
4637 predefined_values = ["test_prompt" ]
47-
38+
4839 with patch ("builtins.open" , side_effect = FileNotFoundError ):
4940 with pytest .raises (SystemExit ) as exc_info :
50- _load_content_with_fallback (
51- "test_prompt" ,
52- predefined_values ,
53- "user" ,
54- "prompt"
55- )
56-
41+ _load_content_with_fallback ("test_prompt" , predefined_values , "user" , "prompt" )
42+
5743 assert exc_info .value .code == 1
5844 captured = capsys .readouterr ()
5945 assert "Pre-defined prompt file 'test_prompt.md' not found in user subfolder." in captured .out
6046
6147 def test_load_content_custom_file_path_not_found (self , capsys ):
6248 """Test error when custom file is not found."""
6349 predefined_values = ["predefined_prompt" ]
64-
50+
6551 with patch ("builtins.open" , side_effect = FileNotFoundError ):
6652 with pytest .raises (SystemExit ) as exc_info :
67- _load_content_with_fallback (
68- "/nonexistent/file.md" ,
69- predefined_values ,
70- "user" ,
71- "prompt"
72- )
73-
53+ _load_content_with_fallback ("/nonexistent/file.md" , predefined_values , "user" , "prompt" )
54+
7455 assert exc_info .value .code == 1
7556 captured = capsys .readouterr ()
7657 assert "Prompt file '/nonexistent/file.md' not found." in captured .out
0 commit comments