spec/helpers/table_screen_searchable.rb in ProMotion-1.0.4 vs spec/helpers/table_screen_searchable.rb in ProMotion-1.1.0.rc1

@@ -1,9 +1,11 @@
 class TableScreenSearchable < TestTableScreen
 
   searchable
 
+  attr_accessor :will_end_search_called, :will_begin_search_called
+
   def table_data
     @search_table_data ||= [{
       cells: [
         { title: "Alabama" },
         { title: "Alaska" },
@@ -55,8 +57,16 @@
         { title: "West Virginia" },
         { title: "Wisconsin" },
         { title: "Wyoming" }
       ]
     }]
+  end
+
+  def will_begin_search
+    self.will_begin_search_called = true
+  end
+
+  def will_end_search
+    self.will_end_search_called = true
   end
 
 end