@@ -797,7 +797,8 @@ def connect_dp(self, dp_id=None):
797797 return connect_msgs
798798
799799 def disconnect_dp (self ):
800- self .valve .datapath_disconnect (time .time ())
800+ valve = self .valves_manager .valves [self .DP_ID ]
801+ valve .datapath_disconnect (time .time ())
801802
802803 def cold_start (self , dp_id = None ):
803804 """
@@ -1292,7 +1293,7 @@ def test_disconnect(self):
12921293 """Test disconnection of DP from controller."""
12931294 valve = self .valves_manager .valves [self .DP_ID ]
12941295 self .assertEqual (1 , int (self .get_prom ('dp_status' )))
1295- self .prom_inc (partial (self . valve .datapath_disconnect , time .time ()), 'of_dp_disconnections_total' )
1296+ self .prom_inc (partial (valve .datapath_disconnect , time .time ()), 'of_dp_disconnections_total' )
12961297 self .assertEqual (0 , int (self .get_prom ('dp_status' )))
12971298
12981299 def test_unexpected_port (self ):
@@ -2139,7 +2140,8 @@ def test_get_config_dict(self):
21392140
21402141 def test_dp_disconnect_cleanup (self ):
21412142 """Test port varz cleanup post dp disconnect"""
2142- port_num = list (self .valve .dp .ports .keys ())[0 ]
2143+ valve = self .valves_manager .valves [self .DP_ID ]
2144+ port_num = list (valve .dp .ports .keys ())[0 ]
21432145 self .port_expected_status (port_num , 1 )
21442146 self .disconnect_dp ()
21452147 self .port_expected_status (port_num , 0 )
0 commit comments