spec/gnawrnip/step_screenshot_spec.rb in gnawrnip-0.0.4 vs spec/gnawrnip/step_screenshot_spec.rb in gnawrnip-0.1.0

@@ -6,14 +6,39 @@
     let :template do
       StepScreenshot
     end
 
     it 'exists failure step template' do
-      expect(TurnipFormatter::Step::Failure.templates).to have_key :screenshot
+      expect(TurnipFormatter::Step::Failure.templates).to have_key template
     end
 
     describe '.build' do
-      subject { template.build('aiueo') }
-      it { should match %r{<img src="data:image/png;base64,aiueo"} }
+      subject { template.build(data_list) }
+
+      context 'has multiple data' do
+        let(:data_list) { ['aiueo', '12345', 'abcde'] }
+        it {
+          should include '<div class="screenshot animation">'
+          should include '<img src="data:image/png;base64,aiueo"/>'
+          should include '<img src="data:image/png;base64,12345"/>'
+          should include '<img src="data:image/png;base64,abcde"/></div>'
+        }
+      end
+
+      context 'has single data' do
+        let(:data_list) { ['aiueo'] }
+        it {
+          should include '<div class="screenshot">'
+          should include '<img src="data:image/png;base64,aiueo"/></div>'
+        }
+      end
+
+      context 'has no data' do
+        let(:data_list) { [] }
+        it {
+          should include '<div class="screenshot">'
+          should include '<img src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAhQ'
+        }
+      end
     end
   end
 end