55import tempfile
66import unittest
77from argparse import Namespace
8- from unittest .mock import MagicMock , patch
8+ from unittest .mock import patch
99
1010
1111def _make_args (** kwargs ):
@@ -97,9 +97,11 @@ class TestWorkflowList(unittest.TestCase):
9797 def test_list_workflows_text (self , _mock_key , mock_list ):
9898 from roboflow .cli .handlers .workflow import _list_workflows
9999
100- mock_list .return_value = {"workflows" : [
101- {"name" : "My Workflow" , "url" : "my-workflow" , "status" : "active" },
102- ]}
100+ mock_list .return_value = {
101+ "workflows" : [
102+ {"name" : "My Workflow" , "url" : "my-workflow" , "status" : "active" },
103+ ]
104+ }
103105 args = _make_args ()
104106 with patch ("builtins.print" ) as mock_print :
105107 _list_workflows (args )
@@ -112,9 +114,11 @@ def test_list_workflows_text(self, _mock_key, mock_list):
112114 def test_list_workflows_json (self , _mock_key , mock_list ):
113115 from roboflow .cli .handlers .workflow import _list_workflows
114116
115- mock_list .return_value = {"workflows" : [
116- {"name" : "WF1" , "url" : "wf-1" , "status" : "active" },
117- ]}
117+ mock_list .return_value = {
118+ "workflows" : [
119+ {"name" : "WF1" , "url" : "wf-1" , "status" : "active" },
120+ ]
121+ }
118122 args = _make_args (json = True )
119123 with patch ("builtins.print" ) as mock_print :
120124 _list_workflows (args )
@@ -238,9 +242,7 @@ class TestWorkflowUpdate(unittest.TestCase):
238242 def test_update_workflow (self , _mock_key , mock_get , mock_update ):
239243 from roboflow .cli .handlers .workflow import _update_workflow
240244
241- mock_get .return_value = {
242- "workflow" : {"id" : "wf-123" , "name" : "My WF" , "url" : "my-wf" , "config" : "{}" }
243- }
245+ mock_get .return_value = {"workflow" : {"id" : "wf-123" , "name" : "My WF" , "url" : "my-wf" , "config" : "{}" }}
244246 mock_update .return_value = {"url" : "my-wf" , "status" : "updated" }
245247 defn = {"blocks" : []}
246248 with tempfile .NamedTemporaryFile (mode = "w" , suffix = ".json" , delete = False ) as f :
@@ -304,10 +306,12 @@ class TestWorkflowVersionList(unittest.TestCase):
304306 def test_list_versions (self , _mock_key , mock_versions ):
305307 from roboflow .cli .handlers .workflow import _list_workflow_versions
306308
307- mock_versions .return_value = {"versions" : [
308- {"version" : "1" , "created" : "2026-01-01" },
309- {"version" : "2" , "created" : "2026-02-01" },
310- ]}
309+ mock_versions .return_value = {
310+ "versions" : [
311+ {"version" : "1" , "created" : "2026-01-01" },
312+ {"version" : "2" , "created" : "2026-02-01" },
313+ ]
314+ }
311315 args = _make_args (workflow_url = "my-wf" )
312316 with patch ("builtins.print" ) as mock_print :
313317 _list_workflow_versions (args )
@@ -340,9 +344,7 @@ def test_fork_workflow_same_workspace(self, _mock_key, mock_fork):
340344 args = _make_args (workflow_url = "my-wf" )
341345 with patch ("builtins.print" ) as mock_print :
342346 _fork_workflow (args )
343- mock_fork .assert_called_once_with (
344- "test-key" , "test-ws" , source_workspace = "test-ws" , source_workflow = "my-wf"
345- )
347+ mock_fork .assert_called_once_with ("test-key" , "test-ws" , source_workspace = "test-ws" , source_workflow = "my-wf" )
346348 printed = mock_print .call_args [0 ][0 ]
347349 self .assertIn ("Forked workflow" , printed )
348350
@@ -356,9 +358,7 @@ def test_fork_workflow_cross_workspace(self, _mock_key, mock_fork):
356358 args = _make_args (workflow_url = "other-ws/my-wf" )
357359 with patch ("builtins.print" ):
358360 _fork_workflow (args )
359- mock_fork .assert_called_once_with (
360- "test-key" , "test-ws" , source_workspace = "other-ws" , source_workflow = "my-wf"
361- )
361+ mock_fork .assert_called_once_with ("test-key" , "test-ws" , source_workspace = "other-ws" , source_workflow = "my-wf" )
362362
363363 @patch ("roboflow.adapters.rfapi.fork_workflow" )
364364 @patch ("roboflow.config.load_roboflow_api_key" , return_value = "test-key" )
0 commit comments