Making OVS Switch Prompts configurable.

[integration/test.git] / test / csit / libraries / SwitchClasses / BaseSwitch.py

diff --git a/test/csit/libraries/SwitchClasses/BaseSwitch.py b/test/csit/libraries/SwitchClasses/BaseSwitch.py
index b09c91d6ad22fcf1fd163185d2017d1e58fa56ca..82b741bf316c846e9d29f45e14dc2598b4424fdf 100644 (file)
--- a/test/csit/libraries/SwitchClasses/BaseSwitch.py
+++ b/test/csit/libraries/SwitchClasses/BaseSwitch.py
@@ -67,6 +67,9 @@ class BaseSwitch(object):
     def set_mgmt_user(self, user):
         self.mgmt_user = user
 
+    def set_mgmt_prompt(self, prompt):
+        self.mgmt_prompt = prompt
+
     def set_ssh_key(self, key):
         self.ssh_key = key