Multiselect from dropdown (#106)

spec/lucky_flow_spec.cr

@@ -211,6 +211,38 @@ describe LuckyFlow do
     flow.select("cars", value: "ford")
     flow.el("#cars").value.should eq "ford"
   end
+
+  it "can choose options in multi select input" do
+    flow = visit_page_with <<-HTML
+      <select name="cars" id="cars" multiple>
+        <option value="ford">Ford</option>
+        <option value="honda">Honda</option>
+        <option value="tesla">Tesla</option>
+        <option value="toyota">Toyota</option>
+      </select>
+    HTML
+
+    flow.select("cars", value: ["honda", "toyota"])
+    flow.el("option[value='ford']").selected?.should be_false
+    flow.el("option[value='honda']").selected?.should be_true
+    flow.el("option[value='tesla']").selected?.should be_false
+    flow.el("option[value='toyota']").selected?.should be_true
+  end
+
+  it "raises error if attempting to select multiple options when not multi select" do
+    flow = visit_page_with <<-HTML
+      <select name="cars" id="cars">
+        <option value="ford">Ford</option>
+        <option value="honda">Honda</option>
+        <option value="tesla">Tesla</option>
+        <option value="toyota">Toyota</option>
+      </select>
+    HTML
+
+    expect_raises(LuckyFlow::InvalidOperationError) do
+      flow.select("cars", value: ["honda", "toyota"])
+    end
+  end
 end
 
 private class FakeProcess

src/lucky_flow.cr

@@ -110,14 +110,26 @@ class LuckyFlow
   # ```crystal
   # select("post:category", value: "rant")
   # ```
-  def select(name_attr : String, value : String)
+  #
+  # If given an Array(String), the select is assumed to have the 'multiple' attribute
+  # and will raise a `LuckyFlow::InvalidMultiSelectError` if it doesn't.
+  #
+  # ```crystal
+  # select("post:tags", value: ["rant", "technology"])
+  # ```
+  #
+  def select(name_attr : String, value : Array(String) | String)
     self.select(field(name_attr), value: value)
   end
 
   def select(element : Element, value : String)
     element.select_option(value)
   end
 
+  def select(element : Element, value : Array(String))
+    element.select_options(value)
+  end
+
   # Fill a form created by Lucky that uses an Avram::SaveOperation
   #
   # Note that Lucky and Avram are required to use this method

src/lucky_flow/element.cr

@@ -1,7 +1,7 @@
 class LuckyFlow::Element
   private getter raw_selector
   getter inner_text
-  delegate text, click, send_keys, displayed?, attribute, property, tag_name, to: element
+  delegate text, click, send_keys, displayed?, selected?, attribute, property, tag_name, to: element
   delegate session, to: LuckyFlow
 
   def initialize(@raw_selector : String, text @inner_text : String? = nil)
@@ -54,4 +54,11 @@ class LuckyFlow::Element
     select_el = Selenium::Helpers::Select.from_element(element)
     select_el.select_by_value(value)
   end
+
+  def select_options(values : Array(String))
+    select_el = Selenium::Helpers::Select.from_element(element)
+    raise LuckyFlow::InvalidMultiSelectError.new unless select_el.multiple?
+
+    values.each { |value| select_el.select_by_value(value) }
+  end
 end

src/lucky_flow/errors.cr

@@ -33,4 +33,13 @@ class LuckyFlow
       super message
     end
   end
+
+  class InvalidOperationError < Error
+  end
+
+  class InvalidMultiSelectError < InvalidOperationError
+    def initialize
+      super "Unable to select multiple options when select element does not have 'multiple' attribute"
+    end
+  end
 end