Skip to content

Commit 658002d

Browse files
committed
Add integration tests for complete cueman workflows
1 parent 8da624c commit 658002d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

cueman/tests/test_integration_workflows.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@ def test_batch_operation_workflow(self, mock_find, mock_stub):
1414
mock_stub.return_value = mock.Mock()
1515
mock_job1 = mock.Mock()
1616
mock_job2 = mock.Mock()
17+
# Mock getLayers to avoid protobuf errors
18+
mock_job1.getLayers.return_value = []
19+
mock_job2.getLayers.return_value = []
1720
mock_find.return_value = [mock_job1, mock_job2]
1821
sys.argv = ['cueman', '-kill', 'job1', 'job2']
1922
main.main(sys.argv)
@@ -26,6 +29,9 @@ def test_frame_management_workflow(self, mock_find, mock_stub):
2629
mock_stub.return_value = mock.Mock()
2730
mock_frame1 = mock.Mock()
2831
mock_frame2 = mock.Mock()
32+
# Mock getLayers for frames if needed
33+
mock_frame1.getLayers = mock.Mock(return_value=[])
34+
mock_frame2.getLayers = mock.Mock(return_value=[])
2935
mock_find.return_value = [mock_frame1, mock_frame2]
3036
sys.argv = ['cueman', '-ll', 'job1', '-state', 'DEAD']
3137
main.main(sys.argv)

0 commit comments

Comments
 (0)