nas/spec/models/home_feed_spec.rb
2025-05-28 12:34:37 +00:00

83 lines
2 KiB
Ruby

# frozen_string_literal: true
require 'rails_helper'
RSpec.describe HomeFeed do
subject { described_class.new(account) }
let(:account) { Fabricate(: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.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]
end
end
context 'when feed is being generated' do
before do
redis.set("account:#{account.id}:regeneration", true)
end
it 'returns nothing' do
results = subject.get(3)
expect(results.map(&:id)).to eq []
end
end
end
describe '#regenerating?' do
context 'when feed is being generated' do
before do
redis.set("account:#{account.id}:regeneration", true)
end
it 'returns `true`' do
expect(subject.regenerating?).to be true
end
end
context 'when feed is not being generated' do
it 'returns `false`' do
expect(subject.regenerating?).to be false
end
end
end
describe '#regeneration_in_progress!' do
it 'sets the corresponding key in redis' do
expect(redis.exists?("account:#{account.id}:regeneration")).to be false
subject.regeneration_in_progress!
expect(redis.exists?("account:#{account.id}:regeneration")).to be true
end
end
describe '#regeneration_finished!' do
it 'removes the corresponding key from redis' do
redis.set("account:#{account.id}:regeneration", true)
subject.regeneration_finished!
expect(redis.exists?("account:#{account.id}:regeneration")).to be false
end
end
end