spec/watirspec/elements/theads_spec.rb in watir-7.2.0 vs spec/watirspec/elements/theads_spec.rb in watir-7.2.1

@@ -1,70 +1,72 @@
 # frozen_string_literal: true
 
 require 'watirspec_helper'
 
-describe 'TableHeaders' do
-  before :each do
-    browser.goto(WatirSpec.url_for('tables.html'))
-  end
-
-  describe 'with selectors' do
-    it 'returns the matching elements' do
-      expect(browser.theads(id: 'tax_headers').to_a).to eq [browser.thead(id: 'tax_headers')]
+module Watir
+  describe 'TableHeaders' do
+    before do
+      browser.goto(WatirSpec.url_for('tables.html'))
     end
-  end
 
-  describe '#length' do
-    it 'returns the correct number of table theads (page context)' do
-      expect(browser.theads.length).to eq 1
+    describe 'with selectors' do
+      it 'returns the matching elements' do
+        expect(browser.theads(id: 'tax_headers').to_a).to eq [browser.thead(id: 'tax_headers')]
+      end
     end
 
-    it 'returns the correct number of table theads (table context)' do
-      expect(browser.table(index: 0).theads.length).to eq 1
-    end
-  end
+    describe '#length' do
+      it 'returns the correct number of table theads (page context)' do
+        expect(browser.theads.length).to eq 1
+      end
 
-  describe '#[]' do
-    it 'returns the row at the given index (page context)' do
-      expect(browser.theads[0].id).to eq 'tax_headers'
+      it 'returns the correct number of table theads (table context)' do
+        expect(browser.table(index: 0).theads.length).to eq 1
+      end
     end
 
-    it 'returns the row at the given index (table context)' do
-      expect(browser.table(index: 0).theads[0].id).to eq 'tax_headers'
-    end
-  end
+    describe '#[]' do
+      it 'returns the row at the given index (page context)' do
+        expect(browser.theads[0].id).to eq 'tax_headers'
+      end
 
-  describe '#each' do
-    it 'iterates through table theads correctly (page context)' do
-      browser.theads.each_with_index do |thead, index|
-        expect(thead.id).to eq browser.thead(index: index).id
+      it 'returns the row at the given index (table context)' do
+        expect(browser.table(index: 0).theads[0].id).to eq 'tax_headers'
       end
     end
 
     describe '#each' do
       it 'iterates through table theads correctly (page context)' do
-        count = 0
-
         browser.theads.each_with_index do |thead, index|
           expect(thead.id).to eq browser.thead(index: index).id
-
-          count += 1
         end
-
-        expect(count).to be > 0
       end
 
-      it 'iterates through table theads correctly (table context)' do
-        table = browser.table(index: 0)
-        count = 0
+      describe '#each' do
+        it 'iterates through table theads correctly (page context)' do
+          count = 0
 
-        table.theads.each_with_index do |thead, index|
-          expect(thead.id).to eq table.thead(index: index).id
+          browser.theads.each_with_index do |thead, index|
+            expect(thead.id).to eq browser.thead(index: index).id
 
-          count += 1
+            count += 1
+          end
+
+          expect(count).to be > 0
         end
 
-        expect(count).to be > 0
+        it 'iterates through table theads correctly (table context)' do
+          table = browser.table(index: 0)
+          count = 0
+
+          table.theads.each_with_index do |thead, index|
+            expect(thead.id).to eq table.thead(index: index).id
+
+            count += 1
+          end
+
+          expect(count).to be > 0
+        end
       end
     end
   end
 end