github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-09 21:27:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

Simplify project load

Browse Source
This commit is contained in:
mijauexe
2025-03-27 18:08:38 +01:00
parent c399210f13
commit 98dc02f690
4 changed files with 31 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/cli/test_cli.py
View File

@@ -167,22 +167,15 @@ def test_show_default_config(capsys):
async def test_list_projects_json(mock_StateManager, capsys):
sm = mock_StateManager.return_value
# Mock the return value to match the expected format
# Format: project_id, project_name, branch_id, branch_name, state_id, step_index, action, created_at
project_id = MagicMock(hex="abcd")
branch_id = MagicMock(hex="1234")
state_id1 = MagicMock(hex="state1")
state_id2 = MagicMock(hex="state2")
state_id3 = MagicMock(hex="state3")
project_id1 = MagicMock(hex="abcd")
project_id2 = MagicMock(hex="efgh")
created_at1 = datetime(2021, 1, 1)
created_at2 = datetime(2021, 1, 2)
created_at3 = datetime(2021, 1, 3)
projects_data = [
(project_id, "project1", branch_id, "branch1", state_id1, 1, "foo", created_at1),
(project_id, "project1", branch_id, "branch1", state_id2, 2, None, created_at2),
(project_id, "project1", branch_id, "branch1", state_id3, 3, "baz", created_at3),
(project_id1, "project1", created_at1, "folder1"),
(project_id2, "project2", created_at2, "folder2"),
]
sm.list_projects = AsyncMock(return_value=projects_data)
@@ -191,25 +184,12 @@ async def test_list_projects_json(mock_StateManager, capsys):
mock_StateManager.assert_called_once_with(None)
sm.list_projects.assert_awaited_once_with()
data = json.loads(capsys.readouterr().out)
captured = capsys.readouterr().out
data = json.loads(captured)
assert data == [
{
"name": "project1",
"id": "abcd",
"updated_at": "2021-01-03T00:00:00",
"branches": [
{
"name": "branch1",
"id": "1234",
"steps": [
{"step": 1, "name": "foo"},
{"step": 2, "name": None},
{"step": 3, "name": "Latest step"},
],
},
],
},
{"id": "abcd", "name": "project1", "folder_name": "folder1", "updated_at": "2021-01-01T00:00:00"},
{"id": "efgh", "name": "project2", "folder_name": "folder2", "updated_at": "2021-01-02T00:00:00"},
]