From 3f5b994ff0674fa8e0e5676ca22f0c347f9d7712 Mon Sep 17 00:00:00 2001
From: Matt Jankowski <mjankowski@thoughtbot.com>
Date: Fri, 5 May 2017 22:00:21 -0400
Subject: Stream entry specs and refactor to use delegate (#2827)

* Add coverage for stream entry delegated methods

* Use delegate with allow_nil to clean up stream entry
---
 spec/models/stream_entry_spec.rb | 48 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

(limited to 'spec/models')

diff --git a/spec/models/stream_entry_spec.rb b/spec/models/stream_entry_spec.rb
index 45bf26899..3b7ff5143 100644
--- a/spec/models/stream_entry_spec.rb
+++ b/spec/models/stream_entry_spec.rb
@@ -26,4 +26,52 @@ RSpec.describe StreamEntry, type: :model do
       expect(status.stream_entry.threaded?).to be false
     end
   end
+
+  describe 'delegated methods' do
+    context 'with a nil status' do
+      subject { described_class.new(status: nil) }
+
+      it 'returns nil for target' do
+        expect(subject.target).to be_nil
+      end
+
+      it 'returns nil for title' do
+        expect(subject.title).to be_nil
+      end
+
+      it 'returns nil for content' do
+        expect(subject.content).to be_nil
+      end
+
+      it 'returns nil for thread' do
+        expect(subject.thread).to be_nil
+      end
+    end
+
+    context 'with a real status' do
+      let(:original) { Fabricate(:status, text: 'Test status') }
+      let(:status) { Fabricate(:status, reblog: original, thread: original) }
+      subject { described_class.new(status: status) }
+
+      it 'delegates target' do
+        expect(status.target).not_to be_nil
+        expect(subject.target).to eq(status.target)
+      end
+
+      it 'delegates title' do
+        expect(status.title).not_to be_nil
+        expect(subject.title).to eq(status.title)
+      end
+
+      it 'delegates content' do
+        expect(status.content).not_to be_nil
+        expect(subject.content).to eq(status.content)
+      end
+
+      it 'delegates thread' do
+        expect(status.thread).not_to be_nil
+        expect(subject.thread).to eq(status.thread)
+      end
+    end
+  end
 end
-- 
cgit