[C] Tests

Insynia · Sep 5, 2016 · d0d8fa7 · d0d8fa7
1 parent b84b753
commit d0d8fa7
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/lib/timeline_rails/view_helpers.rb b/lib/timeline_rails/view_helpers.rb
@@ -6,7 +6,7 @@ module ViewHelpers
 
     def timeline_wrapper(&block)
       output = '<div class="timeline">'
-      output << capture(&block)
+      output << capture(&block) if block_given?
       output << '</div>'
       raw(output)
     end

diff --git a/spec/internal/db/combustion_test.sqlite b/spec/internal/db/combustion_test.sqlite
diff --git a/spec/timeline_rails/helpers/view_helpers_spec.rb b/spec/timeline_rails/helpers/view_helpers_spec.rb
@@ -5,6 +5,7 @@ def count_substrings(str, sub)
 end
 
 describe TimelineRails::ViewHelpers, type: :helper do
+  include ActionView::Helpers
   include TimelineRails::ViewHelpers
 
   describe '#timeline_block' do
@@ -28,4 +29,16 @@ def count_substrings(str, sub)
       end
     end
   end
+
+  describe '#timeline_wrapper' do
+    context 'when no block provided' do
+      subject { timeline_wrapper }
+
+      it 'has one timeline tag' do
+        expect(count_substrings(subject, '<div class="timeline">')).to eq 1
+      end
+
+      #TODO: Add check of yield
+    end
+  end
 end