117 lines
2.8 KiB
Ruby
117 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rails_helper'
|
|
|
|
describe ApplicationRequest do
|
|
before do
|
|
ApplicationRequest.enable
|
|
ApplicationRequest.last_flush = Time.now.utc
|
|
end
|
|
|
|
after do
|
|
ApplicationRequest.disable
|
|
ApplicationRequest.clear_cache!
|
|
end
|
|
|
|
def inc(key, opts = nil)
|
|
ApplicationRequest.increment!(key, opts)
|
|
end
|
|
|
|
def disable_date_flush!
|
|
freeze_time(Time.now)
|
|
ApplicationRequest.last_flush = Time.now.utc
|
|
end
|
|
|
|
context "readonly test" do
|
|
it 'works even if redis is in readonly' do
|
|
disable_date_flush!
|
|
|
|
inc(:http_total)
|
|
inc(:http_total)
|
|
|
|
Discourse.redis.without_namespace.stubs(:incr).raises(Redis::CommandError.new("READONLY"))
|
|
Discourse.redis.without_namespace.stubs(:eval).raises(Redis::CommandError.new("READONLY"))
|
|
|
|
# flush will be deferred no error raised
|
|
inc(:http_total, autoflush: 3)
|
|
ApplicationRequest.write_cache!
|
|
|
|
Discourse.redis.without_namespace.unstub(:incr)
|
|
Discourse.redis.without_namespace.unstub(:eval)
|
|
|
|
inc(:http_total, autoflush: 3)
|
|
expect(ApplicationRequest.http_total.first.count).to eq(3)
|
|
end
|
|
end
|
|
|
|
it 'logs nothing for an unflushed increment' do
|
|
ApplicationRequest.increment!(:page_view_anon)
|
|
expect(ApplicationRequest.count).to eq(0)
|
|
end
|
|
|
|
it 'can automatically flush' do
|
|
disable_date_flush!
|
|
|
|
inc(:http_total)
|
|
inc(:http_total)
|
|
inc(:http_total, autoflush: 3)
|
|
|
|
expect(ApplicationRequest.http_total.first.count).to eq(3)
|
|
|
|
inc(:http_total)
|
|
inc(:http_total)
|
|
inc(:http_total, autoflush: 3)
|
|
|
|
expect(ApplicationRequest.http_total.first.count).to eq(6)
|
|
end
|
|
|
|
it 'can flush based on time' do
|
|
t1 = Time.now.utc.at_midnight
|
|
freeze_time(t1)
|
|
ApplicationRequest.write_cache!
|
|
inc(:http_total)
|
|
expect(ApplicationRequest.count).to eq(0)
|
|
|
|
freeze_time(t1 + ApplicationRequest.autoflush_seconds + 1)
|
|
inc(:http_total)
|
|
|
|
expect(ApplicationRequest.count).to eq(1)
|
|
end
|
|
|
|
it 'flushes yesterdays results' do
|
|
t1 = Time.now.utc.at_midnight
|
|
freeze_time(t1)
|
|
inc(:http_total)
|
|
freeze_time(t1.tomorrow)
|
|
inc(:http_total)
|
|
|
|
ApplicationRequest.write_cache!
|
|
expect(ApplicationRequest.count).to eq(2)
|
|
end
|
|
|
|
it 'clears cache correctly' do
|
|
# otherwise we have test pollution
|
|
inc(:page_view_anon)
|
|
ApplicationRequest.clear_cache!
|
|
ApplicationRequest.write_cache!
|
|
|
|
expect(ApplicationRequest.count).to eq(0)
|
|
end
|
|
|
|
it 'logs a few counts once flushed' do
|
|
time = Time.now.at_midnight
|
|
freeze_time(time)
|
|
|
|
3.times { inc(:http_total) }
|
|
2.times { inc(:http_2xx) }
|
|
4.times { inc(:http_3xx) }
|
|
|
|
ApplicationRequest.write_cache!
|
|
|
|
expect(ApplicationRequest.http_total.first.count).to eq(3)
|
|
expect(ApplicationRequest.http_2xx.first.count).to eq(2)
|
|
expect(ApplicationRequest.http_3xx.first.count).to eq(4)
|
|
|
|
end
|
|
end
|