Added support for Rspec and Cucumber

lib/page_record/class_actions.rb

@@ -5,7 +5,7 @@ def self.method_missing(action)
 			raw_action = /(.*)\?/.match(action)
 			begin
 				if raw_action
-					raw_action_for(raw_action[0])
+					action_for?(raw_action[1])
 				else
 					action_for(action)
 				end
@@ -21,12 +21,14 @@ def self.method_missing(action)
 
 
 		def self.action_for(action)
-			element = raw_action_for(action)
-			element.native.click
+			element = action_for?(action)
+			element.click
+			element
 		end
 
-		def self.raw_action_for(action)
-			@page.find("[data-action-for='#{action}']")
+
+		def self.action_for?(action)
+			page.find("[data-action-for='#{action}']")
 		end
 	end
 end

lib/page_record/cucumber.rb

@@ -0,0 +1,5 @@
+require 'page_record'
+
+Before do
+  PageRecord::PageRecord.page = page
+end

lib/page_record/instance_actions.rb

@@ -4,7 +4,7 @@ def method_missing(action)
 			raw_action = /(.*)\?/.match(action)
 			begin
 				if raw_action
-					raw_action_for(raw_action[0])
+					action_for?(raw_action[1])
 				else
 					action_for(action)
 				end
@@ -16,13 +16,14 @@ def method_missing(action)
 
 	private
 
+
 		def action_for(action)
-			element = raw_action_for(action)
-			element.native.click
+			element = action_for?(action)
+			element.click
 			element
 		end
 
-		def raw_action_for(action)
+		def action_for?(action)
 			@record.find("[data-action-for='#{action}']")
 		end
 	end

lib/page_record/rspec.rb

@@ -0,0 +1,7 @@
+require 'page_record'
+
+RSpec.configure do |config|
+	config.before(:each) do
+    PageRecord::PageRecord.page = session
+  end
+end