1 files changed, 26 insertions, 3 deletions

diff --git a/vid-automation/src/main/java/vid/automation/test/infra/SelectOption.java b/vid-automation/src/main/java/vid/automation/test/infra/SelectOption.java
index 048174e99..dc792cfc3 100644
--- a/vid-automation/src/main/java/vid/automation/test/infra/SelectOption.java
+++ b/vid-automation/src/main/java/vid/automation/test/infra/SelectOption.java
@@ -20,9 +20,13 @@ public class SelectOption {
         return select;
     }
 
-    public static void byIdAndVisibleText(String id, String text) {
+    public static Select byIdAndVisibleText(String id, String text) {
         Select selectlist = new Select(Get.byId(id));
-        selectlist.selectByVisibleText(text);
+        if(text != null) {
+            selectlist.selectByVisibleText(text);
+        }
+
+        return selectlist;
     }
 
     public static void byClassAndVisibleText(String className, String text) {
@@ -32,7 +36,14 @@ public class SelectOption {
             byIdAndVisibleText(id, text);
         });
     }
-
+    public static List<WebElement> getList(String selectDataTestId) {
+        Select selectList = GeneralUIUtils.getSelectList(null, selectDataTestId);
+        return selectList.getOptions();
+    }
+    public static String getSelectedOption(String selectDataTestId) {
+        Select selectList = GeneralUIUtils.getSelectList(null, selectDataTestId);
+        return selectList.getFirstSelectedOption().getText();
+    }
     public static void byTestIdAndVisibleText(String displayName, String selectDataTestId) {
         GeneralUIUtils.getSelectList(displayName, selectDataTestId);
     }
@@ -44,4 +55,16 @@ public class SelectOption {
         Click.byClass(Constants.MULTI_SELECT_UNSELECTED_CLASS);
 
     }
+
+    public static void selectOptionsFromMultiselectById(String multiSelectId, List<String> options) {
+        Click.byId(multiSelectId);
+        try {
+            Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+        for(String option:options) {
+            Click.byClassAndVisibleText(Constants.MULTI_SELECT_UNSELECTED_CLASS, option);
+        }
+    }
 }