diff options
Diffstat (limited to 'spec/lib/gitlab/database/health_status_spec.rb')
-rw-r--r-- | spec/lib/gitlab/database/health_status_spec.rb | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/spec/lib/gitlab/database/health_status_spec.rb b/spec/lib/gitlab/database/health_status_spec.rb index 4a2b9eee45a..95f74929b84 100644 --- a/spec/lib/gitlab/database/health_status_spec.rb +++ b/spec/lib/gitlab/database/health_status_spec.rb @@ -21,9 +21,11 @@ RSpec.describe Gitlab::Database::HealthStatus, feature_category: :database do let(:autovacuum_indicator_class) { health_status::Indicators::AutovacuumActiveOnTable } let(:wal_indicator_class) { health_status::Indicators::WriteAheadLog } let(:patroni_apdex_indicator_class) { health_status::Indicators::PatroniApdex } + let(:wal_rate_indicator_class) { health_status::Indicators::WalRate } let(:autovacuum_indicator) { instance_double(autovacuum_indicator_class) } let(:wal_indicator) { instance_double(wal_indicator_class) } let(:patroni_apdex_indicator) { instance_double(patroni_apdex_indicator_class) } + let(:wal_rate_indicator) { instance_double(wal_rate_indicator_class) } before do allow(autovacuum_indicator_class).to receive(:new).with(health_context).and_return(autovacuum_indicator) @@ -39,11 +41,17 @@ RSpec.describe Gitlab::Database::HealthStatus, feature_category: :database do expect(autovacuum_indicator).to receive(:evaluate).and_return(normal_signal) expect(wal_indicator_class).to receive(:new).with(health_context).and_return(wal_indicator) expect(wal_indicator).to receive(:evaluate).and_return(not_available_signal) - expect(patroni_apdex_indicator_class).to receive(:new).with(health_context) - .and_return(patroni_apdex_indicator) + expect(patroni_apdex_indicator_class).to receive(:new).with(health_context).and_return(patroni_apdex_indicator) expect(patroni_apdex_indicator).to receive(:evaluate).and_return(not_available_signal) - - expect(evaluate).to contain_exactly(normal_signal, not_available_signal, not_available_signal) + expect(wal_rate_indicator_class).to receive(:new).with(health_context).and_return(wal_rate_indicator) + expect(wal_rate_indicator).to receive(:evaluate).and_return(not_available_signal) + + expect(evaluate).to contain_exactly( + normal_signal, + not_available_signal, + not_available_signal, + not_available_signal + ) end end |