spec/unit/lib/web/section_spec.rb in howitzer-2.0.0 vs spec/unit/lib/web/section_spec.rb in howitzer-2.0.1

@@ -1,16 +1,18 @@
 require 'spec_helper'
 require 'howitzer/web/section_dsl'
 require 'howitzer/web/section'
-require 'howitzer/web/element_dsl'
 RSpec.describe Howitzer::Web::Section do
   describe 'element dsl methods' do
     let(:parent) { double }
     let(:capybara_context) { double }
 
     let(:klass) { Class.new(described_class) }
     let(:klass_object) { klass.new(parent, capybara_context) }
-    before { allow(klass_object).to receive(:capybara_context) { kontext } }
+
+    it 'returns correct capybara context' do
+      expect(klass_object.capybara_context).to eq(capybara_context)
+    end
 
     include_examples :element_dsl
   end
 
   describe 'DSL' do