cartodb-4.42/spec/support/import_database_connection.rb

38 lines
987 B
Ruby
Raw Normal View History

2024-04-06 13:25:13 +08:00
module CartoDB
class ImportDatabaseConnection
def self.connection
@@connection ||= nil
if @@connection
@@connection
else
c = ::Sequel.connect('postgres://postgres:@localhost:5432/cartodb_importer_test')
begin
c.test_connection
@@connection = c
rescue StandardError
c = ::Sequel.connect('postgres://postgres:@localhost:5432')
c.run <<-SQL
CREATE DATABASE cartodb_importer_test
WITH TEMPLATE = template_postgis
OWNER = postgres
SQL
@@connection = ::Sequel.connect('postgres://postgres:@localhost:5432/cartodb_importer_test')
end
return @@connection
end
end
def self.drop
@@connection.disconnect
@@connection = nil
begin
c = ::Sequel.connect('postgres://postgres:@localhost:5432')
c.run "DROP DATABASE cartodb_importer_test"
rescue StandardError => e
raise e
end
return true
end
end
end