diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2023-12-19 14:01:45 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2023-12-19 14:01:45 +0300 |
commit | 9297025d0b7ddf095eb618dfaaab2ff8f2018d8b (patch) | |
tree | 865198c01d1824a9b098127baa3ab980c9cd2c06 /spec/support/helpers/next_found_instance_of.rb | |
parent | 6372471f43ee03c05a7c1f8b0c6ac6b8a7431dbe (diff) |
Add latest changes from gitlab-org/gitlab@16-7-stable-eev16.7.0-rc42
Diffstat (limited to 'spec/support/helpers/next_found_instance_of.rb')
-rw-r--r-- | spec/support/helpers/next_found_instance_of.rb | 61 |
1 files changed, 0 insertions, 61 deletions
diff --git a/spec/support/helpers/next_found_instance_of.rb b/spec/support/helpers/next_found_instance_of.rb deleted file mode 100644 index f53798c1856..00000000000 --- a/spec/support/helpers/next_found_instance_of.rb +++ /dev/null @@ -1,61 +0,0 @@ -# frozen_string_literal: true - -module NextFoundInstanceOf - ERROR_MESSAGE = 'NextFoundInstanceOf mock helpers can only be used with ActiveRecord targets' - HELPER_METHOD_PATTERN = /(?:allow|expect)_next_found_(?<number>\d+)_instances_of/ - - def method_missing(method_name, ...) - return super unless match_data = method_name.match(HELPER_METHOD_PATTERN) - - helper_method = method_name.to_s.sub("_#{match_data[:number]}", '') - - public_send(helper_method, *args, match_data[:number].to_i, &block) - end - - def expect_next_found_instance_of(klass, &block) - expect_next_found_instances_of(klass, nil, &block) - end - - def expect_next_found_instances_of(klass, number) - check_if_active_record!(klass) - - stub_allocate(expect(klass), klass, number) do |expectation| - yield(expectation) - end - end - - def allow_next_found_instance_of(klass, &block) - allow_next_found_instances_of(klass, nil, &block) - end - - def allow_next_found_instances_of(klass, number) - check_if_active_record!(klass) - - stub_allocate(allow(klass), klass, number) do |allowance| - yield(allowance) - end - end - - private - - def check_if_active_record!(klass) - raise ArgumentError, ERROR_MESSAGE unless klass < ActiveRecord::Base - end - - def stub_allocate(target, klass, number) - stub = receive(:allocate) - stub.exactly(number).times if number - - target.to stub.and_wrap_original do |method| - method.call.tap do |allocation| - # ActiveRecord::Core.allocate returns a frozen object: - # https://github.com/rails/rails/blob/291a3d2ef29a3842d1156ada7526f4ee60dd2b59/activerecord/lib/active_record/core.rb#L620 - # It's unexpected behavior and probably a bug in Rails - # Let's work it around by setting the attributes to default to unfreeze the object for now - allocation.instance_variable_set(:@attributes, klass._default_attributes) - - yield(allocation) - end - end - end -end |