Add support for selected multiple tests

Signed-off-by: Dante Su <[email protected]>
sonic-net · Jun 28, 2022 · 8e2f7a4 · 8e2f7a4
1 parent f67e776
commit 8e2f7a4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -405,7 +405,8 @@ def __init__(
 
     def create_servers(self):
         for i in range(NUM_PORTS):
-            server = VirtualServer(self.ctn_sw.name, self.ctn_sw_pid, i)
+            name = self.ctn_sw.name if hasattr(self, 'ctn_sw') else self.dvsname
+            server = VirtualServer(name, self.ctn_sw_pid, i)
             self.servers.append(server)
 
     def reset_dbs(self):