mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 15:49:55 +08:00
427d54b2b0
Zeitwerk simplifies working with dependencies in dev and makes it easier reloading class chains. We no longer need to use Rails "require_dependency" anywhere and instead can just use standard Ruby patterns to require files. This is a far reaching change and we expect some followups here.
94 lines
2.3 KiB
Ruby
94 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rails_helper'
|
|
|
|
describe Migration::TableDropper do
|
|
|
|
def table_exists?(table_name)
|
|
DB.exec(<<~SQL) > 0
|
|
SELECT 1
|
|
FROM INFORMATION_SCHEMA.TABLES
|
|
WHERE table_schema = 'public' AND
|
|
table_name = '#{table_name}'
|
|
SQL
|
|
end
|
|
|
|
def function_exists?(function_name, schema_name = 'public')
|
|
DB.exec(<<~SQL) > 0
|
|
SELECT 1
|
|
FROM information_schema.routines
|
|
WHERE routine_type = 'FUNCTION' AND
|
|
routine_name = '#{function_name}' AND
|
|
specific_schema = '#{schema_name}'
|
|
SQL
|
|
end
|
|
|
|
let(:table_name) { 'table_with_old_name' }
|
|
|
|
before do
|
|
DB.exec "CREATE TABLE #{table_name} (topic_id INTEGER)"
|
|
|
|
DB.exec <<~SQL
|
|
INSERT INTO #{table_name} (topic_id) VALUES (1)
|
|
SQL
|
|
end
|
|
|
|
describe ".execute_drop" do
|
|
it "should drop the table" do
|
|
Migration::TableDropper.execute_drop(table_name)
|
|
|
|
expect(table_exists?(table_name)).to eq(false)
|
|
end
|
|
end
|
|
|
|
describe ".readonly_only_table" do
|
|
before do
|
|
Migration::TableDropper.read_only_table(table_name)
|
|
end
|
|
|
|
after do
|
|
ActiveRecord::Base.connection.reset!
|
|
|
|
DB.exec(<<~SQL)
|
|
DROP TABLE IF EXISTS #{table_name};
|
|
DROP FUNCTION IF EXISTS #{Migration::BaseDropper.readonly_function_name(table_name)} CASCADE;
|
|
SQL
|
|
end
|
|
|
|
it 'should be droppable' do
|
|
Migration::TableDropper.execute_drop(table_name)
|
|
|
|
expect(has_trigger?(Migration::BaseDropper.readonly_trigger_name(
|
|
table_name
|
|
))).to eq(false)
|
|
|
|
expect(table_exists?(table_name)).to eq(false)
|
|
end
|
|
|
|
it "should drop the read_only function" do
|
|
Migration::TableDropper.execute_drop(table_name)
|
|
|
|
schema_name, function_name = Migration::BaseDropper
|
|
.readonly_function_name(table_name)
|
|
.delete_suffix('()').split('.')
|
|
|
|
expect(function_exists?(function_name, schema_name)).to eq(false)
|
|
end
|
|
|
|
it 'should prevent insertions to the table' do
|
|
begin
|
|
DB.exec <<~SQL
|
|
INSERT INTO #{table_name} (topic_id) VALUES (2)
|
|
SQL
|
|
rescue PG::RaiseException => e
|
|
[
|
|
"Discourse: #{table_name} is read only",
|
|
'discourse_functions.raise_table_with_old_name_readonly()'
|
|
].each do |message|
|
|
expect(e.message).to include(message)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|