ros-apartment 2.4.0 → 2.7.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (139) hide show
  1. checksums.yaml +4 -4
  2. data/.rubocop.yml +5 -0
  3. data/.rubocop_todo.yml +1 -1
  4. data/.story_branch.yml +1 -0
  5. data/.travis.yml +13 -30
  6. data/Appraisals +4 -17
  7. data/Gemfile +2 -2
  8. data/Guardfile +3 -1
  9. data/HISTORY.md +90 -0
  10. data/README.md +31 -16
  11. data/Rakefile +34 -22
  12. data/TODO.md +0 -1
  13. data/gemfiles/rails_5_0.gemfile +10 -11
  14. data/gemfiles/rails_5_1.gemfile +10 -11
  15. data/gemfiles/rails_5_2.gemfile +9 -10
  16. data/gemfiles/rails_6_0.gemfile +10 -11
  17. data/gemfiles/rails_master.gemfile +10 -11
  18. data/lib/apartment.rb +31 -13
  19. data/lib/apartment/active_record/connection_handling.rb +17 -0
  20. data/lib/apartment/active_record/internal_metadata.rb +11 -0
  21. data/lib/apartment/active_record/log_subscriber.rb +41 -0
  22. data/lib/apartment/active_record/schema_migration.rb +13 -0
  23. data/lib/apartment/adapters/abstract_adapter.rb +17 -13
  24. data/lib/apartment/adapters/jdbc_postgresql_adapter.rb +9 -4
  25. data/lib/apartment/adapters/mysql2_adapter.rb +2 -0
  26. data/lib/apartment/adapters/postgresql_adapter.rb +33 -4
  27. data/lib/apartment/adapters/sqlite3_adapter.rb +2 -0
  28. data/lib/apartment/console.rb +7 -0
  29. data/lib/apartment/custom_console.rb +23 -7
  30. data/lib/apartment/model.rb +27 -0
  31. data/lib/apartment/railtie.rb +19 -15
  32. data/lib/apartment/tasks/task_helper.rb +35 -0
  33. data/lib/apartment/tenant.rb +16 -5
  34. data/lib/apartment/version.rb +1 -1
  35. data/lib/generators/apartment/install/templates/apartment.rb +5 -0
  36. data/lib/tasks/apartment.rake +16 -37
  37. data/{apartment.gemspec → ros-apartment.gemspec} +12 -6
  38. metadata +16 -213
  39. data/.github/workflows/.rubocop-linter.yml +0 -22
  40. data/spec/adapters/jdbc_mysql_adapter_spec.rb +0 -20
  41. data/spec/adapters/jdbc_postgresql_adapter_spec.rb +0 -39
  42. data/spec/adapters/mysql2_adapter_spec.rb +0 -61
  43. data/spec/adapters/postgresql_adapter_spec.rb +0 -63
  44. data/spec/adapters/sqlite3_adapter_spec.rb +0 -101
  45. data/spec/apartment_spec.rb +0 -13
  46. data/spec/config/database.yml.sample +0 -49
  47. data/spec/dummy/Rakefile +0 -7
  48. data/spec/dummy/app/controllers/application_controller.rb +0 -7
  49. data/spec/dummy/app/helpers/application_helper.rb +0 -4
  50. data/spec/dummy/app/models/application_record.rb +0 -6
  51. data/spec/dummy/app/models/company.rb +0 -5
  52. data/spec/dummy/app/models/user.rb +0 -5
  53. data/spec/dummy/app/views/application/index.html.erb +0 -1
  54. data/spec/dummy/app/views/layouts/application.html.erb +0 -14
  55. data/spec/dummy/config.ru +0 -6
  56. data/spec/dummy/config/application.rb +0 -51
  57. data/spec/dummy/config/boot.rb +0 -13
  58. data/spec/dummy/config/database.yml.sample +0 -44
  59. data/spec/dummy/config/environment.rb +0 -7
  60. data/spec/dummy/config/environments/development.rb +0 -29
  61. data/spec/dummy/config/environments/production.rb +0 -53
  62. data/spec/dummy/config/environments/test.rb +0 -36
  63. data/spec/dummy/config/initializers/apartment.rb +0 -6
  64. data/spec/dummy/config/initializers/backtrace_silencers.rb +0 -8
  65. data/spec/dummy/config/initializers/inflections.rb +0 -11
  66. data/spec/dummy/config/initializers/mime_types.rb +0 -6
  67. data/spec/dummy/config/initializers/secret_token.rb +0 -9
  68. data/spec/dummy/config/initializers/session_store.rb +0 -10
  69. data/spec/dummy/config/locales/en.yml +0 -5
  70. data/spec/dummy/config/routes.rb +0 -5
  71. data/spec/dummy/db/migrate/20110613152810_create_dummy_models.rb +0 -39
  72. data/spec/dummy/db/migrate/20111202022214_create_table_books.rb +0 -14
  73. data/spec/dummy/db/migrate/20180415260934_create_public_tokens.rb +0 -13
  74. data/spec/dummy/db/schema.rb +0 -55
  75. data/spec/dummy/db/seeds.rb +0 -5
  76. data/spec/dummy/db/seeds/import.rb +0 -7
  77. data/spec/dummy/public/404.html +0 -26
  78. data/spec/dummy/public/422.html +0 -26
  79. data/spec/dummy/public/500.html +0 -26
  80. data/spec/dummy/public/favicon.ico +0 -0
  81. data/spec/dummy/public/stylesheets/.gitkeep +0 -0
  82. data/spec/dummy/script/rails +0 -8
  83. data/spec/dummy_engine/.gitignore +0 -8
  84. data/spec/dummy_engine/Gemfile +0 -15
  85. data/spec/dummy_engine/Rakefile +0 -34
  86. data/spec/dummy_engine/bin/rails +0 -12
  87. data/spec/dummy_engine/config/initializers/apartment.rb +0 -52
  88. data/spec/dummy_engine/dummy_engine.gemspec +0 -24
  89. data/spec/dummy_engine/lib/dummy_engine.rb +0 -6
  90. data/spec/dummy_engine/lib/dummy_engine/engine.rb +0 -6
  91. data/spec/dummy_engine/lib/dummy_engine/version.rb +0 -5
  92. data/spec/dummy_engine/test/dummy/Rakefile +0 -6
  93. data/spec/dummy_engine/test/dummy/config.ru +0 -6
  94. data/spec/dummy_engine/test/dummy/config/application.rb +0 -24
  95. data/spec/dummy_engine/test/dummy/config/boot.rb +0 -7
  96. data/spec/dummy_engine/test/dummy/config/database.yml +0 -25
  97. data/spec/dummy_engine/test/dummy/config/environment.rb +0 -7
  98. data/spec/dummy_engine/test/dummy/config/environments/development.rb +0 -39
  99. data/spec/dummy_engine/test/dummy/config/environments/production.rb +0 -80
  100. data/spec/dummy_engine/test/dummy/config/environments/test.rb +0 -41
  101. data/spec/dummy_engine/test/dummy/config/initializers/assets.rb +0 -10
  102. data/spec/dummy_engine/test/dummy/config/initializers/backtrace_silencers.rb +0 -8
  103. data/spec/dummy_engine/test/dummy/config/initializers/cookies_serializer.rb +0 -5
  104. data/spec/dummy_engine/test/dummy/config/initializers/filter_parameter_logging.rb +0 -6
  105. data/spec/dummy_engine/test/dummy/config/initializers/inflections.rb +0 -17
  106. data/spec/dummy_engine/test/dummy/config/initializers/mime_types.rb +0 -5
  107. data/spec/dummy_engine/test/dummy/config/initializers/session_store.rb +0 -5
  108. data/spec/dummy_engine/test/dummy/config/initializers/wrap_parameters.rb +0 -16
  109. data/spec/dummy_engine/test/dummy/config/locales/en.yml +0 -23
  110. data/spec/dummy_engine/test/dummy/config/routes.rb +0 -58
  111. data/spec/dummy_engine/test/dummy/config/secrets.yml +0 -22
  112. data/spec/examples/connection_adapter_examples.rb +0 -44
  113. data/spec/examples/generic_adapter_custom_configuration_example.rb +0 -93
  114. data/spec/examples/generic_adapter_examples.rb +0 -164
  115. data/spec/examples/schema_adapter_examples.rb +0 -239
  116. data/spec/integration/apartment_rake_integration_spec.rb +0 -107
  117. data/spec/integration/query_caching_spec.rb +0 -83
  118. data/spec/integration/use_within_an_engine_spec.rb +0 -28
  119. data/spec/schemas/v1.rb +0 -15
  120. data/spec/schemas/v2.rb +0 -41
  121. data/spec/schemas/v3.rb +0 -47
  122. data/spec/spec_helper.rb +0 -63
  123. data/spec/support/apartment_helpers.rb +0 -47
  124. data/spec/support/capybara_sessions.rb +0 -15
  125. data/spec/support/config.rb +0 -13
  126. data/spec/support/contexts.rb +0 -54
  127. data/spec/support/requirements.rb +0 -48
  128. data/spec/support/setup.rb +0 -46
  129. data/spec/tasks/apartment_rake_spec.rb +0 -124
  130. data/spec/tenant_spec.rb +0 -194
  131. data/spec/unit/config_spec.rb +0 -111
  132. data/spec/unit/elevators/domain_spec.rb +0 -33
  133. data/spec/unit/elevators/first_subdomain_spec.rb +0 -26
  134. data/spec/unit/elevators/generic_spec.rb +0 -55
  135. data/spec/unit/elevators/host_hash_spec.rb +0 -33
  136. data/spec/unit/elevators/host_spec.rb +0 -89
  137. data/spec/unit/elevators/subdomain_spec.rb +0 -77
  138. data/spec/unit/migrator_spec.rb +0 -78
  139. data/spec/unit/reloader_spec.rb +0 -24
@@ -1,23 +0,0 @@
1
- # Files in the config/locales directory are used for internationalization
2
- # and are automatically loaded by Rails. If you want to use locales other
3
- # than English, add the necessary files in this directory.
4
- #
5
- # To use the locales, use `I18n.t`:
6
- #
7
- # I18n.t 'hello'
8
- #
9
- # In views, this is aliased to just `t`:
10
- #
11
- # <%= t('hello') %>
12
- #
13
- # To use a different locale, set it with `I18n.locale`:
14
- #
15
- # I18n.locale = :es
16
- #
17
- # This would use the information in config/locales/es.yml.
18
- #
19
- # To learn more, please read the Rails Internationalization guide
20
- # available at http://guides.rubyonrails.org/i18n.html.
21
-
22
- en:
23
- hello: "Hello world"
@@ -1,58 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- Rails.application.routes.draw do
4
- # The priority is based upon order of creation: first created -> highest priority.
5
- # See how all your routes lay out with "rake routes".
6
-
7
- # You can have the root of your site routed with "root"
8
- # root 'welcome#index'
9
-
10
- # Example of regular route:
11
- # get 'products/:id' => 'catalog#view'
12
-
13
- # Example of named route that can be invoked with purchase_url(id: product.id)
14
- # get 'products/:id/purchase' => 'catalog#purchase', as: :purchase
15
-
16
- # Example resource route (maps HTTP verbs to controller actions automatically):
17
- # resources :products
18
-
19
- # Example resource route with options:
20
- # resources :products do
21
- # member do
22
- # get 'short'
23
- # post 'toggle'
24
- # end
25
- #
26
- # collection do
27
- # get 'sold'
28
- # end
29
- # end
30
-
31
- # Example resource route with sub-resources:
32
- # resources :products do
33
- # resources :comments, :sales
34
- # resource :seller
35
- # end
36
-
37
- # Example resource route with more complex sub-resources:
38
- # resources :products do
39
- # resources :comments
40
- # resources :sales do
41
- # get 'recent', on: :collection
42
- # end
43
- # end
44
-
45
- # Example resource route with concerns:
46
- # concern :toggleable do
47
- # post 'toggle'
48
- # end
49
- # resources :posts, concerns: :toggleable
50
- # resources :photos, concerns: :toggleable
51
-
52
- # Example resource route within a namespace:
53
- # namespace :admin do
54
- # # Directs /admin/products/* to Admin::ProductsController
55
- # # (app/controllers/admin/products_controller.rb)
56
- # resources :products
57
- # end
58
- end
@@ -1,22 +0,0 @@
1
- # Be sure to restart your server when you modify this file.
2
-
3
- # Your secret key is used for verifying the integrity of signed cookies.
4
- # If you change this key, all old signed cookies will become invalid!
5
-
6
- # Make sure the secret is at least 30 characters and all random,
7
- # no regular words or you'll be exposed to dictionary attacks.
8
- # You can use `rake secret` to generate a secure secret key.
9
-
10
- # Make sure the secrets in this file are kept private
11
- # if you're sharing your code publicly.
12
-
13
- development:
14
- secret_key_base: bb62b819b585a74e69c797f9d03d5a004d8fe82a8e7a7da6fa2f7923030713b7b087c12cc7a918e71073c38afb343f7223d22ba3f1b223b7e76dbf8d5b65fa2c
15
-
16
- test:
17
- secret_key_base: 67945d3b189c71dffef98de2bb7c14d6fb059679c115ca3cddf65c88babe130afe4d583560d0e308b017dd76ce305bef4159d876de9fd893952d9cbf269c8476
18
-
19
- # Do not keep production secrets in the repository,
20
- # instead read values from the environment.
21
- production:
22
- secret_key_base: <%= ENV["SECRET_KEY_BASE"] %>
@@ -1,44 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- require 'spec_helper'
4
-
5
- shared_examples_for 'a connection based apartment adapter' do
6
- include Apartment::Spec::AdapterRequirements
7
-
8
- let(:default_tenant) { subject.switch { ActiveRecord::Base.connection.current_database } }
9
-
10
- describe '#init' do
11
- after do
12
- # Apartment::Tenant.init creates per model connection.
13
- # Remove the connection after testing not to unintentionally keep the connection across tests.
14
- Apartment.excluded_models.each do |excluded_model|
15
- excluded_model.constantize.remove_connection
16
- end
17
- end
18
-
19
- it 'should process model exclusions' do
20
- Apartment.configure do |config|
21
- config.excluded_models = ['Company']
22
- end
23
- Apartment::Tenant.init
24
-
25
- expect(Company.connection.object_id).not_to eq(ActiveRecord::Base.connection.object_id)
26
- end
27
- end
28
-
29
- describe '#drop' do
30
- it 'should raise an error for unknown database' do
31
- expect do
32
- subject.drop 'unknown_database'
33
- end.to raise_error(Apartment::TenantNotFound)
34
- end
35
- end
36
-
37
- describe '#switch!' do
38
- it 'should raise an error if database is invalid' do
39
- expect do
40
- subject.switch! 'unknown_database'
41
- end.to raise_error(Apartment::TenantNotFound)
42
- end
43
- end
44
- end
@@ -1,93 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- require 'spec_helper'
4
-
5
- shared_examples_for 'a generic apartment adapter able to handle custom configuration' do
6
- let(:custom_tenant_name) { 'test_tenantwwww' }
7
- let(:db) { |example| example.metadata[:database] }
8
- let(:custom_tenant_names) do
9
- {
10
- custom_tenant_name => custom_db_conf
11
- }
12
- end
13
-
14
- before do
15
- Apartment.tenant_names = custom_tenant_names
16
- Apartment.with_multi_server_setup = true
17
- end
18
-
19
- after do
20
- Apartment.with_multi_server_setup = false
21
- end
22
-
23
- context 'database key taken from specific config' do
24
- let(:expected_args) { custom_db_conf }
25
-
26
- describe '#create' do
27
- it 'should establish_connection with the separate connection with expected args' do
28
- expect(Apartment::Adapters::AbstractAdapter::SeparateDbConnectionHandler).to receive(:establish_connection).with(expected_args).and_call_original
29
-
30
- # because we dont have another server to connect to it errors
31
- # what matters is establish_connection receives proper args
32
- expect { subject.create(custom_tenant_name) }.to raise_error(Apartment::TenantExists)
33
- end
34
- end
35
-
36
- describe '#drop' do
37
- it 'should establish_connection with the separate connection with expected args' do
38
- expect(Apartment::Adapters::AbstractAdapter::SeparateDbConnectionHandler).to receive(:establish_connection).with(expected_args).and_call_original
39
-
40
- # because we dont have another server to connect to it errors
41
- # what matters is establish_connection receives proper args
42
- expect { subject.drop(custom_tenant_name) }.to raise_error(Apartment::TenantNotFound)
43
- end
44
- end
45
- end
46
-
47
- context 'database key from tenant name' do
48
- let(:expected_args) do
49
- custom_db_conf.tap { |args| args.delete(:database) }
50
- end
51
-
52
- describe '#switch!' do
53
- it 'should connect to new db' do
54
- expect(Apartment).to receive(:establish_connection) do |args|
55
- db_name = args.delete(:database)
56
-
57
- expect(args).to eq expected_args
58
- expect(db_name).to match custom_tenant_name
59
-
60
- # we only need to check args, then we short circuit
61
- # in order to avoid the mess due to the `establish_connection` override
62
- raise ActiveRecord::ActiveRecordError
63
- end
64
-
65
- expect { subject.switch!(custom_tenant_name) }.to raise_error(Apartment::TenantNotFound)
66
- end
67
- end
68
- end
69
-
70
- def specific_connection
71
- {
72
- postgresql: {
73
- adapter: 'postgresql',
74
- database: 'override_database',
75
- password: 'override_password',
76
- username: 'overridepostgres'
77
- },
78
- mysql: {
79
- adapter: 'mysql2',
80
- database: 'override_database',
81
- username: 'root'
82
- },
83
- sqlite: {
84
- adapter: 'sqlite3',
85
- database: 'override_database'
86
- }
87
- }
88
- end
89
-
90
- def custom_db_conf
91
- specific_connection[db.to_sym].with_indifferent_access
92
- end
93
- end
@@ -1,164 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- require 'spec_helper'
4
-
5
- shared_examples_for 'a generic apartment adapter' do
6
- include Apartment::Spec::AdapterRequirements
7
-
8
- before do
9
- Apartment.prepend_environment = false
10
- Apartment.append_environment = false
11
- end
12
-
13
- describe '#init' do
14
- it 'should not retain a connection after railtie' do
15
- # this test should work on rails >= 4, the connection pool code is
16
- # completely different for 3.2 so we'd have to have a messy conditional..
17
- unless Rails::VERSION::MAJOR < 4
18
- ActiveRecord::Base.connection_pool.disconnect!
19
-
20
- Apartment::Railtie.config.to_prepare_blocks.map(&:call)
21
-
22
- num_available_connections = Apartment.connection_class.connection_pool
23
- .instance_variable_get(:@available)
24
- .instance_variable_get(:@queue)
25
- .size
26
-
27
- expect(num_available_connections).to eq(1)
28
- end
29
- end
30
- end
31
-
32
- #
33
- # Creates happen already in our before_filter
34
- #
35
- describe '#create' do
36
- it 'should create the new databases' do
37
- expect(tenant_names).to include(db1)
38
- expect(tenant_names).to include(db2)
39
- end
40
-
41
- it 'should load schema.rb to new schema' do
42
- subject.switch(db1) do
43
- expect(connection.tables).to include('companies')
44
- end
45
- end
46
-
47
- it 'should yield to block if passed and reset' do
48
- subject.drop(db2) # so we don't get errors on creation
49
-
50
- @count = 0 # set our variable so its visible in and outside of blocks
51
-
52
- subject.create(db2) do
53
- @count = User.count
54
- expect(subject.current).to eq(db2)
55
- User.create
56
- end
57
-
58
- expect(subject.current).not_to eq(db2)
59
-
60
- subject.switch(db2) { expect(User.count).to eq(@count + 1) }
61
- end
62
-
63
- it 'should raise error when the schema.rb is missing unless Apartment.use_sql is set to true' do
64
- next if Apartment.use_sql
65
-
66
- subject.drop(db1)
67
- begin
68
- Dir.mktmpdir do |tmpdir|
69
- Apartment.database_schema_file = "#{tmpdir}/schema.rb"
70
- expect do
71
- subject.create(db1)
72
- end.to raise_error(Apartment::FileNotFound)
73
- end
74
- ensure
75
- Apartment.remove_instance_variable(:@database_schema_file)
76
- end
77
- end
78
- end
79
-
80
- describe '#drop' do
81
- it 'should remove the db' do
82
- subject.drop db1
83
- expect(tenant_names).not_to include(db1)
84
- end
85
- end
86
-
87
- describe '#switch!' do
88
- it 'should connect to new db' do
89
- subject.switch!(db1)
90
- expect(subject.current).to eq(db1)
91
- end
92
-
93
- it 'should reset connection if database is nil' do
94
- subject.switch!
95
- expect(subject.current).to eq(default_tenant)
96
- end
97
-
98
- it 'should raise an error if database is invalid' do
99
- expect do
100
- subject.switch! 'unknown_database'
101
- end.to raise_error(Apartment::ApartmentError)
102
- end
103
- end
104
-
105
- describe '#switch' do
106
- it 'connects and resets the tenant' do
107
- subject.switch(db1) do
108
- expect(subject.current).to eq(db1)
109
- end
110
- expect(subject.current).to eq(default_tenant)
111
- end
112
-
113
- # We're often finding when using Apartment in tests, the `current` (ie the previously connect to db)
114
- # gets dropped, but switch will try to return to that db in a test. We should just reset if it doesn't exist
115
- it 'should not throw exception if current is no longer accessible' do
116
- subject.switch!(db2)
117
-
118
- expect do
119
- subject.switch(db1) { subject.drop(db2) }
120
- end.to_not raise_error
121
- end
122
- end
123
-
124
- describe '#reset' do
125
- it 'should reset connection' do
126
- subject.switch!(db1)
127
- subject.reset
128
- expect(subject.current).to eq(default_tenant)
129
- end
130
- end
131
-
132
- describe '#current' do
133
- it 'should return the current db name' do
134
- subject.switch!(db1)
135
- expect(subject.current).to eq(db1)
136
- end
137
- end
138
-
139
- describe '#each' do
140
- it 'iterates over each tenant by default' do
141
- result = []
142
- Apartment.tenant_names = [db2, db1]
143
-
144
- subject.each do |tenant|
145
- result << tenant
146
- expect(subject.current).to eq(tenant)
147
- end
148
-
149
- expect(result).to eq([db2, db1])
150
- end
151
-
152
- it 'iterates over the given tenants' do
153
- result = []
154
- Apartment.tenant_names = [db2]
155
-
156
- subject.each([db2]) do |tenant|
157
- result << tenant
158
- expect(subject.current).to eq(tenant)
159
- end
160
-
161
- expect(result).to eq([db2])
162
- end
163
- end
164
- end
@@ -1,239 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- require 'spec_helper'
4
-
5
- # rubocop:disable Metrics/BlockLength
6
- shared_examples_for 'a schema based apartment adapter' do
7
- include Apartment::Spec::AdapterRequirements
8
-
9
- let(:schema1) { db1 }
10
- let(:schema2) { db2 }
11
- let(:public_schema) { default_tenant }
12
-
13
- describe '#init' do
14
- before do
15
- Apartment.configure do |config|
16
- config.excluded_models = ['Company']
17
- end
18
- end
19
-
20
- after do
21
- # Apartment::Tenant.init creates per model connection.
22
- # Remove the connection after testing not to unintentionally keep the connection across tests.
23
- Apartment.excluded_models.each do |excluded_model|
24
- excluded_model.constantize.remove_connection
25
- end
26
- end
27
-
28
- it 'should process model exclusions' do
29
- Apartment::Tenant.init
30
-
31
- expect(Company.table_name).to eq('public.companies')
32
- end
33
-
34
- context 'with a default_schema', default_schema: true do
35
- it 'should set the proper table_name on excluded_models' do
36
- Apartment::Tenant.init
37
-
38
- expect(Company.table_name).to eq("#{default_schema}.companies")
39
- end
40
-
41
- it 'sets the search_path correctly' do
42
- Apartment::Tenant.init
43
-
44
- expect(User.connection.schema_search_path).to match(/|#{default_schema}|/)
45
- end
46
- end
47
-
48
- context 'persistent_schemas', persistent_schemas: true do
49
- it 'sets the persistent schemas in the schema_search_path' do
50
- Apartment::Tenant.init
51
- expect(connection.schema_search_path).to end_with persistent_schemas.map { |schema| %("#{schema}") }.join(', ')
52
- end
53
- end
54
- end
55
-
56
- #
57
- # Creates happen already in our before_filter
58
- #
59
- describe '#create' do
60
- it 'should load schema.rb to new schema' do
61
- connection.schema_search_path = schema1
62
- expect(connection.tables).to include('companies')
63
- end
64
-
65
- it 'should yield to block if passed and reset' do
66
- subject.drop(schema2) # so we don't get errors on creation
67
-
68
- @count = 0 # set our variable so its visible in and outside of blocks
69
-
70
- subject.create(schema2) do
71
- @count = User.count
72
- expect(connection.schema_search_path).to start_with %("#{schema2}")
73
- User.create
74
- end
75
-
76
- expect(connection.schema_search_path).not_to start_with %("#{schema2}")
77
-
78
- subject.switch(schema2) { expect(User.count).to eq(@count + 1) }
79
- end
80
-
81
- context 'numeric database names' do
82
- let(:db) { 1234 }
83
- it 'should allow them' do
84
- expect do
85
- subject.create(db)
86
- end.to_not raise_error
87
- expect(tenant_names).to include(db.to_s)
88
- end
89
-
90
- after { subject.drop(db) }
91
- end
92
- end
93
-
94
- describe '#drop' do
95
- it 'should raise an error for unknown database' do
96
- expect do
97
- subject.drop 'unknown_database'
98
- end.to raise_error(Apartment::TenantNotFound)
99
- end
100
-
101
- context 'numeric database names' do
102
- let(:db) { 1234 }
103
-
104
- it 'should be able to drop them' do
105
- subject.create(db)
106
- expect do
107
- subject.drop(db)
108
- end.to_not raise_error
109
- expect(tenant_names).not_to include(db.to_s)
110
- end
111
-
112
- after do
113
- begin
114
- subject.drop(db)
115
- rescue StandardError => _e
116
- nil
117
- end
118
- end
119
- end
120
- end
121
-
122
- describe '#switch' do
123
- it 'connects and resets' do
124
- subject.switch(schema1) do
125
- expect(connection.schema_search_path).to start_with %("#{schema1}")
126
- end
127
-
128
- expect(connection.schema_search_path).to start_with %("#{public_schema}")
129
- end
130
- end
131
-
132
- describe '#reset' do
133
- it 'should reset connection' do
134
- subject.switch!(schema1)
135
- subject.reset
136
- expect(connection.schema_search_path).to start_with %("#{public_schema}")
137
- end
138
-
139
- context 'with default_schema', default_schema: true do
140
- it 'should reset to the default schema' do
141
- subject.switch!(schema1)
142
- subject.reset
143
- expect(connection.schema_search_path).to start_with %("#{default_schema}")
144
- end
145
- end
146
-
147
- context 'persistent_schemas', persistent_schemas: true do
148
- before do
149
- subject.switch!(schema1)
150
- subject.reset
151
- end
152
-
153
- it 'maintains the persistent schemas in the schema_search_path' do
154
- expect(connection.schema_search_path).to end_with persistent_schemas.map { |schema| %("#{schema}") }.join(', ')
155
- end
156
-
157
- context 'with default_schema', default_schema: true do
158
- it 'prioritizes the switched schema to front of schema_search_path' do
159
- subject.reset # need to re-call this as the default_schema wasn't set at the time that the above reset ran
160
- expect(connection.schema_search_path).to start_with %("#{default_schema}")
161
- end
162
- end
163
- end
164
- end
165
-
166
- describe '#switch!' do
167
- it 'should connect to new schema' do
168
- subject.switch!(schema1)
169
- expect(connection.schema_search_path).to start_with %("#{schema1}")
170
- end
171
-
172
- it 'should reset connection if database is nil' do
173
- subject.switch!
174
- expect(connection.schema_search_path).to eq(%("#{public_schema}"))
175
- end
176
-
177
- it 'should raise an error if schema is invalid' do
178
- expect do
179
- subject.switch! 'unknown_schema'
180
- end.to raise_error(Apartment::TenantNotFound)
181
- end
182
-
183
- context 'numeric databases' do
184
- let(:db) { 1234 }
185
-
186
- it 'should connect to them' do
187
- subject.create(db)
188
- expect do
189
- subject.switch!(db)
190
- end.to_not raise_error
191
-
192
- expect(connection.schema_search_path).to start_with %("#{db}")
193
- end
194
-
195
- after { subject.drop(db) }
196
- end
197
-
198
- describe 'with default_schema specified', default_schema: true do
199
- before do
200
- subject.switch!(schema1)
201
- end
202
-
203
- it 'should switch out the default schema rather than public' do
204
- expect(connection.schema_search_path).not_to include default_schema
205
- end
206
-
207
- it 'should still switch to the switched schema' do
208
- expect(connection.schema_search_path).to start_with %("#{schema1}")
209
- end
210
- end
211
-
212
- context 'persistent_schemas', persistent_schemas: true do
213
- before { subject.switch!(schema1) }
214
-
215
- it 'maintains the persistent schemas in the schema_search_path' do
216
- expect(connection.schema_search_path).to end_with persistent_schemas.map { |schema| %("#{schema}") }.join(', ')
217
- end
218
-
219
- it 'prioritizes the switched schema to front of schema_search_path' do
220
- expect(connection.schema_search_path).to start_with %("#{schema1}")
221
- end
222
- end
223
- end
224
-
225
- describe '#current' do
226
- it 'should return the current schema name' do
227
- subject.switch!(schema1)
228
- expect(subject.current).to eq(schema1)
229
- end
230
-
231
- context 'persistent_schemas', persistent_schemas: true do
232
- it 'should exlude persistent_schemas' do
233
- subject.switch!(schema1)
234
- expect(subject.current).to eq(schema1)
235
- end
236
- end
237
- end
238
- end
239
- # rubocop:enable Metrics/BlockLength