1- import pytest
21from unittest .mock import Mock , patch
32
3+ import pytest
4+
45
56@pytest .fixture
67def mock_get_openstack_conn ():
@@ -9,9 +10,8 @@ def mock_get_openstack_conn():
910
1011 with patch (
1112 "openstack_mcp_server.tools.compute_tools.get_openstack_conn" ,
12- return_value = mock_conn
13- ) as mock_func :
14-
13+ return_value = mock_conn ,
14+ ):
1515 yield mock_conn
1616
1717
@@ -22,9 +22,8 @@ def mock_get_openstack_conn_image():
2222
2323 with patch (
2424 "openstack_mcp_server.tools.image_tools.get_openstack_conn" ,
25- return_value = mock_conn
26- ) as mock_func :
27-
25+ return_value = mock_conn ,
26+ ):
2827 yield mock_conn
2928
3029
@@ -35,9 +34,8 @@ def mock_get_openstack_conn_identity():
3534
3635 with patch (
3736 "openstack_mcp_server.tools.identity_tools.get_openstack_conn" ,
38- return_value = mock_conn
39- ) as mock_func :
40-
37+ return_value = mock_conn ,
38+ ):
4139 yield mock_conn
4240
4341
@@ -54,12 +52,12 @@ def mock_openstack_base():
5452
5553
5654@pytest .fixture
57- def mock_openstack_connect_neutron ():
58- """Mock get_openstack_conn function for neutron_tools ."""
55+ def mock_openstack_connect_network ():
56+ """Mock get_openstack_conn function for network_tools ."""
5957 mock_conn = Mock ()
6058
6159 with patch (
62- "openstack_mcp_server.tools.neutron_tools .get_openstack_conn" ,
60+ "openstack_mcp_server.tools.network_tools .get_openstack_conn" ,
6361 return_value = mock_conn ,
6462 ):
6563 yield mock_conn
0 commit comments