|
44 | 44 | "5": "scheduler.0",
|
45 | 45 | "6": "scheduler.0",
|
46 | 46 | "7": ""}
|
| 47 | +BLOCK_DATA_PLANE_SCHEDULER_NAME = 'scheduler.block_data_plane' |
47 | 48 |
|
48 | 49 |
|
49 | 50 | class ThriftInterface(BaseTest):
|
@@ -238,16 +239,15 @@ def disable_mellanox_egress_data_plane(self, ptf_port_list):
|
238 | 239 | dut_port = self.get_dut_port(ptf_port)
|
239 | 240 | dut_port_list.append(dut_port)
|
240 | 241 | self.original_dut_port_queue_scheduler_map = self.get_queue_scheduler_name(dut_port_list)
|
241 |
| - block_data_plane_scheduler_name = 'scheduler.block_data_plane' |
242 | 242 | cmd_set_block_data_plane_scheduler = \
|
243 |
| - f'sonic-db-cli CONFIG_DB hset "SCHEDULER|{block_data_plane_scheduler_name}" "type" DWRR "weight" 15 "pir" 1' |
| 243 | + f'sonic-db-cli CONFIG_DB hset "SCHEDULER|{BLOCK_DATA_PLANE_SCHEDULER_NAME}" "type" DWRR "weight" 15 "pir" 1' |
244 | 244 |
|
245 | 245 | self.exec_cmd_on_dut(self.server, self.test_params['dut_username'], self.test_params['dut_password'],
|
246 | 246 | cmd_set_block_data_plane_scheduler)
|
247 | 247 | for dut_port in dut_port_list:
|
248 | 248 | for q in DATA_PLANE_QUEUE_LIST:
|
249 | 249 | cmd_block_q = \
|
250 |
| - f" sonic-db-cli CONFIG_DB hset 'QUEUE|{dut_port}|{q}' scheduler {block_data_plane_scheduler_name}" |
| 250 | + f" sonic-db-cli CONFIG_DB hset 'QUEUE|{dut_port}|{q}' scheduler {BLOCK_DATA_PLANE_SCHEDULER_NAME}" |
251 | 251 | self.exec_cmd_on_dut(
|
252 | 252 | self.server, self.test_params['dut_username'], self.test_params['dut_password'], cmd_block_q)
|
253 | 253 |
|
@@ -276,6 +276,20 @@ def enable_mellanox_egress_data_plane(self, ptf_port_list):
|
276 | 276 | self.exec_cmd_on_dut(
|
277 | 277 | self.server, self.test_params['dut_username'],
|
278 | 278 | self.test_params['dut_password'], cmd_recover_q_scheduler_config)
|
| 279 | + self.remove_block_data_plan_scheduler() |
| 280 | + |
| 281 | + def remove_block_data_plan_scheduler(self): |
| 282 | + get_block_data_plane_scheduler_name = \ |
| 283 | + f"sonic-db-cli CONFIG_DB keys 'SCHEDULER|{BLOCK_DATA_PLANE_SCHEDULER_NAME}'" |
| 284 | + scheduler_name, _, _ = self.exec_cmd_on_dut(self.server, |
| 285 | + self.test_params['dut_username'], |
| 286 | + self.test_params['dut_password'], |
| 287 | + get_block_data_plane_scheduler_name) |
| 288 | + if isinstance(scheduler_name, list) and BLOCK_DATA_PLANE_SCHEDULER_NAME in scheduler_name[0]: |
| 289 | + cmd_del_block_data_plane_scheduler = \ |
| 290 | + f'sonic-db-cli CONFIG_DB del "SCHEDULER|{BLOCK_DATA_PLANE_SCHEDULER_NAME}"' |
| 291 | + self.exec_cmd_on_dut(self.server, self.test_params['dut_username'], self.test_params['dut_password'], |
| 292 | + cmd_del_block_data_plane_scheduler) |
279 | 293 |
|
280 | 294 |
|
281 | 295 | class ThriftInterfaceDataPlane(ThriftInterface):
|
|
0 commit comments