mirror of https://github.com/glitch-soc/mastodon
135 lines
4.6 KiB
Ruby
135 lines
4.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rails_helper'
|
|
|
|
RSpec.describe Admin::SystemCheck::ElasticsearchCheck do
|
|
subject(:check) { described_class.new(user) }
|
|
|
|
let(:user) { Fabricate(:user) }
|
|
|
|
it_behaves_like 'a check available to devops users'
|
|
|
|
describe 'pass?' do
|
|
context 'when chewy is enabled' do
|
|
before do
|
|
allow(Chewy).to receive(:enabled?).and_return(true)
|
|
allow(Chewy.client.cluster).to receive(:health).and_return({ 'status' => 'green', 'number_of_nodes' => 1 })
|
|
allow(Chewy.client.indices).to receive_messages(get_mapping: {
|
|
AccountsIndex.index_name => AccountsIndex.mappings_hash.deep_stringify_keys,
|
|
StatusesIndex.index_name => StatusesIndex.mappings_hash.deep_stringify_keys,
|
|
PublicStatusesIndex.index_name => PublicStatusesIndex.mappings_hash.deep_stringify_keys,
|
|
InstancesIndex.index_name => InstancesIndex.mappings_hash.deep_stringify_keys,
|
|
TagsIndex.index_name => TagsIndex.mappings_hash.deep_stringify_keys,
|
|
}, get_settings: {
|
|
'chewy_specifications' => {
|
|
'settings' => {
|
|
'index' => {
|
|
'number_of_replicas' => 0,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
end
|
|
|
|
context 'when running version is present and high enough' do
|
|
before do
|
|
allow(Chewy.client).to receive(:info)
|
|
.and_return({ 'version' => { 'number' => '999.99.9' } })
|
|
end
|
|
|
|
it 'returns true' do
|
|
expect(check.pass?).to be true
|
|
end
|
|
end
|
|
|
|
context 'when running version is present and too low' do
|
|
context 'when compatible version is too low' do
|
|
before do
|
|
allow(Chewy.client).to receive(:info)
|
|
.and_return({ 'version' => { 'number' => '1.2.3', 'minimum_wire_compatibility_version' => '1.0' } })
|
|
end
|
|
|
|
it 'returns false' do
|
|
expect(check.pass?).to be false
|
|
end
|
|
end
|
|
|
|
context 'when compatible version is high enough' do
|
|
before do
|
|
allow(Chewy.client).to receive(:info)
|
|
.and_return({ 'version' => { 'number' => '1.2.3', 'minimum_wire_compatibility_version' => '99.9' } })
|
|
end
|
|
|
|
it 'returns true' do
|
|
expect(check.pass?).to be true
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when running version is missing' do
|
|
before { stub_elasticsearch_error }
|
|
|
|
it 'returns false' do
|
|
expect(check.pass?).to be false
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when chewy is not enabled' do
|
|
before { allow(Chewy).to receive(:enabled?).and_return(false) }
|
|
|
|
it 'returns true' do
|
|
expect(check.pass?).to be true
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'message' do
|
|
before do
|
|
allow(Chewy).to receive(:enabled?).and_return(true)
|
|
allow(Chewy.client.cluster).to receive(:health).and_return({ 'status' => 'green', 'number_of_nodes' => 1 })
|
|
allow(Chewy.client.indices).to receive(:get_mapping).and_return({
|
|
AccountsIndex.index_name => AccountsIndex.mappings_hash.deep_stringify_keys,
|
|
StatusesIndex.index_name => StatusesIndex.mappings_hash.deep_stringify_keys,
|
|
PublicStatusesIndex.index_name => PublicStatusesIndex.mappings_hash.deep_stringify_keys,
|
|
InstancesIndex.index_name => InstancesIndex.mappings_hash.deep_stringify_keys,
|
|
TagsIndex.index_name => TagsIndex.mappings_hash.deep_stringify_keys,
|
|
})
|
|
end
|
|
|
|
context 'when running version is present' do
|
|
before { allow(Chewy.client).to receive(:info).and_return({ 'version' => { 'number' => '1.2.3' } }) }
|
|
|
|
it 'sends class name symbol to message instance' do
|
|
allow(Admin::SystemCheck::Message).to receive(:new)
|
|
.with(:elasticsearch_version_check, anything)
|
|
|
|
check.message
|
|
|
|
expect(Admin::SystemCheck::Message).to have_received(:new)
|
|
.with(:elasticsearch_version_check, 'Elasticsearch 1.2.3 is running while 7.x is required')
|
|
end
|
|
end
|
|
|
|
context 'when running version is missing' do
|
|
before { stub_elasticsearch_error }
|
|
|
|
it 'sends class name symbol to message instance' do
|
|
allow(Admin::SystemCheck::Message).to receive(:new)
|
|
.with(:elasticsearch_running_check)
|
|
|
|
check.message
|
|
|
|
expect(Admin::SystemCheck::Message).to have_received(:new)
|
|
.with(:elasticsearch_running_check)
|
|
end
|
|
end
|
|
end
|
|
|
|
def stub_elasticsearch_error
|
|
client = instance_double(Elasticsearch::Client)
|
|
allow(client).to receive(:info).and_raise(Elasticsearch::Transport::Transport::Error)
|
|
allow(Chewy).to receive(:client).and_return(client)
|
|
end
|
|
end
|