mirror of
https://github.com/lunaisnotaboy/mastodon.git
synced 2024-12-25 14:46:46 +00:00
5db3a14388
* Rework tests * Add tests * Fix HomeFeed#get with min_id fetching from database * Minor code cleanup and optimizations * Add tests * Take DMs into account when fetching home TL from database * Fix not listing own DMs in Home timeline * Add tests * Please CodeClimate
110 lines
3.3 KiB
Ruby
110 lines
3.3 KiB
Ruby
require 'rails_helper'
|
|
|
|
RSpec.describe HomeFeed, type: :model do
|
|
let(:account) { Fabricate(:account) }
|
|
let(:followed) { Fabricate(:account) }
|
|
let(:other) { Fabricate(:account) }
|
|
|
|
subject { described_class.new(account) }
|
|
|
|
describe '#get' do
|
|
before do
|
|
account.follow!(followed)
|
|
|
|
Fabricate(:status, account: account, id: 1)
|
|
Fabricate(:status, account: account, id: 2)
|
|
status = Fabricate(:status, account: followed, id: 3)
|
|
Fabricate(:mention, account: account, status: status)
|
|
Fabricate(:status, account: account, id: 10)
|
|
Fabricate(:status, account: other, id: 11)
|
|
Fabricate(:status, account: followed, id: 12, visibility: :private)
|
|
Fabricate(:status, account: followed, id: 13, visibility: :direct)
|
|
Fabricate(:status, account: account, id: 14, visibility: :direct)
|
|
dm = Fabricate(:status, account: followed, id: 15, visibility: :direct)
|
|
Fabricate(:mention, account: account, status: dm)
|
|
end
|
|
|
|
context 'when feed is generated' do
|
|
before do
|
|
FeedManager.instance.populate_home(account)
|
|
|
|
# Add direct messages because populate_home does not do that
|
|
Redis.current.zadd(
|
|
FeedManager.instance.key(:home, account.id),
|
|
[[14, 14], [15, 15]]
|
|
)
|
|
end
|
|
|
|
it 'gets statuses with ids in the range from redis with database' do
|
|
results = subject.get(5)
|
|
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10, 3]
|
|
expect(results.first.attributes.keys).to eq %w(id updated_at)
|
|
end
|
|
|
|
it 'with since_id present' do
|
|
results = subject.get(5, nil, 3, nil)
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10]
|
|
end
|
|
|
|
it 'with min_id present' do
|
|
results = subject.get(3, nil, nil, 0)
|
|
expect(results.map(&:id)).to eq [3, 2, 1]
|
|
end
|
|
end
|
|
|
|
context 'when feed is only partial' do
|
|
before do
|
|
FeedManager.instance.populate_home(account)
|
|
|
|
# Add direct messages because populate_home does not do that
|
|
Redis.current.zadd(
|
|
FeedManager.instance.key(:home, account.id),
|
|
[[14, 14], [15, 15]]
|
|
)
|
|
|
|
Redis.current.zremrangebyrank(FeedManager.instance.key(:home, account.id), 0, -2)
|
|
end
|
|
|
|
it 'gets statuses with ids in the range from redis with database' do
|
|
results = subject.get(5)
|
|
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10, 3]
|
|
expect(results.first.attributes.keys).to eq %w(id updated_at)
|
|
end
|
|
|
|
it 'with since_id present' do
|
|
results = subject.get(5, nil, 3, nil)
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10]
|
|
end
|
|
|
|
it 'with min_id present' do
|
|
results = subject.get(3, nil, nil, 0)
|
|
expect(results.map(&:id)).to eq [3, 2, 1]
|
|
end
|
|
end
|
|
|
|
context 'when feed is being generated' do
|
|
before do
|
|
Redis.current.set("account:#{account.id}:regeneration", true)
|
|
end
|
|
|
|
it 'returns from database' do
|
|
results = subject.get(5)
|
|
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10, 3]
|
|
end
|
|
|
|
it 'with since_id present' do
|
|
results = subject.get(5, nil, 3, nil)
|
|
expect(results.map(&:id)).to eq [15, 14, 12, 10]
|
|
end
|
|
|
|
it 'with min_id present' do
|
|
results = subject.get(3, nil, nil, 0)
|
|
expect(results.map(&:id)).to eq [3, 2, 1]
|
|
end
|
|
end
|
|
end
|
|
end
|