@@ -79,21 +79,22 @@ def test_when_successful(self, mocker, port_id, trunk_id):
79
79
80
80
81
81
class TestHandleRouterInterfaceRemoval :
82
- def test_case_for_non_router (self , mocker , port_payload ):
82
+ def test_case_for_non_router (self , mocker , port_db_payload ):
83
83
mock_sb_removal = mocker .patch (
84
84
"neutron_understack.routers.handle_subport_removal"
85
85
)
86
86
mock_localnet_removal = mocker .patch (
87
87
"neutron_understack.routers.delete_uplink_port"
88
88
)
89
89
90
- port_payload .metadata ["port" ]["device_owner" ] = "not_a_router"
91
- handle_router_interface_removal (None , None , None , port_payload )
90
+ port_db_payload .metadata ["port_db" ].device_owner = "not_a_router"
91
+
92
+ handle_router_interface_removal (None , None , None , port_db_payload )
92
93
93
94
mock_sb_removal .assert_not_called ()
94
95
mock_localnet_removal .assert_not_called ()
95
96
96
- def test_when_network_in_use (self , mocker , port_payload ):
97
+ def test_when_network_in_use (self , mocker , port_db_payload ):
97
98
mock_sb_removal = mocker .patch (
98
99
"neutron_understack.routers.handle_subport_removal"
99
100
)
@@ -105,12 +106,14 @@ def test_when_network_in_use(self, mocker, port_payload):
105
106
return_value = False ,
106
107
)
107
108
108
- handle_router_interface_removal (None , None , None , port_payload )
109
+ handle_router_interface_removal (None , None , None , port_db_payload )
109
110
110
111
mock_sb_removal .assert_not_called ()
111
112
mock_localnet_removal .assert_not_called ()
112
113
113
- def test_last_port_on_network (self , mocker , port_object , port_payload , network_id ):
114
+ def test_last_port_on_network (
115
+ self , mocker , port_object , port_db_payload , network_id
116
+ ):
114
117
mock_sb_removal = mocker .patch (
115
118
"neutron_understack.routers.handle_subport_removal"
116
119
)
@@ -133,7 +136,7 @@ def test_last_port_on_network(self, mocker, port_object, port_payload, network_i
133
136
)
134
137
delete_shared_port = mocker .patch .object (port_object , "delete" )
135
138
136
- handle_router_interface_removal (None , None , None , port_payload )
139
+ handle_router_interface_removal (None , None , None , port_db_payload )
137
140
138
141
mock_sb_removal .assert_called_once_with (port_object )
139
142
mock_localnet_removal .assert_called_once_with (fake_segment , str (network_id ))
0 commit comments