ibm_db 5.2.0-x86-mingw32 → 5.3.2-x86-mingw32
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGES +9 -0
- data/LICENSE +55 -18
- data/ext/Makefile +15 -13
- data/ext/ibm_db.c +62 -57
- data/ext/ibm_db.o +0 -0
- data/ext/ibm_db.so +0 -0
- data/ext/mkmf.log +26 -24
- data/ext/ruby_ibm_db_cli.c +1 -0
- data/ext/ruby_ibm_db_cli.o +0 -0
- data/lib/active_record/connection_adapters/ibm_db_adapter.rb +1463 -1279
- data/lib/ibm_db.so +1 -0
- data/lib/mswin32/rb3x/i386/ruby30/ibm_db.so +0 -0
- data/test/active_record/connection_adapters/fake_adapter.rb +5 -2
- data/test/activejob/destroy_association_async_test.rb +305 -0
- data/test/activejob/destroy_async_job_not_present_test.rb +31 -0
- data/test/activejob/helper.rb +15 -0
- data/test/assets/schema_dump_5_1.yml +345 -0
- data/test/cases/adapter_prevent_writes_test.rb +334 -0
- data/test/cases/adapter_test.rb +432 -218
- data/test/cases/adapters/mysql2/active_schema_test.rb +85 -75
- data/test/cases/adapters/mysql2/auto_increment_test.rb +34 -0
- data/test/cases/adapters/mysql2/bind_parameter_test.rb +5 -3
- data/test/cases/adapters/mysql2/boolean_test.rb +6 -4
- data/test/cases/adapters/mysql2/case_sensitivity_test.rb +26 -24
- data/test/cases/adapters/mysql2/charset_collation_test.rb +20 -17
- data/test/cases/adapters/mysql2/connection_test.rb +48 -50
- data/test/cases/adapters/mysql2/count_deleted_rows_with_lock_test.rb +28 -0
- data/test/cases/adapters/mysql2/datetime_precision_quoting_test.rb +23 -19
- data/test/cases/adapters/mysql2/enum_test.rb +32 -11
- data/test/cases/adapters/mysql2/explain_test.rb +13 -11
- data/test/cases/adapters/mysql2/json_test.rb +17 -188
- data/test/cases/adapters/mysql2/mysql2_adapter_prevent_writes_test.rb +208 -0
- data/test/cases/adapters/mysql2/mysql2_adapter_test.rb +183 -28
- data/test/cases/adapters/mysql2/nested_deadlock_test.rb +75 -0
- data/test/cases/adapters/mysql2/optimizer_hints_test.rb +69 -0
- data/test/cases/adapters/mysql2/schema_migrations_test.rb +26 -21
- data/test/cases/adapters/mysql2/schema_test.rb +24 -22
- data/test/cases/adapters/mysql2/set_test.rb +32 -0
- data/test/cases/adapters/mysql2/sp_test.rb +10 -8
- data/test/cases/adapters/mysql2/sql_types_test.rb +8 -6
- data/test/cases/adapters/mysql2/table_options_test.rb +93 -10
- data/test/cases/adapters/mysql2/transaction_test.rb +151 -0
- data/test/cases/adapters/mysql2/unsigned_type_test.rb +11 -9
- data/test/cases/adapters/mysql2/virtual_column_test.rb +66 -0
- data/test/cases/adapters/postgresql/active_schema_test.rb +40 -25
- data/test/cases/adapters/postgresql/array_test.rb +118 -63
- data/test/cases/adapters/postgresql/bit_string_test.rb +12 -10
- data/test/cases/adapters/postgresql/bytea_test.rb +26 -25
- data/test/cases/adapters/postgresql/case_insensitive_test.rb +10 -9
- data/test/cases/adapters/postgresql/change_schema_test.rb +7 -5
- data/test/cases/adapters/postgresql/cidr_test.rb +2 -0
- data/test/cases/adapters/postgresql/citext_test.rb +58 -58
- data/test/cases/adapters/postgresql/collation_test.rb +17 -15
- data/test/cases/adapters/postgresql/composite_test.rb +25 -23
- data/test/cases/adapters/postgresql/connection_test.rb +73 -85
- data/test/cases/adapters/postgresql/create_unlogged_tables_test.rb +74 -0
- data/test/cases/adapters/postgresql/datatype_test.rb +19 -22
- data/test/cases/adapters/postgresql/date_test.rb +42 -0
- data/test/cases/adapters/postgresql/domain_test.rb +9 -7
- data/test/cases/adapters/postgresql/enum_test.rb +12 -10
- data/test/cases/adapters/postgresql/explain_test.rb +10 -8
- data/test/cases/adapters/postgresql/extension_migration_test.rb +13 -12
- data/test/cases/adapters/postgresql/foreign_table_test.rb +109 -0
- data/test/cases/adapters/postgresql/full_text_test.rb +8 -6
- data/test/cases/adapters/postgresql/geometric_test.rb +57 -63
- data/test/cases/adapters/postgresql/hstore_test.rb +288 -280
- data/test/cases/adapters/postgresql/infinity_test.rb +54 -15
- data/test/cases/adapters/postgresql/integer_test.rb +2 -0
- data/test/cases/adapters/postgresql/interval_test.rb +99 -0
- data/test/cases/adapters/postgresql/json_test.rb +16 -201
- data/test/cases/adapters/postgresql/ltree_test.rb +14 -16
- data/test/cases/adapters/postgresql/money_test.rb +47 -16
- data/test/cases/adapters/postgresql/network_test.rb +36 -28
- data/test/cases/adapters/postgresql/numbers_test.rb +7 -5
- data/test/cases/adapters/postgresql/optimizer_hints_test.rb +71 -0
- data/test/cases/adapters/postgresql/partitions_test.rb +22 -0
- data/test/cases/adapters/postgresql/postgresql_adapter_prevent_writes_test.rb +205 -0
- data/test/cases/adapters/postgresql/postgresql_adapter_test.rb +178 -136
- data/test/cases/adapters/postgresql/prepared_statements_disabled_test.rb +27 -0
- data/test/cases/adapters/postgresql/quoting_test.rb +12 -6
- data/test/cases/adapters/postgresql/range_test.rb +406 -292
- data/test/cases/adapters/postgresql/referential_integrity_test.rb +16 -15
- data/test/cases/adapters/postgresql/rename_table_test.rb +9 -8
- data/test/cases/adapters/postgresql/schema_authorization_test.rb +14 -23
- data/test/cases/adapters/postgresql/schema_test.rb +207 -91
- data/test/cases/adapters/postgresql/serial_test.rb +9 -7
- data/test/cases/adapters/postgresql/statement_pool_test.rb +26 -6
- data/test/cases/adapters/postgresql/timestamp_test.rb +17 -15
- data/test/cases/adapters/postgresql/transaction_nested_test.rb +114 -0
- data/test/cases/adapters/postgresql/transaction_test.rb +189 -0
- data/test/cases/adapters/postgresql/type_lookup_test.rb +12 -10
- data/test/cases/adapters/postgresql/utils_test.rb +11 -9
- data/test/cases/adapters/postgresql/uuid_test.rb +226 -109
- data/test/cases/adapters/postgresql/xml_test.rb +10 -14
- data/test/cases/adapters/sqlite3/collation_test.rb +26 -15
- data/test/cases/adapters/sqlite3/copy_table_test.rb +31 -28
- data/test/cases/adapters/sqlite3/explain_test.rb +13 -11
- data/test/cases/adapters/sqlite3/json_test.rb +29 -0
- data/test/cases/adapters/sqlite3/quoting_test.rb +35 -57
- data/test/cases/adapters/sqlite3/sqlite3_adapter_prevent_writes_test.rb +186 -0
- data/test/cases/adapters/sqlite3/sqlite3_adapter_test.rb +318 -131
- data/test/cases/adapters/sqlite3/sqlite3_create_folder_test.rb +11 -11
- data/test/cases/adapters/sqlite3/statement_pool_test.rb +7 -6
- data/test/cases/adapters/sqlite3/transaction_test.rb +123 -0
- data/test/cases/aggregations_test.rb +14 -12
- data/test/cases/annotate_test.rb +46 -0
- data/test/cases/ar_schema_test.rb +153 -86
- data/test/cases/arel/attributes/attribute_test.rb +1145 -0
- data/test/cases/arel/attributes/math_test.rb +83 -0
- data/test/cases/arel/attributes_test.rb +27 -0
- data/test/cases/arel/collectors/bind_test.rb +40 -0
- data/test/cases/arel/collectors/composite_test.rb +47 -0
- data/test/cases/arel/collectors/sql_string_test.rb +41 -0
- data/test/cases/arel/collectors/substitute_bind_collector_test.rb +48 -0
- data/test/cases/arel/crud_test.rb +65 -0
- data/test/cases/arel/delete_manager_test.rb +53 -0
- data/test/cases/arel/factory_methods_test.rb +46 -0
- data/test/cases/arel/helper.rb +45 -0
- data/test/cases/arel/insert_manager_test.rb +241 -0
- data/test/cases/arel/nodes/and_test.rb +30 -0
- data/test/cases/arel/nodes/as_test.rb +36 -0
- data/test/cases/arel/nodes/ascending_test.rb +46 -0
- data/test/cases/arel/nodes/bin_test.rb +35 -0
- data/test/cases/arel/nodes/binary_test.rb +29 -0
- data/test/cases/arel/nodes/bind_param_test.rb +22 -0
- data/test/cases/arel/nodes/case_test.rb +96 -0
- data/test/cases/arel/nodes/casted_test.rb +18 -0
- data/test/cases/arel/nodes/comment_test.rb +22 -0
- data/test/cases/arel/nodes/count_test.rb +35 -0
- data/test/cases/arel/nodes/delete_statement_test.rb +36 -0
- data/test/cases/arel/nodes/descending_test.rb +46 -0
- data/test/cases/arel/nodes/distinct_test.rb +21 -0
- data/test/cases/arel/nodes/equality_test.rb +62 -0
- data/test/cases/arel/nodes/extract_test.rb +43 -0
- data/test/cases/arel/nodes/false_test.rb +21 -0
- data/test/cases/arel/nodes/grouping_test.rb +26 -0
- data/test/cases/arel/nodes/infix_operation_test.rb +42 -0
- data/test/cases/arel/nodes/insert_statement_test.rb +44 -0
- data/test/cases/arel/nodes/named_function_test.rb +48 -0
- data/test/cases/arel/nodes/node_test.rb +22 -0
- data/test/cases/arel/nodes/not_test.rb +31 -0
- data/test/cases/arel/nodes/or_test.rb +36 -0
- data/test/cases/arel/nodes/over_test.rb +69 -0
- data/test/cases/arel/nodes/select_core_test.rb +79 -0
- data/test/cases/arel/nodes/select_statement_test.rb +51 -0
- data/test/cases/arel/nodes/sql_literal_test.rb +75 -0
- data/test/cases/arel/nodes/sum_test.rb +35 -0
- data/test/cases/arel/nodes/table_alias_test.rb +29 -0
- data/test/cases/arel/nodes/true_test.rb +21 -0
- data/test/cases/arel/nodes/unary_operation_test.rb +41 -0
- data/test/cases/arel/nodes/update_statement_test.rb +60 -0
- data/test/cases/arel/nodes/window_test.rb +81 -0
- data/test/cases/arel/nodes_test.rb +34 -0
- data/test/cases/arel/select_manager_test.rb +1238 -0
- data/test/cases/arel/support/fake_record.rb +135 -0
- data/test/cases/arel/table_test.rb +216 -0
- data/test/cases/arel/update_manager_test.rb +126 -0
- data/test/cases/arel/visitors/dispatch_contamination_test.rb +78 -0
- data/test/cases/arel/visitors/dot_test.rb +90 -0
- data/test/cases/arel/visitors/mysql_test.rb +157 -0
- data/test/cases/arel/visitors/postgres_test.rb +366 -0
- data/test/cases/arel/visitors/sqlite_test.rb +75 -0
- data/test/cases/arel/visitors/to_sql_test.rb +750 -0
- data/test/cases/associations/belongs_to_associations_test.rb +510 -158
- data/test/cases/associations/bidirectional_destroy_dependencies_test.rb +4 -2
- data/test/cases/associations/callbacks_test.rb +56 -38
- data/test/cases/associations/cascaded_eager_loading_test.rb +118 -61
- data/test/cases/associations/eager_load_includes_full_sti_class_test.rb +138 -18
- data/test/cases/associations/eager_load_nested_include_test.rb +38 -37
- data/test/cases/associations/eager_singularization_test.rb +21 -21
- data/test/cases/associations/eager_test.rb +559 -415
- data/test/cases/associations/extension_test.rb +18 -12
- data/test/cases/associations/has_and_belongs_to_many_associations_test.rb +234 -213
- data/test/cases/associations/has_many_associations_test.rb +1038 -465
- data/test/cases/associations/has_many_through_associations_test.rb +558 -249
- data/test/cases/associations/has_one_associations_test.rb +294 -129
- data/test/cases/associations/has_one_through_associations_test.rb +121 -75
- data/test/cases/associations/inner_join_association_test.rb +114 -38
- data/test/cases/associations/inverse_associations_test.rb +606 -398
- data/test/cases/associations/join_model_test.rb +158 -148
- data/test/cases/associations/left_outer_join_association_test.rb +59 -24
- data/test/cases/associations/nested_through_associations_test.rb +166 -109
- data/test/cases/associations/required_test.rb +35 -10
- data/test/cases/associations_test.rb +241 -110
- data/test/cases/attribute_methods/read_test.rb +11 -11
- data/test/cases/attribute_methods_test.rb +413 -298
- data/test/cases/attributes_test.rb +145 -27
- data/test/cases/autosave_association_test.rb +681 -436
- data/test/cases/base_prevent_writes_test.rb +229 -0
- data/test/cases/base_test.rb +599 -542
- data/test/cases/batches_test.rb +288 -82
- data/test/cases/binary_test.rb +26 -31
- data/test/cases/bind_parameter_test.rb +194 -21
- data/test/cases/boolean_test.rb +52 -0
- data/test/cases/cache_key_test.rb +110 -5
- data/test/cases/calculations_test.rb +740 -177
- data/test/cases/callbacks_test.rb +74 -207
- data/test/cases/clone_test.rb +15 -10
- data/test/cases/coders/json_test.rb +2 -0
- data/test/cases/coders/yaml_column_test.rb +16 -13
- data/test/cases/collection_cache_key_test.rb +177 -20
- data/test/cases/column_alias_test.rb +9 -7
- data/test/cases/column_definition_test.rb +10 -68
- data/test/cases/comment_test.rb +166 -107
- data/test/cases/connection_adapters/adapter_leasing_test.rb +14 -10
- data/test/cases/connection_adapters/connection_handler_test.rb +358 -51
- data/test/cases/connection_adapters/connection_handlers_multi_db_test.rb +400 -0
- data/test/cases/connection_adapters/connection_handlers_multi_pool_config_test.rb +103 -0
- data/test/cases/connection_adapters/connection_handlers_sharding_db_test.rb +499 -0
- data/test/cases/connection_adapters/connection_swapping_nested_test.rb +457 -0
- data/test/cases/connection_adapters/legacy_connection_handlers_multi_db_test.rb +486 -0
- data/test/cases/connection_adapters/legacy_connection_handlers_sharding_db_test.rb +586 -0
- data/test/cases/connection_adapters/merge_and_resolve_default_url_config_test.rb +319 -138
- data/test/cases/connection_adapters/mysql_type_lookup_test.rb +62 -50
- data/test/cases/connection_adapters/schema_cache_test.rb +259 -26
- data/test/cases/connection_adapters/type_lookup_test.rb +96 -95
- data/test/cases/connection_management_test.rb +13 -11
- data/test/cases/connection_pool_test.rb +316 -83
- data/test/cases/core_test.rb +82 -58
- data/test/cases/counter_cache_test.rb +204 -50
- data/test/cases/custom_locking_test.rb +5 -3
- data/test/cases/database_configurations/hash_config_test.rb +74 -0
- data/test/cases/database_configurations/resolver_test.rb +150 -0
- data/test/cases/database_configurations_test.rb +145 -0
- data/test/cases/database_selector_test.rb +296 -0
- data/test/cases/database_statements_test.rb +18 -16
- data/test/cases/date_test.rb +8 -16
- data/test/cases/date_time_precision_test.rb +100 -78
- data/test/cases/date_time_test.rb +23 -8
- data/test/cases/defaults_test.rb +106 -71
- data/test/cases/delegated_type_test.rb +57 -0
- data/test/cases/dirty_test.rb +419 -223
- data/test/cases/disconnected_test.rb +6 -6
- data/test/cases/dup_test.rb +54 -27
- data/test/cases/enum_test.rb +461 -82
- data/test/cases/errors_test.rb +7 -7
- data/test/cases/explain_subscriber_test.rb +17 -15
- data/test/cases/explain_test.rb +11 -19
- data/test/cases/filter_attributes_test.rb +153 -0
- data/test/cases/finder_respond_to_test.rb +14 -14
- data/test/cases/finder_test.rb +669 -287
- data/test/cases/fixture_set/file_test.rb +34 -38
- data/test/cases/fixtures_test.rb +833 -176
- data/test/cases/forbidden_attributes_protection_test.rb +32 -67
- data/test/cases/habtm_destroy_order_test.rb +25 -25
- data/test/cases/helper.rb +78 -49
- data/test/cases/hot_compatibility_test.rb +33 -32
- data/test/cases/i18n_test.rb +18 -17
- data/test/cases/inheritance_test.rb +180 -115
- data/test/cases/insert_all_test.rb +489 -0
- data/test/cases/instrumentation_test.rb +101 -0
- data/test/cases/integration_test.rb +119 -31
- data/test/cases/invalid_connection_test.rb +18 -16
- data/test/cases/invertible_migration_test.rb +183 -43
- data/test/cases/json_attribute_test.rb +35 -0
- data/test/cases/json_serialization_test.rb +57 -58
- data/test/cases/json_shared_test_cases.rb +290 -0
- data/test/cases/locking_test.rb +413 -119
- data/test/cases/log_subscriber_test.rb +68 -26
- data/test/cases/marshal_serialization_test.rb +39 -0
- data/test/cases/migration/change_schema_test.rb +118 -72
- data/test/cases/migration/change_table_test.rb +138 -30
- data/test/cases/migration/check_constraint_test.rb +162 -0
- data/test/cases/migration/column_attributes_test.rb +45 -35
- data/test/cases/migration/column_positioning_test.rb +18 -6
- data/test/cases/migration/columns_test.rb +93 -77
- data/test/cases/migration/command_recorder_test.rb +121 -34
- data/test/cases/migration/compatibility_test.rb +578 -23
- data/test/cases/migration/create_join_table_test.rb +35 -25
- data/test/cases/migration/foreign_key_test.rb +503 -284
- data/test/cases/migration/helper.rb +4 -3
- data/test/cases/migration/index_test.rb +119 -70
- data/test/cases/migration/logger_test.rb +9 -6
- data/test/cases/migration/pending_migrations_test.rb +88 -34
- data/test/cases/migration/references_foreign_key_test.rb +164 -150
- data/test/cases/migration/references_index_test.rb +38 -19
- data/test/cases/migration/references_statements_test.rb +15 -14
- data/test/cases/migration/rename_table_test.rb +53 -30
- data/test/cases/migration_test.rb +637 -269
- data/test/cases/migrator_test.rb +191 -135
- data/test/cases/mixin_test.rb +7 -11
- data/test/cases/modules_test.rb +36 -34
- data/test/cases/multi_db_migrator_test.rb +223 -0
- data/test/cases/multiparameter_attributes_test.rb +60 -33
- data/test/cases/multiple_db_test.rb +16 -22
- data/test/cases/nested_attributes_test.rb +341 -320
- data/test/cases/nested_attributes_with_callbacks_test.rb +26 -24
- data/test/cases/null_relation_test.rb +84 -0
- data/test/cases/numeric_data_test.rb +93 -0
- data/test/cases/persistence_test.rb +361 -269
- data/test/cases/pooled_connections_test.rb +18 -26
- data/test/cases/prepared_statement_status_test.rb +48 -0
- data/test/cases/primary_keys_test.rb +210 -104
- data/test/cases/query_cache_test.rb +610 -141
- data/test/cases/quoting_test.rb +132 -31
- data/test/cases/readonly_test.rb +49 -48
- data/test/cases/reaper_test.rb +146 -32
- data/test/cases/reflection_test.rb +167 -156
- data/test/cases/relation/delegation_test.rb +49 -36
- data/test/cases/relation/delete_all_test.rb +117 -0
- data/test/cases/relation/merging_test.rb +319 -42
- data/test/cases/relation/mutation_test.rb +55 -93
- data/test/cases/relation/or_test.rb +129 -29
- data/test/cases/relation/predicate_builder_test.rb +21 -6
- data/test/cases/relation/record_fetch_warning_test.rb +5 -3
- data/test/cases/relation/select_test.rb +67 -0
- data/test/cases/relation/update_all_test.rb +317 -0
- data/test/cases/relation/where_chain_test.rb +68 -32
- data/test/cases/relation/where_clause_test.rb +136 -61
- data/test/cases/relation/where_test.rb +155 -48
- data/test/cases/relation_test.rb +266 -112
- data/test/cases/relations_test.rb +969 -744
- data/test/cases/reload_models_test.rb +13 -9
- data/test/cases/reserved_word_test.rb +141 -0
- data/test/cases/result_test.rb +68 -17
- data/test/cases/sanitize_test.rb +87 -71
- data/test/cases/schema_dumper_test.rb +221 -128
- data/test/cases/schema_loading_test.rb +3 -2
- data/test/cases/scoping/default_scoping_test.rb +185 -144
- data/test/cases/scoping/named_scoping_test.rb +177 -89
- data/test/cases/scoping/relation_scoping_test.rb +197 -75
- data/test/cases/secure_token_test.rb +18 -3
- data/test/cases/serialization_test.rb +30 -28
- data/test/cases/serialized_attribute_test.rb +133 -42
- data/test/cases/signed_id_test.rb +168 -0
- data/test/cases/statement_cache_test.rb +41 -24
- data/test/cases/statement_invalid_test.rb +42 -0
- data/test/cases/store_test.rb +180 -55
- data/test/cases/strict_loading_test.rb +473 -0
- data/test/cases/suppressor_test.rb +26 -12
- data/test/cases/tasks/database_tasks_test.rb +1258 -194
- data/test/cases/tasks/mysql_rake_test.rb +370 -298
- data/test/cases/tasks/postgresql_rake_test.rb +481 -251
- data/test/cases/tasks/sqlite_rake_test.rb +225 -178
- data/test/cases/test_case.rb +51 -40
- data/test/cases/test_databases_test.rb +79 -0
- data/test/cases/test_fixtures_test.rb +79 -19
- data/test/cases/time_precision_test.rb +98 -76
- data/test/cases/timestamp_test.rb +102 -99
- data/test/cases/touch_later_test.rb +12 -10
- data/test/cases/transaction_callbacks_test.rb +344 -90
- data/test/cases/transaction_isolation_test.rb +12 -12
- data/test/cases/transactions_test.rb +612 -162
- data/test/cases/type/adapter_specific_registry_test.rb +14 -2
- data/test/cases/type/date_time_test.rb +4 -2
- data/test/cases/type/integer_test.rb +4 -2
- data/test/cases/type/string_test.rb +10 -8
- data/test/cases/type/time_test.rb +28 -0
- data/test/cases/type/type_map_test.rb +29 -28
- data/test/cases/type/unsigned_integer_test.rb +19 -0
- data/test/cases/type_test.rb +2 -0
- data/test/cases/types_test.rb +3 -1
- data/test/cases/unconnected_test.rb +14 -1
- data/test/cases/unsafe_raw_sql_test.rb +274 -0
- data/test/cases/validations/absence_validation_test.rb +19 -17
- data/test/cases/validations/association_validation_test.rb +30 -28
- data/test/cases/validations/i18n_generate_message_validation_test.rb +34 -16
- data/test/cases/validations/i18n_validation_test.rb +22 -21
- data/test/cases/validations/length_validation_test.rb +34 -33
- data/test/cases/validations/numericality_validation_test.rb +181 -0
- data/test/cases/validations/presence_validation_test.rb +21 -19
- data/test/cases/validations/uniqueness_validation_test.rb +156 -86
- data/test/cases/validations_repair_helper.rb +2 -0
- data/test/cases/validations_test.rb +61 -26
- data/test/cases/view_test.rb +122 -116
- data/test/cases/yaml_serialization_test.rb +79 -34
- data/test/config.example.yml +19 -19
- data/test/config.rb +3 -1
- data/test/config.yml +16 -6
- data/test/fixtures/all/namespaced/accounts.yml +2 -0
- data/test/fixtures/author_addresses.yml +1 -8
- data/test/fixtures/authors.yml +1 -7
- data/test/fixtures/binaries.yml +4 -0
- data/test/fixtures/books.yml +9 -2
- data/test/fixtures/categories_posts.yml +3 -0
- data/test/fixtures/citations.yml +5 -0
- data/test/fixtures/comments.yml +7 -0
- data/test/fixtures/companies.yml +5 -0
- data/test/fixtures/computers.yml +2 -0
- data/test/fixtures/customers.yml +10 -1
- data/test/fixtures/developers.yml +1 -1
- data/test/fixtures/essays.yml +10 -0
- data/test/fixtures/faces.yml +3 -3
- data/test/fixtures/humans.yml +5 -0
- data/test/fixtures/interests.yml +7 -7
- data/test/fixtures/memberships.yml +7 -0
- data/test/fixtures/minimalistics.yml +3 -0
- data/test/fixtures/mixed_case_monkeys.yml +2 -2
- data/test/fixtures/naked/yml/courses_with_invalid_key.yml +3 -0
- data/test/fixtures/naked/yml/parrots.yml +1 -0
- data/test/fixtures/other_books.yml +26 -0
- data/test/fixtures/other_posts.yml +1 -0
- data/test/fixtures/parrots.yml +7 -1
- data/test/fixtures/pirates.yml +3 -0
- data/test/fixtures/posts.yml +11 -3
- data/test/fixtures/readers.yml +6 -0
- data/test/fixtures/reserved_words/values.yml +2 -2
- data/test/fixtures/sponsors.yml +3 -0
- data/test/fixtures/strict_zines.yml +2 -0
- data/test/fixtures/subscribers.yml +1 -1
- data/test/fixtures/tasks.yml +1 -1
- data/test/fixtures/warehouse-things.yml +3 -0
- data/test/migrations/10_urban/9_add_expressions.rb +2 -0
- data/test/migrations/decimal/1_give_me_big_numbers.rb +6 -4
- data/test/migrations/magic/1_currencies_have_symbols.rb +3 -2
- data/test/migrations/missing/1000_people_have_middle_names.rb +2 -0
- data/test/migrations/missing/1_people_have_last_names.rb +2 -0
- data/test/migrations/missing/3_we_need_reminders.rb +2 -0
- data/test/migrations/missing/4_innocent_jointable.rb +3 -1
- data/test/migrations/rename/1_we_need_things.rb +2 -0
- data/test/migrations/rename/2_rename_things.rb +2 -0
- data/test/migrations/to_copy/1_people_have_hobbies.rb +3 -1
- data/test/migrations/to_copy/2_people_have_descriptions.rb +3 -1
- data/test/migrations/to_copy2/1_create_articles.rb +2 -0
- data/test/migrations/to_copy2/2_create_comments.rb +3 -1
- data/test/migrations/to_copy_with_name_collision/1_people_have_hobbies.rb +3 -1
- data/test/migrations/to_copy_with_timestamps/20090101010101_people_have_hobbies.rb +3 -1
- data/test/migrations/to_copy_with_timestamps/20090101010202_people_have_descriptions.rb +3 -1
- data/test/migrations/to_copy_with_timestamps2/20090101010101_create_articles.rb +2 -0
- data/test/migrations/to_copy_with_timestamps2/20090101010202_create_comments.rb +2 -0
- data/test/migrations/valid/1_valid_people_have_last_names.rb +2 -0
- data/test/migrations/valid/2_we_need_reminders.rb +2 -0
- data/test/migrations/valid/3_innocent_jointable.rb +3 -1
- data/test/migrations/valid_with_subdirectories/1_valid_people_have_last_names.rb +2 -0
- data/test/migrations/valid_with_subdirectories/sub/2_we_need_reminders.rb +2 -0
- data/test/migrations/valid_with_subdirectories/sub1/3_innocent_jointable.rb +3 -1
- data/test/migrations/valid_with_timestamps/20100101010101_valid_with_timestamps_people_have_last_names.rb +2 -0
- data/test/migrations/valid_with_timestamps/20100201010101_valid_with_timestamps_we_need_reminders.rb +2 -0
- data/test/migrations/valid_with_timestamps/20100301010101_valid_with_timestamps_innocent_jointable.rb +3 -1
- data/test/migrations/version_check/20131219224947_migration_version_check.rb +2 -0
- data/test/models/account.rb +46 -0
- data/test/models/admin/account.rb +3 -1
- data/test/models/admin/randomly_named_c1.rb +2 -0
- data/test/models/admin/user.rb +16 -8
- data/test/models/admin.rb +4 -2
- data/test/models/aircraft.rb +3 -1
- data/test/models/arunit2_model.rb +2 -0
- data/test/models/author.rb +153 -102
- data/test/models/auto_id.rb +2 -0
- data/test/models/autoloadable/extra_firm.rb +2 -0
- data/test/models/binary.rb +3 -1
- data/test/models/binary_field.rb +6 -0
- data/test/models/bird.rb +13 -1
- data/test/models/book.rb +14 -4
- data/test/models/book_destroy_async.rb +24 -0
- data/test/models/boolean.rb +5 -0
- data/test/models/bulb.rb +13 -4
- data/test/models/cake_designer.rb +2 -0
- data/test/models/car.rb +17 -10
- data/test/models/carrier.rb +2 -0
- data/test/models/cart.rb +5 -0
- data/test/models/cat.rb +2 -0
- data/test/models/categorization.rb +8 -6
- data/test/models/category.rb +28 -16
- data/test/models/chef.rb +2 -0
- data/test/models/citation.rb +5 -1
- data/test/models/club.rb +13 -10
- data/test/models/college.rb +4 -2
- data/test/models/column.rb +2 -0
- data/test/models/column_name.rb +2 -0
- data/test/models/comment.rb +32 -10
- data/test/models/company.rb +102 -106
- data/test/models/company_in_module.rb +27 -26
- data/test/models/computer.rb +3 -1
- data/test/models/contact.rb +15 -13
- data/test/models/content.rb +5 -3
- data/test/models/contract.rb +21 -3
- data/test/models/country.rb +2 -4
- data/test/models/course.rb +3 -1
- data/test/models/customer.rb +10 -8
- data/test/models/customer_carrier.rb +2 -0
- data/test/models/dashboard.rb +2 -0
- data/test/models/default.rb +2 -0
- data/test/models/department.rb +2 -0
- data/test/models/destroy_async_parent.rb +15 -0
- data/test/models/destroy_async_parent_soft_delete.rb +20 -0
- data/test/models/developer.rb +152 -85
- data/test/models/dl_keyed_belongs_to.rb +13 -0
- data/test/models/dl_keyed_belongs_to_soft_delete.rb +19 -0
- data/test/models/dl_keyed_has_many.rb +5 -0
- data/test/models/dl_keyed_has_many_through.rb +5 -0
- data/test/models/dl_keyed_has_one.rb +5 -0
- data/test/models/dl_keyed_join.rb +10 -0
- data/test/models/dog.rb +2 -0
- data/test/models/dog_lover.rb +2 -0
- data/test/models/doubloon.rb +3 -1
- data/test/models/drink_designer.rb +17 -0
- data/test/models/edge.rb +4 -2
- data/test/models/electron.rb +2 -0
- data/test/models/engine.rb +3 -2
- data/test/models/entrant.rb +2 -0
- data/test/models/entry.rb +5 -0
- data/test/models/essay.rb +6 -3
- data/test/models/essay_destroy_async.rb +12 -0
- data/test/models/event.rb +3 -1
- data/test/models/eye.rb +5 -3
- data/test/models/face.rb +14 -6
- data/test/models/family.rb +6 -0
- data/test/models/family_tree.rb +6 -0
- data/test/models/friendship.rb +5 -3
- data/test/models/frog.rb +8 -0
- data/test/models/guid.rb +3 -1
- data/test/models/guitar.rb +2 -0
- data/test/models/hotel.rb +5 -3
- data/test/models/human.rb +39 -0
- data/test/models/image.rb +3 -1
- data/test/models/interest.rb +14 -3
- data/test/models/invoice.rb +4 -2
- data/test/models/item.rb +3 -1
- data/test/models/job.rb +5 -3
- data/test/models/joke.rb +4 -2
- data/test/models/keyboard.rb +3 -1
- data/test/models/legacy_thing.rb +2 -0
- data/test/models/lesson.rb +2 -0
- data/test/models/line_item.rb +3 -1
- data/test/models/liquid.rb +2 -0
- data/test/models/matey.rb +3 -1
- data/test/models/measurement.rb +4 -0
- data/test/models/member.rb +23 -20
- data/test/models/member_detail.rb +3 -0
- data/test/models/member_type.rb +2 -0
- data/test/models/membership.rb +4 -1
- data/test/models/mentor.rb +3 -1
- data/test/models/message.rb +5 -0
- data/test/models/minimalistic.rb +2 -0
- data/test/models/minivan.rb +3 -2
- data/test/models/mixed_case_monkey.rb +3 -1
- data/test/models/molecule.rb +2 -0
- data/test/models/mouse.rb +6 -0
- data/test/models/movie.rb +2 -0
- data/test/models/node.rb +4 -2
- data/test/models/non_primary_key.rb +2 -0
- data/test/models/notification.rb +2 -0
- data/test/models/numeric_data.rb +12 -0
- data/test/models/order.rb +4 -2
- data/test/models/organization.rb +9 -7
- data/test/models/other_dog.rb +3 -1
- data/test/models/owner.rb +6 -4
- data/test/models/parrot.rb +12 -4
- data/test/models/person.rb +59 -54
- data/test/models/personal_legacy_thing.rb +3 -1
- data/test/models/pet.rb +4 -2
- data/test/models/pet_treasure.rb +2 -0
- data/test/models/pirate.rb +67 -43
- data/test/models/possession.rb +3 -1
- data/test/models/post.rb +184 -86
- data/test/models/price_estimate.rb +11 -1
- data/test/models/professor.rb +3 -1
- data/test/models/project.rb +14 -12
- data/test/models/publisher/article.rb +2 -0
- data/test/models/publisher/magazine.rb +2 -0
- data/test/models/publisher.rb +2 -0
- data/test/models/randomly_named_c1.rb +2 -0
- data/test/models/rating.rb +5 -1
- data/test/models/reader.rb +7 -5
- data/test/models/recipe.rb +2 -0
- data/test/models/record.rb +2 -0
- data/test/models/reference.rb +6 -3
- data/test/models/reply.rb +39 -21
- data/test/models/room.rb +6 -0
- data/test/models/section.rb +6 -0
- data/test/models/seminar.rb +6 -0
- data/test/models/session.rb +6 -0
- data/test/models/ship.rb +12 -9
- data/test/models/ship_part.rb +5 -3
- data/test/models/shop.rb +4 -2
- data/test/models/shop_account.rb +2 -0
- data/test/models/speedometer.rb +2 -0
- data/test/models/sponsor.rb +8 -5
- data/test/models/squeak.rb +6 -0
- data/test/models/strict_zine.rb +7 -0
- data/test/models/string_key_object.rb +2 -0
- data/test/models/student.rb +2 -0
- data/test/models/subscriber.rb +4 -2
- data/test/models/subscription.rb +5 -1
- data/test/models/tag.rb +6 -3
- data/test/models/tagging.rb +13 -6
- data/test/models/task.rb +2 -0
- data/test/models/topic.rb +54 -19
- data/test/models/toy.rb +4 -0
- data/test/models/traffic_light.rb +2 -0
- data/test/models/treasure.rb +5 -3
- data/test/models/treaty.rb +2 -4
- data/test/models/tree.rb +2 -0
- data/test/models/tuning_peg.rb +2 -0
- data/test/models/tyre.rb +2 -0
- data/test/models/user.rb +12 -4
- data/test/models/uuid_child.rb +2 -0
- data/test/models/uuid_item.rb +2 -0
- data/test/models/uuid_parent.rb +2 -0
- data/test/models/vegetables.rb +12 -3
- data/test/models/vertex.rb +6 -4
- data/test/models/warehouse_thing.rb +2 -0
- data/test/models/wheel.rb +3 -1
- data/test/models/without_table.rb +3 -1
- data/test/models/zine.rb +3 -1
- data/test/schema/mysql2_specific_schema.rb +49 -35
- data/test/schema/oracle_specific_schema.rb +13 -15
- data/test/schema/postgresql_specific_schema.rb +51 -40
- data/test/schema/schema.rb +334 -154
- data/test/schema/sqlite_specific_schema.rb +9 -16
- data/test/support/config.rb +26 -26
- data/test/support/connection.rb +14 -8
- data/test/support/connection_helper.rb +3 -1
- data/test/support/ddl_helper.rb +2 -0
- data/test/support/marshal_compatibility_fixtures/IBM_DB/rails_6_0_topic.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/IBM_DB/rails_6_0_topic_associations.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/Mysql2/rails_6_0_topic.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/Mysql2/rails_6_0_topic_associations.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/PostgreSQL/rails_6_0_topic.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/PostgreSQL/rails_6_0_topic_associations.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/SQLite/rails_6_0_topic.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/SQLite/rails_6_0_topic_associations.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/legacy_6_0_record_mysql.dump +0 -0
- data/test/support/marshal_compatibility_fixtures/legacy_relation.dump +0 -0
- data/test/support/schema_dumping_helper.rb +2 -0
- data/test/support/stubs/strong_parameters.rb +40 -0
- data/test/support/yaml_compatibility_fixtures/rails_v1_mysql.yml +206 -0
- data/test/support/yaml_compatibility_fixtures/rails_v2.yml +55 -0
- metadata +190 -14
@@ -0,0 +1,499 @@
|
|
1
|
+
# frozen_string_literal: true
|
2
|
+
|
3
|
+
require "cases/helper"
|
4
|
+
require "models/person"
|
5
|
+
|
6
|
+
module ActiveRecord
|
7
|
+
module ConnectionAdapters
|
8
|
+
class ConnectionHandlersShardingDbTest < ActiveRecord::TestCase
|
9
|
+
self.use_transactional_tests = false
|
10
|
+
|
11
|
+
fixtures :people
|
12
|
+
|
13
|
+
def setup
|
14
|
+
@handler = ConnectionHandler.new
|
15
|
+
@owner_name = "ActiveRecord::Base"
|
16
|
+
db_config = ActiveRecord::Base.configurations.configs_for(env_name: "arunit", name: "primary")
|
17
|
+
@rw_pool = @handler.establish_connection(db_config)
|
18
|
+
@ro_pool = @handler.establish_connection(db_config, role: :reading)
|
19
|
+
end
|
20
|
+
|
21
|
+
def teardown
|
22
|
+
clean_up_connection_handler
|
23
|
+
end
|
24
|
+
|
25
|
+
unless in_memory_db?
|
26
|
+
def test_establishing_a_connection_in_connected_to_block_uses_current_role_and_shard
|
27
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :shard_one) do
|
28
|
+
db_config = ActiveRecord::Base.configurations.configs_for(env_name: "arunit", name: "primary")
|
29
|
+
ActiveRecord::Base.establish_connection(db_config)
|
30
|
+
assert_nothing_raised { Person.first }
|
31
|
+
|
32
|
+
assert_equal [:default, :shard_one], ActiveRecord::Base.connection_handler.send(:owner_to_pool_manager).fetch("ActiveRecord::Base").instance_variable_get(:@name_to_role_mapping).values.flat_map(&:keys).uniq
|
33
|
+
end
|
34
|
+
end
|
35
|
+
|
36
|
+
def test_establish_connection_using_3_levels_config
|
37
|
+
previous_env, ENV["RAILS_ENV"] = ENV["RAILS_ENV"], "default_env"
|
38
|
+
|
39
|
+
config = {
|
40
|
+
"default_env" => {
|
41
|
+
"primary" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3" },
|
42
|
+
"primary_shard_one" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3" },
|
43
|
+
}
|
44
|
+
}
|
45
|
+
|
46
|
+
@prev_configs, ActiveRecord::Base.configurations = ActiveRecord::Base.configurations, config
|
47
|
+
|
48
|
+
ActiveRecord::Base.connects_to(shards: {
|
49
|
+
default: { writing: :primary },
|
50
|
+
shard_one: { writing: :primary_shard_one }
|
51
|
+
})
|
52
|
+
|
53
|
+
base_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base")
|
54
|
+
default_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", shard: :default)
|
55
|
+
|
56
|
+
assert_equal base_pool, default_pool
|
57
|
+
assert_equal "test/db/primary.sqlite3", default_pool.db_config.database
|
58
|
+
assert_equal "primary", default_pool.db_config.name
|
59
|
+
|
60
|
+
assert_not_nil pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", shard: :shard_one)
|
61
|
+
assert_equal "test/db/primary_shard_one.sqlite3", pool.db_config.database
|
62
|
+
assert_equal "primary_shard_one", pool.db_config.name
|
63
|
+
ensure
|
64
|
+
ActiveRecord::Base.configurations = @prev_configs
|
65
|
+
ActiveRecord::Base.establish_connection(:arunit)
|
66
|
+
ENV["RAILS_ENV"] = previous_env
|
67
|
+
end
|
68
|
+
|
69
|
+
def test_establish_connection_using_3_levels_config_with_shards_and_replica
|
70
|
+
previous_env, ENV["RAILS_ENV"] = ENV["RAILS_ENV"], "default_env"
|
71
|
+
|
72
|
+
config = {
|
73
|
+
"default_env" => {
|
74
|
+
"primary" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3" },
|
75
|
+
"primary_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3", "replica" => true },
|
76
|
+
"primary_shard_one" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3" },
|
77
|
+
"primary_shard_one_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3", "replica" => true }
|
78
|
+
}
|
79
|
+
}
|
80
|
+
|
81
|
+
@prev_configs, ActiveRecord::Base.configurations = ActiveRecord::Base.configurations, config
|
82
|
+
|
83
|
+
ActiveRecord::Base.connects_to(shards: {
|
84
|
+
default: { writing: :primary, reading: :primary_replica },
|
85
|
+
shard_one: { writing: :primary_shard_one, reading: :primary_shard_one_replica }
|
86
|
+
})
|
87
|
+
|
88
|
+
default_writing_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", shard: :default)
|
89
|
+
base_writing_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base")
|
90
|
+
assert_equal base_writing_pool, default_writing_pool
|
91
|
+
assert_equal "test/db/primary.sqlite3", default_writing_pool.db_config.database
|
92
|
+
assert_equal "primary", default_writing_pool.db_config.name
|
93
|
+
|
94
|
+
default_reading_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", role: :reading, shard: :default)
|
95
|
+
base_reading_pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", role: :reading)
|
96
|
+
assert_equal base_reading_pool, default_reading_pool
|
97
|
+
assert_equal "test/db/primary.sqlite3", default_reading_pool.db_config.database
|
98
|
+
assert_equal "primary_replica", default_reading_pool.db_config.name
|
99
|
+
|
100
|
+
assert_not_nil pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", shard: :shard_one)
|
101
|
+
assert_equal "test/db/primary_shard_one.sqlite3", pool.db_config.database
|
102
|
+
assert_equal "primary_shard_one", pool.db_config.name
|
103
|
+
|
104
|
+
assert_not_nil pool = ActiveRecord::Base.connection_handler.retrieve_connection_pool("ActiveRecord::Base", role: :reading, shard: :shard_one)
|
105
|
+
assert_equal "test/db/primary_shard_one.sqlite3", pool.db_config.database
|
106
|
+
assert_equal "primary_shard_one_replica", pool.db_config.name
|
107
|
+
ensure
|
108
|
+
ActiveRecord::Base.configurations = @prev_configs
|
109
|
+
ActiveRecord::Base.establish_connection(:arunit)
|
110
|
+
ENV["RAILS_ENV"] = previous_env
|
111
|
+
end
|
112
|
+
|
113
|
+
def test_switching_connections_via_handler
|
114
|
+
previous_env, ENV["RAILS_ENV"] = ENV["RAILS_ENV"], "default_env"
|
115
|
+
|
116
|
+
config = {
|
117
|
+
"default_env" => {
|
118
|
+
"primary" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3" },
|
119
|
+
"primary_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3", "replica" => true },
|
120
|
+
"primary_shard_one" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3" },
|
121
|
+
"primary_shard_one_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3", "replica" => true }
|
122
|
+
}
|
123
|
+
}
|
124
|
+
|
125
|
+
@prev_configs, ActiveRecord::Base.configurations = ActiveRecord::Base.configurations, config
|
126
|
+
|
127
|
+
ActiveRecord::Base.connects_to(shards: {
|
128
|
+
default: { writing: :primary, reading: :primary_replica },
|
129
|
+
shard_one: { writing: :primary_shard_one, reading: :primary_shard_one_replica }
|
130
|
+
})
|
131
|
+
|
132
|
+
ActiveRecord::Base.connected_to(role: :reading, shard: :default) do
|
133
|
+
assert_equal :reading, ActiveRecord::Base.current_role
|
134
|
+
assert ActiveRecord::Base.connected_to?(role: :reading, shard: :default)
|
135
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :default)
|
136
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :shard_one)
|
137
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :shard_one)
|
138
|
+
assert_predicate ActiveRecord::Base.connection, :preventing_writes?
|
139
|
+
end
|
140
|
+
|
141
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :default) do
|
142
|
+
assert_equal :writing, ActiveRecord::Base.current_role
|
143
|
+
assert ActiveRecord::Base.connected_to?(role: :writing, shard: :default)
|
144
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :default)
|
145
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :shard_one)
|
146
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :shard_one)
|
147
|
+
assert_not_predicate ActiveRecord::Base.connection, :preventing_writes?
|
148
|
+
end
|
149
|
+
|
150
|
+
ActiveRecord::Base.connected_to(role: :reading, shard: :shard_one) do
|
151
|
+
assert_equal :reading, ActiveRecord::Base.current_role
|
152
|
+
assert ActiveRecord::Base.connected_to?(role: :reading, shard: :shard_one)
|
153
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :shard_one)
|
154
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :default)
|
155
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :default)
|
156
|
+
assert_predicate ActiveRecord::Base.connection, :preventing_writes?
|
157
|
+
end
|
158
|
+
|
159
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :shard_one) do
|
160
|
+
assert_equal :writing, ActiveRecord::Base.current_role
|
161
|
+
assert ActiveRecord::Base.connected_to?(role: :writing, shard: :shard_one)
|
162
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :shard_one)
|
163
|
+
assert_not ActiveRecord::Base.connected_to?(role: :reading, shard: :default)
|
164
|
+
assert_not ActiveRecord::Base.connected_to?(role: :writing, shard: :default)
|
165
|
+
assert_not_predicate ActiveRecord::Base.connection, :preventing_writes?
|
166
|
+
end
|
167
|
+
ensure
|
168
|
+
ActiveRecord::Base.configurations = @prev_configs
|
169
|
+
ActiveRecord::Base.establish_connection(:arunit)
|
170
|
+
ENV["RAILS_ENV"] = previous_env
|
171
|
+
end
|
172
|
+
|
173
|
+
def test_retrieves_proper_connection_with_nested_connected_to
|
174
|
+
previous_env, ENV["RAILS_ENV"] = ENV["RAILS_ENV"], "default_env"
|
175
|
+
|
176
|
+
config = {
|
177
|
+
"default_env" => {
|
178
|
+
"primary" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3" },
|
179
|
+
"primary_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary.sqlite3", "replica" => true },
|
180
|
+
"primary_shard_one" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3" },
|
181
|
+
"primary_shard_one_replica" => { "adapter" => "sqlite3", "database" => "test/db/primary_shard_one.sqlite3", "replica" => true }
|
182
|
+
}
|
183
|
+
}
|
184
|
+
|
185
|
+
@prev_configs, ActiveRecord::Base.configurations = ActiveRecord::Base.configurations, config
|
186
|
+
|
187
|
+
ActiveRecord::Base.connects_to(shards: {
|
188
|
+
default: { writing: :primary, reading: :primary_replica },
|
189
|
+
shard_one: { writing: :primary_shard_one, reading: :primary_shard_one_replica }
|
190
|
+
})
|
191
|
+
|
192
|
+
ActiveRecord::Base.connected_to(role: :reading, shard: :shard_one) do
|
193
|
+
# Uses the correct connection
|
194
|
+
assert_equal "primary_shard_one_replica", ActiveRecord::Base.connection_pool.db_config.name
|
195
|
+
|
196
|
+
# Uses the shard currently in use
|
197
|
+
ActiveRecord::Base.connected_to(role: :writing) do
|
198
|
+
assert_equal "primary_shard_one", ActiveRecord::Base.connection_pool.db_config.name
|
199
|
+
end
|
200
|
+
|
201
|
+
# Allows overriding the shard as well
|
202
|
+
ActiveRecord::Base.connected_to(role: :reading, shard: :default) do
|
203
|
+
assert_equal "primary_replica", ActiveRecord::Base.connection_pool.db_config.name
|
204
|
+
end
|
205
|
+
|
206
|
+
# Resets correctly
|
207
|
+
assert_equal "primary_shard_one_replica", ActiveRecord::Base.connection_pool.db_config.name
|
208
|
+
end
|
209
|
+
ensure
|
210
|
+
ActiveRecord::Base.configurations = @prev_configs
|
211
|
+
ActiveRecord::Base.establish_connection(:arunit)
|
212
|
+
ENV["RAILS_ENV"] = previous_env
|
213
|
+
end
|
214
|
+
|
215
|
+
class SecondaryBase < ActiveRecord::Base
|
216
|
+
self.abstract_class = true
|
217
|
+
end
|
218
|
+
|
219
|
+
def test_connected_to_raises_without_a_shard_or_role
|
220
|
+
error = assert_raises(ArgumentError) do
|
221
|
+
ActiveRecord::Base.connected_to { }
|
222
|
+
end
|
223
|
+
assert_equal "must provide a `shard` and/or `role`.", error.message
|
224
|
+
end
|
225
|
+
|
226
|
+
def test_connects_to_raises_with_a_shard_and_database_key
|
227
|
+
error = assert_raises(ArgumentError) do
|
228
|
+
ActiveRecord::Base.connects_to(database: { writing: :arunit }, shards: { shard_one: { writing: :arunit } })
|
229
|
+
end
|
230
|
+
assert_equal "`connects_to` can only accept a `database` or `shards` argument, but not both arguments.", error.message
|
231
|
+
end
|
232
|
+
|
233
|
+
def test_retrieve_connection_pool_with_invalid_shard
|
234
|
+
assert_not_nil @handler.retrieve_connection_pool("ActiveRecord::Base")
|
235
|
+
assert_nil @handler.retrieve_connection_pool("ActiveRecord::Base", shard: :foo)
|
236
|
+
|
237
|
+
assert_not_nil @handler.retrieve_connection_pool("ActiveRecord::Base", role: :reading)
|
238
|
+
assert_nil @handler.retrieve_connection_pool("ActiveRecord::Base", role: :reading, shard: :foo)
|
239
|
+
end
|
240
|
+
|
241
|
+
def test_calling_connected_to_on_a_non_existent_shard_raises
|
242
|
+
ActiveRecord::Base.connects_to(shards: {
|
243
|
+
default: { writing: :arunit, reading: :arunit }
|
244
|
+
})
|
245
|
+
|
246
|
+
error = assert_raises ActiveRecord::ConnectionNotEstablished do
|
247
|
+
ActiveRecord::Base.connected_to(role: :reading, shard: :foo) do
|
248
|
+
Person.first
|
249
|
+
end
|
250
|
+
end
|
251
|
+
|
252
|
+
assert_equal "No connection pool for 'ActiveRecord::Base' found for the 'foo' shard.", error.message
|
253
|
+
end
|
254
|
+
end
|
255
|
+
|
256
|
+
class SecondaryBase < ActiveRecord::Base
|
257
|
+
self.abstract_class = true
|
258
|
+
end
|
259
|
+
|
260
|
+
class ShardConnectionTestModel < SecondaryBase
|
261
|
+
end
|
262
|
+
|
263
|
+
class SomeOtherBase < ActiveRecord::Base
|
264
|
+
self.abstract_class = true
|
265
|
+
end
|
266
|
+
|
267
|
+
class ShardConnectionTestModelB < SomeOtherBase
|
268
|
+
end
|
269
|
+
|
270
|
+
def test_same_shards_across_clusters
|
271
|
+
SecondaryBase.connects_to shards: { one: { writing: { database: ":memory:", adapter: "sqlite3" } } }
|
272
|
+
SomeOtherBase.connects_to shards: { one: { writing: { database: ":memory:", adapter: "sqlite3" } } }
|
273
|
+
|
274
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :one) do
|
275
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
276
|
+
ShardConnectionTestModel.create!(shard_key: "test_model_default")
|
277
|
+
|
278
|
+
ShardConnectionTestModelB.connection.execute("CREATE TABLE `shard_connection_test_model_bs` (shard_key VARCHAR (255))")
|
279
|
+
ShardConnectionTestModelB.create!(shard_key: "test_model_b_default")
|
280
|
+
|
281
|
+
assert_equal "test_model_default", ShardConnectionTestModel.where(shard_key: "test_model_default").first.shard_key
|
282
|
+
assert_equal "test_model_b_default", ShardConnectionTestModelB.where(shard_key: "test_model_b_default").first.shard_key
|
283
|
+
end
|
284
|
+
end
|
285
|
+
|
286
|
+
def test_sharding_separation
|
287
|
+
SecondaryBase.connects_to shards: {
|
288
|
+
default: { writing: { database: ":memory:", adapter: "sqlite3" } },
|
289
|
+
one: { writing: { database: ":memory:", adapter: "sqlite3" } }
|
290
|
+
}
|
291
|
+
|
292
|
+
[:default, :one].each do |shard_name|
|
293
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: shard_name) do
|
294
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
295
|
+
end
|
296
|
+
end
|
297
|
+
|
298
|
+
# Create a record on :default
|
299
|
+
ShardConnectionTestModel.create!(shard_key: "foo")
|
300
|
+
|
301
|
+
# Make sure we can read it when explicitly connecting to :default
|
302
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :default) do
|
303
|
+
assert ShardConnectionTestModel.find_by_shard_key("foo")
|
304
|
+
end
|
305
|
+
|
306
|
+
# Switch to shard and make sure we can't read the record from :default
|
307
|
+
# Also add a new record on :one
|
308
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :one) do
|
309
|
+
assert_not ShardConnectionTestModel.find_by_shard_key("foo")
|
310
|
+
ShardConnectionTestModel.create!(shard_key: "bar")
|
311
|
+
end
|
312
|
+
|
313
|
+
# Make sure we can't read the record from :one but can read the record
|
314
|
+
# from :default
|
315
|
+
assert_not ShardConnectionTestModel.find_by_shard_key("bar")
|
316
|
+
assert ShardConnectionTestModel.find_by_shard_key("foo")
|
317
|
+
end
|
318
|
+
|
319
|
+
def test_swapping_shards_globally_in_a_multi_threaded_environment
|
320
|
+
tf_default = Tempfile.open "shard_key_default"
|
321
|
+
tf_shard_one = Tempfile.open "shard_key_one"
|
322
|
+
|
323
|
+
SecondaryBase.connects_to shards: {
|
324
|
+
default: { writing: { database: tf_default.path, adapter: "sqlite3" } },
|
325
|
+
one: { writing: { database: tf_shard_one.path, adapter: "sqlite3" } }
|
326
|
+
}
|
327
|
+
|
328
|
+
[:default, :one].each do |shard_name|
|
329
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: shard_name) do
|
330
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
331
|
+
ShardConnectionTestModel.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}')")
|
332
|
+
end
|
333
|
+
end
|
334
|
+
|
335
|
+
shard_one_latch = Concurrent::CountDownLatch.new
|
336
|
+
shard_default_latch = Concurrent::CountDownLatch.new
|
337
|
+
|
338
|
+
ShardConnectionTestModel.connection
|
339
|
+
|
340
|
+
thread = Thread.new do
|
341
|
+
ShardConnectionTestModel.connection
|
342
|
+
|
343
|
+
shard_default_latch.wait
|
344
|
+
assert_equal "shard_key_default", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
345
|
+
shard_one_latch.count_down
|
346
|
+
end
|
347
|
+
|
348
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: :one) do
|
349
|
+
shard_default_latch.count_down
|
350
|
+
assert_equal "shard_key_one", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
351
|
+
shard_one_latch.wait
|
352
|
+
end
|
353
|
+
|
354
|
+
thread.join
|
355
|
+
ensure
|
356
|
+
tf_shard_one.close
|
357
|
+
tf_shard_one.unlink
|
358
|
+
tf_default.close
|
359
|
+
tf_default.unlink
|
360
|
+
end
|
361
|
+
|
362
|
+
def test_swapping_shards_and_roles_in_a_multi_threaded_environment
|
363
|
+
tf_default = Tempfile.open "shard_key_default"
|
364
|
+
tf_shard_one = Tempfile.open "shard_key_one"
|
365
|
+
tf_default_reading = Tempfile.open "shard_key_default_reading"
|
366
|
+
tf_shard_one_reading = Tempfile.open "shard_key_one_reading"
|
367
|
+
|
368
|
+
SecondaryBase.connects_to shards: {
|
369
|
+
default: { writing: { database: tf_default.path, adapter: "sqlite3" }, secondary: { database: tf_default_reading.path, adapter: "sqlite3" } },
|
370
|
+
one: { writing: { database: tf_shard_one.path, adapter: "sqlite3" }, secondary: { database: tf_shard_one_reading.path, adapter: "sqlite3" } }
|
371
|
+
}
|
372
|
+
|
373
|
+
[:default, :one].each do |shard_name|
|
374
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: shard_name) do
|
375
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
376
|
+
ShardConnectionTestModel.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}')")
|
377
|
+
end
|
378
|
+
|
379
|
+
ActiveRecord::Base.connected_to(role: :secondary, shard: shard_name) do
|
380
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
381
|
+
ShardConnectionTestModel.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}_secondary')")
|
382
|
+
end
|
383
|
+
end
|
384
|
+
|
385
|
+
shard_one_latch = Concurrent::CountDownLatch.new
|
386
|
+
shard_default_latch = Concurrent::CountDownLatch.new
|
387
|
+
|
388
|
+
ShardConnectionTestModel.connection
|
389
|
+
|
390
|
+
thread = Thread.new do
|
391
|
+
ShardConnectionTestModel.connection
|
392
|
+
|
393
|
+
shard_default_latch.wait
|
394
|
+
assert_equal "shard_key_default", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
395
|
+
shard_one_latch.count_down
|
396
|
+
end
|
397
|
+
|
398
|
+
ActiveRecord::Base.connected_to(shard: :one, role: :secondary) do
|
399
|
+
shard_default_latch.count_down
|
400
|
+
assert_equal "shard_key_one_secondary", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
401
|
+
shard_one_latch.wait
|
402
|
+
end
|
403
|
+
|
404
|
+
thread.join
|
405
|
+
ensure
|
406
|
+
tf_shard_one.close
|
407
|
+
tf_shard_one.unlink
|
408
|
+
tf_default.close
|
409
|
+
tf_default.unlink
|
410
|
+
tf_shard_one_reading.close
|
411
|
+
tf_shard_one_reading.unlink
|
412
|
+
tf_default_reading.close
|
413
|
+
tf_default_reading.unlink
|
414
|
+
end
|
415
|
+
|
416
|
+
def test_swapping_granular_shards_and_roles_in_a_multi_threaded_environment
|
417
|
+
tf_default = Tempfile.open "shard_key_default"
|
418
|
+
tf_shard_one = Tempfile.open "shard_key_one"
|
419
|
+
tf_default_reading = Tempfile.open "shard_key_default_reading"
|
420
|
+
tf_shard_one_reading = Tempfile.open "shard_key_one_reading"
|
421
|
+
tf_default2 = Tempfile.open "shard_key_default2"
|
422
|
+
tf_shard_one2 = Tempfile.open "shard_key_one2"
|
423
|
+
tf_default_reading2 = Tempfile.open "shard_key_default_reading2"
|
424
|
+
tf_shard_one_reading2 = Tempfile.open "shard_key_one_reading2"
|
425
|
+
|
426
|
+
SecondaryBase.connects_to shards: {
|
427
|
+
default: { writing: { database: tf_default.path, adapter: "sqlite3" }, secondary: { database: tf_default_reading.path, adapter: "sqlite3" } },
|
428
|
+
one: { writing: { database: tf_shard_one.path, adapter: "sqlite3" }, secondary: { database: tf_shard_one_reading.path, adapter: "sqlite3" } }
|
429
|
+
}
|
430
|
+
|
431
|
+
SomeOtherBase.connects_to shards: {
|
432
|
+
default: { writing: { database: tf_default2.path, adapter: "sqlite3" }, secondary: { database: tf_default_reading2.path, adapter: "sqlite3" } },
|
433
|
+
one: { writing: { database: tf_shard_one2.path, adapter: "sqlite3" }, secondary: { database: tf_shard_one_reading2.path, adapter: "sqlite3" } }
|
434
|
+
}
|
435
|
+
|
436
|
+
[:default, :one].each do |shard_name|
|
437
|
+
ActiveRecord::Base.connected_to(role: :writing, shard: shard_name) do
|
438
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
439
|
+
ShardConnectionTestModel.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}')")
|
440
|
+
ShardConnectionTestModelB.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
441
|
+
ShardConnectionTestModelB.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}_b')")
|
442
|
+
end
|
443
|
+
|
444
|
+
ActiveRecord::Base.connected_to(role: :secondary, shard: shard_name) do
|
445
|
+
ShardConnectionTestModel.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
446
|
+
ShardConnectionTestModel.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}_secondary')")
|
447
|
+
ShardConnectionTestModelB.connection.execute("CREATE TABLE `shard_connection_test_models` (shard_key VARCHAR (255))")
|
448
|
+
ShardConnectionTestModelB.connection.execute("INSERT INTO `shard_connection_test_models` VALUES ('shard_key_#{shard_name}_secondary_b')")
|
449
|
+
end
|
450
|
+
end
|
451
|
+
|
452
|
+
shard_one_latch = Concurrent::CountDownLatch.new
|
453
|
+
shard_default_latch = Concurrent::CountDownLatch.new
|
454
|
+
|
455
|
+
ShardConnectionTestModel.connection
|
456
|
+
|
457
|
+
thread = Thread.new do
|
458
|
+
ShardConnectionTestModel.connection
|
459
|
+
|
460
|
+
shard_default_latch.wait
|
461
|
+
assert_equal "shard_key_default", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
462
|
+
assert_equal "shard_key_default_b", ShardConnectionTestModelB.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
463
|
+
shard_one_latch.count_down
|
464
|
+
end
|
465
|
+
|
466
|
+
SecondaryBase.connected_to(shard: :one, role: :secondary) do
|
467
|
+
shard_default_latch.count_down
|
468
|
+
assert_equal "shard_key_one_secondary", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
469
|
+
assert_equal "shard_key_default_b", ShardConnectionTestModelB.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
470
|
+
|
471
|
+
SomeOtherBase.connected_to(shard: :one, role: :secondary) do
|
472
|
+
assert_equal "shard_key_one_secondary", ShardConnectionTestModel.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
473
|
+
assert_equal "shard_key_one_secondary_b", ShardConnectionTestModelB.connection.select_value("SELECT shard_key from shard_connection_test_models")
|
474
|
+
end
|
475
|
+
shard_one_latch.wait
|
476
|
+
end
|
477
|
+
|
478
|
+
thread.join
|
479
|
+
ensure
|
480
|
+
tf_shard_one.close
|
481
|
+
tf_shard_one.unlink
|
482
|
+
tf_default.close
|
483
|
+
tf_default.unlink
|
484
|
+
tf_shard_one_reading.close
|
485
|
+
tf_shard_one_reading.unlink
|
486
|
+
tf_default_reading.close
|
487
|
+
tf_default_reading.unlink
|
488
|
+
tf_shard_one2.close
|
489
|
+
tf_shard_one2.unlink
|
490
|
+
tf_default2.close
|
491
|
+
tf_default2.unlink
|
492
|
+
tf_shard_one_reading2.close
|
493
|
+
tf_shard_one_reading2.unlink
|
494
|
+
tf_default_reading2.close
|
495
|
+
tf_default_reading2.unlink
|
496
|
+
end
|
497
|
+
end
|
498
|
+
end
|
499
|
+
end
|