1 files changed, 45 insertions, 0 deletions
diff --git a/spec/models/home_feed_spec.rb b/spec/models/home_feed_spec.rb
new file mode 100644
index 000000000..3acb997f1
--- /dev/null
+++ b/spec/models/home_feed_spec.rb
@@ -0,0 +1,45 @@
+require 'rails_helper'
+
+RSpec.describe HomeFeed, type: :model do
+  let(:account) { Fabricate(:account) }
+
+  subject { described_class.new(account) }
+
+  describe '#get' do
+    before do
+      Fabricate(:status, account: account, id: 1)
+      Fabricate(:status, account: account, id: 2)
+      Fabricate(:status, account: account, id: 3)
+      Fabricate(:status, account: account, id: 10)
+    end
+
+    context 'when feed is generated' do
+      before do
+        Redis.current.zadd(
+          FeedManager.instance.key(:home, account.id),
+          [[4, 4], [3, 3], [2, 2], [1, 1]]
+        )
+      end
+
+      it 'gets statuses with ids in the range from redis' do
+        results = subject.get(3)
+
+        expect(results.map(&:id)).to eq [3, 2]
+        expect(results.first.attributes.keys).to eq %w(id updated_at)
+      end
+    end
+
+    context 'when feed is being generated' do
+      before do
+        Redis.current.set("account:#{account.id}:regeneration", true)
+      end
+
+      it 'gets statuses with ids in the range from database' do
+        results = subject.get(3)
+
+        expect(results.map(&:id)).to eq [10, 3, 2]
+        expect(results.first.attributes.keys).to include('id', 'updated_at')
+      end
+    end
+  end
+end