spec/element_spec.rb in page_magic-1.0.0.alpha20 vs spec/element_spec.rb in page_magic-1.0.0.alpha21

@@ -8,14 +8,14 @@
         include PageMagic
         url '/elements'
       end
     end
 
-    let(:page) do
-      page_class.new.tap(&:visit)
-    end
+    let(:session) { page_class.visit(application: rack_app) }
 
+    let(:page) { session.current_page }
+
     describe 'inheriting' do
       it 'lets you create custom elements' do
         custom_element = Class.new(described_class) do
           text_field :form_field, id: 'field_id'
 
@@ -60,17 +60,14 @@
     end
 
     describe '#browser_element' do
       let!(:browser) { double('browser') }
 
-      it 'calls the on_load hook' do
-      end
-
       context 'options supplied to selector' do
         it 'passes them on to the cappybara finder method' do
           options = { count: 1 }
           xpath_selector = '//div/input'
-          expect(Capybara.current_session).to receive(:find).with(:xpath, xpath_selector, options)
+          expect(page.session.raw_session).to receive(:find).with(:xpath, xpath_selector, options)
           described_class.new(:my_input,
                               page,
                               type: :text_field,
                               selector: { xpath: xpath_selector }.merge(options)).browser_element
         end