karafka-testing 2.6.1 → 2.6.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8273ddabdc2941d14ccc5e64957e3a6413523641edbffffb17b224cb1b7fb23a
-  data.tar.gz: 23509825159602b0ac1ca9cc05810fdcc9183ae96fe30f02d8e51ba6cda878ad
+  metadata.gz: 2be14092c2b929e48d6849d330f913b9000006217a48a2ac0d9cbadc1a43fe77
+  data.tar.gz: c575f2aee39ad9c80ac92c32729572cc882d6951ec5adbb056fc690ca330cd1f
 SHA512:
-  metadata.gz: 1474258c50223d692343c89666446546e958023afd86b7d4b8acf8996f701be27345e309fa631a0a1cfbb1aac9fb9c8901811c6f700ae6fe643eb811629b1646
-  data.tar.gz: 8e741e257da85b1cc06993fd3920a9ebadaa7276ad89be449a976ace801d70957212363c3b9b2e361648951cfe8420795a629fee3a2e6a8bd8763fae36b19a7c
+  metadata.gz: 384fc90788d47afbfa37d7fddb81a229c89fdfc5157b433f0cf5124bbba0309d9cb099c6fb63d3678eba786334352cad8cd257b09227f37b084baf54d88dd7ad
+  data.tar.gz: 5a11804abcad1b3987e5d4c77a77c12cc9d39568cf3c833dcc0286300af64e077dd728bfda5b2cba714445623d9ee93b9ab4d22ba2d84ddb8b10c1a84e05f48c

data/.github/workflows/ci.yml

@@ -24,7 +24,6 @@ jobs:
           - '4.0'
           - '3.4'
           - '3.3'
-          - '3.2'
         include:
           - ruby: '4.0'
             coverage: 'true'
@@ -37,7 +36,7 @@ jobs:
         run: "[ -e $APT_DEPS ] || sudo apt-get install -y --no-install-recommends $APT_DEPS"
 
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
           ruby-version: ${{matrix.ruby}}
           bundler: 'latest'
@@ -67,9 +66,9 @@ jobs:
         with:
           fetch-depth: 0
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
-          ruby-version: '4.0.6'
+          ruby-version: '4.0.7'
           bundler-cache: true
       - name: Run rubocop
         run: bundle exec rubocop
@@ -82,9 +81,9 @@ jobs:
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
-          ruby-version: '4.0.6'
+          ruby-version: '4.0.7'
           bundler-cache: true
       - run: bundle exec yard-lint lib/
 
@@ -96,9 +95,9 @@ jobs:
         with:
           fetch-depth: 0
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
-          ruby-version: '4.0.6'
+          ruby-version: '4.0.7'
           bundler: 'latest'
       - name: Install dependencies
         run: bundle install --jobs 4 --retry 3
@@ -113,9 +112,9 @@ jobs:
         with:
           fetch-depth: 0
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
-          ruby-version: '4.0.6'
+          ruby-version: '4.0.7'
           bundler: 'latest'
       - name: Install dependencies
         run: bundle install --jobs 4 --retry 3
@@ -130,7 +129,7 @@ jobs:
         with:
           fetch-depth: 0
       - name: Set up Node.js
-        uses: actions/setup-node@820762786026740c76f36085b0efc47a31fe5020 # v7.0.0
+        uses: actions/setup-node@949feb2413d6458794dcd2491c4babbbce0c15c1 # v7.1.0
         with:
           node-version: '20'
           cache: 'npm'

data/.github/workflows/push.yml

@@ -24,7 +24,7 @@ jobs:
           fetch-depth: 0
 
       - name: Set up Ruby
-        uses: ruby/setup-ruby@95ef2b042f9d7a56d8268cba8559e2842e2ad01b # v1.321.0
+        uses: ruby/setup-ruby@824c0e29fdfc7d689c191c27acc969d2d04b7dd8 # v1.331.0
         with:
           bundler-cache: false
 
@@ -32,4 +32,4 @@ jobs:
         run: |
           bundle install --jobs 4 --retry 3
 
-      - uses: rubygems/release-gem@052cc82692552de3ef2b81fd670e41d13cba8092 # v1.4.0
+      - uses: rubygems/release-gem@7f9650160c1a4e7989fdc9855807bdbd421d8b6b # v1.4.1

data/.rubocop.yml

@@ -12,7 +12,7 @@ inherit_gem:
 
 AllCops:
   NewCops: enable
-  TargetRubyVersion: 3.2
+  TargetRubyVersion: 3.3
 
 # Layout
 Layout/LineLength:

data/.ruby-version

@@ -1 +1 @@
-4.0.6
+4.0.7

data/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Karafka Testing Changelog
 
+## 2.6.3 (2026-10-25)
+- **[EOL]** Drop Ruby 3.2 support; require Ruby 3.3+.
+- [Maintenance] Align internal naming and YARD docs with the Karafka ecosystem conventions.
+- [Maintenance] Drop `ostruct` from the test dependencies.
+- [Fix] Make the Minitest helpers work without Mocha. They no longer fail with `undefined method 'stubs'` when Mocha is not loaded, and still use Mocha when it is.
+- [Fix] Set `metadata.message` on test messages and give `headers: nil` empty `raw_headers`, as a real run does.
+- [Fix] Make `produce(partition: N)` set N as the batch and consumer partition, not only the message one. With mixed partitions the first message decides.
+
+## 2.6.2 (2026-08-13)
+- [Fix] Read the `strategy_selector` and `expansions_selector` settings from their new `consumer_groups` location. Requires Karafka `>= 2.6.0`.
+
 ## 2.6.1 (2026-08-05)
 - [Fix] Fix the dependency constraints.
 

data/Gemfile

@@ -7,7 +7,6 @@ gemspec
 group :test do
   gem "minitest"
   gem "mocha"
-  gem "ostruct"
   gem "rspec"
   gem "simplecov"
   gem "warning"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    karafka-testing (2.6.1)
+    karafka-testing (2.6.3)
       karafka (>= 2.6.0.rc2, < 2.7.0)
       waterdrop (>= 2.10.0)
 
@@ -12,22 +12,22 @@ GEM
     drb (2.2.3)
     ffi (1.17.4)
     ffi (1.17.4-x86_64-linux-gnu)
-    json (2.20.0)
-    karafka (2.6.0.rc2)
-      karafka-core (>= 2.6.0, < 2.7.0)
+    json (3.0.2)
+    karafka (2.6.1)
+      karafka-core (>= 2.6.2, < 2.7.0)
       karafka-rdkafka (>= 0.28.0)
-      waterdrop (>= 2.10.1, < 3.0.0)
+      waterdrop (>= 2.10.2, < 3.0.0)
       zeitwerk (~> 2.3)
-    karafka-core (2.6.2)
+    karafka-core (2.6.4)
       karafka-rdkafka (>= 0.20.0)
       logger (>= 1.6.0)
-    karafka-rdkafka (0.28.0)
+    karafka-rdkafka (0.30.1)
       ffi (~> 1.17.1)
       json (> 2.0)
       logger
       mini_portile2 (~> 2.6)
       rake (> 12)
-    karafka-rdkafka (0.28.0-x86_64-linux-gnu)
+    karafka-rdkafka (0.30.1-x86_64-linux-gnu)
       ffi (~> 1.17.1)
       json (> 2.0)
       logger
@@ -40,7 +40,6 @@ GEM
       prism (~> 1.5)
     mocha (3.1.0)
       ruby2_keywords (>= 0.0.5)
-    ostruct (0.6.3)
     prism (1.9.0)
     rake (13.4.2)
     rspec (3.13.2)
@@ -57,13 +56,13 @@ GEM
       rspec-support (~> 3.13.0)
     rspec-support (3.13.7)
     ruby2_keywords (0.0.5)
-    simplecov (1.0.3)
+    simplecov (1.3.2)
     warning (1.6.0)
-    waterdrop (2.10.2)
+    waterdrop (2.10.5)
       karafka-core (>= 2.5.12, < 3.0.0)
       karafka-rdkafka (>= 0.24.0)
       zeitwerk (~> 2.3)
-    zeitwerk (2.8.2)
+    zeitwerk (2.8.3)
 
 PLATFORMS
   ruby
@@ -73,7 +72,6 @@ DEPENDENCIES
   karafka-testing!
   minitest
   mocha
-  ostruct
   rspec
   simplecov
   warning

data/README.md

@@ -1,10 +1,10 @@
 # Karafka Testing library
 
-[![Build Status](https://github.com/karafka/karafka-testing/workflows/ci/badge.svg)](https://github.com/karafka/karafka-testing/actions?query=workflow%3Aci)
-[![Gem Version](https://badge.fury.io/rb/karafka-testing.svg)](http://badge.fury.io/rb/karafka-testing)
+[![Build Status](https://github.com/karafka/karafka-testing/actions/workflows/ci.yml/badge.svg)](https://github.com/karafka/karafka-testing/actions/workflows/ci.yml)
+[![Gem Version](https://badge.fury.io/rb/karafka-testing.svg)](https://badge.fury.io/rb/karafka-testing)
 [![Join the chat at https://slack.karafka.io](https://raw.githubusercontent.com/karafka/misc/master/slack.svg)](https://slack.karafka.io)
 
-Karafka-Testing is a library that provides RSpec helpers, to make testing of Karafka consumers and producer much easier.
+Karafka-Testing is a library that provides RSpec and Minitest helpers that make testing of Karafka consumers and producers much easier.
 
 ## Installation and usage
 
@@ -14,6 +14,6 @@ Karafka-Testing is a library that provides RSpec helpers, to make testing of Kar
 
 First, thank you for considering contributing to the Karafka ecosystem! It's people like you that make the open source community such a great community!
 
-Each pull request must pass all the RSpec specs, integration tests and meet our quality requirements.
+Each pull request must pass all the tests, including integration tests, and meet our quality requirements.
 
 Fork it, update and wait for the Github Actions results.

data/karafka-testing.gemspec

@@ -22,7 +22,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "karafka", ">= 2.6.0.rc2", "< 2.7.0"
   spec.add_dependency "waterdrop", ">= 2.10.0"
 
-  spec.required_ruby_version = ">= 3.2.0"
+  spec.required_ruby_version = ">= 3.3.0"
 
   spec.metadata = {
     "homepage_uri" => "https://karafka.io",

data/lib/karafka/testing/helpers.rb

@@ -8,19 +8,19 @@ module Karafka
         # Finds all the routing topics matching requested topic within all topics or within
         # provided consumer group based on name
         #
-        # @param requested_topic [String] requested topic name
+        # @param topic_name [String] requested topic name
         # @param requested_consumer_group [String] requested consumer group or nil to look in all
         # @return [Array<Karafka::Routing::Topic>] all matching topics
         #
         # @note Since we run the lookup on subscription groups, the search will automatically
         #   expand with matching patterns
-        def karafka_consumer_find_candidate_topics(requested_topic, requested_consumer_group)
+        def karafka_consumer_find_candidate_topics(topic_name, requested_consumer_group)
           subscription_groups = karafka_consumer_find_subscription_groups(requested_consumer_group)
 
           candidate_topics = []
 
           subscription_groups.each do |group|
-            topic = group.topics.find(requested_topic.to_s)
+            topic = group.topics.find(topic_name.to_s)
 
             next unless topic
 
@@ -36,6 +36,20 @@ module Karafka
           candidate_topics.uniq { |topic| topic.subscription_group.name }
         end
 
+        # Builds a coordinator for given topic partition, as a consumer gets in a real run
+        #
+        # @param topic [Karafka::Routing::Topic] topic of the consumer
+        # @param partition [Integer] partition the consumer works on
+        # @return [Karafka::Processing::ConsumerGroups::Coordinator] coordinator for the partition
+        def karafka_build_coordinator(topic, partition)
+          coordinator = Karafka::Processing::CoordinatorsBuffer
+            .new(Karafka::Routing::Topics.new([topic]))
+            .find_or_create(topic.name, partition)
+
+          coordinator.seek_offset = 0
+          coordinator
+        end
+
         # Finds subscription groups from the requested consumer group or selects all if no
         # consumer group specified
         # @param requested_consumer_group [String] requested consumer group or nil to look in all

data/lib/karafka/testing/minitest/helpers.rb

@@ -39,7 +39,23 @@ module Karafka
 
               @_karafka_consumer_mappings = {}
 
-              Karafka.producer.stubs(:client).returns(@_karafka_producer_client)
+              # Mocha is a dev dependency of many apps without being used in their Karafka tests,
+              # so we stub with it only when its patches are actually available
+              if Object.const_defined?("Mocha", false) && Karafka.producer.respond_to?(:stubs)
+                Karafka.producer.stubs(:client).returns(@_karafka_producer_client)
+              else
+                producer_client = @_karafka_producer_client
+                Karafka.producer.define_singleton_method(:client) { producer_client }
+                @_karafka_producer_client_stubbed = true
+              end
+            end
+
+            # Mocha removes its own stubs, so we only remove the one we defined
+            teardown_flow = lambda do
+              next unless @_karafka_producer_client_stubbed
+
+              Karafka.producer.singleton_class.remove_method(:client)
+              @_karafka_producer_client_stubbed = false
             end
 
             if base.to_s == "Minitest::Spec"
@@ -60,29 +76,44 @@ module Karafka
 
               base.prepend(setup_mod)
             end
+
+            # `after_teardown` runs even when a test overrides `teardown` without calling `super`
+            teardown_mod = Module.new do
+              define_method(:after_teardown) do
+                super()
+              ensure
+                instance_exec(&teardown_flow)
+              end
+            end
+
+            base.prepend(teardown_mod)
           end
         end
 
         # Creates a consumer instance for a given topic
         #
-        # @param requested_topic [String, Symbol] name of the topic for which we want to
-        #   create a consumer instance
+        # @param topic_name [String, Symbol] name of the topic for which we want to create a
+        #   consumer instance
         # @param requested_consumer_group [String, Symbol, nil] optional name of the consumer group
         #   if we have multiple consumer groups listening on the same topic
         # @return [Object] Karafka consumer instance
         # @raise [Karafka::Testing::Errors::TopicNotFoundError] raised when we're unable to find
         #   topic that was requested
+        # @raise [Karafka::Testing::Errors::TopicInManyConsumerGroupsError] raised when the topic is
+        #   in many consumer groups and no consumer group was requested
+        # @raise [Karafka::Testing::Errors::ConsumerGroupNotFoundError] raised when the requested
+        #   consumer group does not exist
         #
         # @example Creates a consumer instance with settings for `my_requested_topic`
         #   consumer = @karafka.consumer_for(:my_requested_topic)
-        def _karafka_consumer_for(requested_topic, requested_consumer_group = nil)
+        def _karafka_consumer_for(topic_name, requested_consumer_group = nil)
           selected_topics = Testing::Helpers.karafka_consumer_find_candidate_topics(
-            requested_topic.to_s,
+            topic_name.to_s,
             requested_consumer_group.to_s
           )
 
-          raise Errors::TopicInManyConsumerGroupsError, requested_topic if selected_topics.size > 1
-          raise Errors::TopicNotFoundError, requested_topic if selected_topics.empty?
+          raise Errors::TopicInManyConsumerGroupsError, topic_name if selected_topics.size > 1
+          raise Errors::TopicNotFoundError, topic_name if selected_topics.empty?
 
           _karafka_build_consumer_for(selected_topics.first)
         end
@@ -96,7 +127,7 @@ module Karafka
         #   @karafka.produce({ 'hello' => 'world' }.to_json)
         #
         # @example Send a json message to consumer and simulate, that it is partition 6
-        #   @karafka.produce({ 'hello' => 'world' }.to_json, 'partition' => 6)
+        #   @karafka.produce({ 'hello' => 'world' }.to_json, partition: 6)
         def _karafka_add_message_to_consumer_if_needed(message)
           consumer_obj = if message[:consumer_group]
             _karafka_find_consumer_for_message(message)
@@ -129,16 +160,33 @@ module Karafka
             metadata[key] = message.fetch(message_key)
           end
 
-          # Add this message to previously produced messages
-          @_karafka_consumer_messages << Karafka::Messages::Message.new(
+          metadata[:raw_headers] ||= {}
+
+          metadata = Karafka::Messages::Metadata.new(metadata)
+          karafka_message = Karafka::Messages::Message.new(
             message[:payload],
-            Karafka::Messages::Metadata.new(metadata)
+            metadata
           )
+          metadata[:message] = karafka_message
+
+          # Add this message to previously produced messages
+          @_karafka_consumer_messages << karafka_message
+          # A real batch never mixes partitions, so the first message of this consumer decides
+          # the partition
+          partition = (@_karafka_batch_partitions ||= {})[consumer_obj] ||= karafka_message.partition
+
+          if consumer_obj.partition != partition
+            consumer_obj.coordinator = Testing::Helpers.karafka_build_coordinator(
+              consumer_obj.topic,
+              partition
+            )
+          end
+
           # Update batch metadata
           batch_metadata = Karafka::Messages::Builders::BatchMetadata.call(
             @_karafka_consumer_messages,
             consumer_obj.topic,
-            0,
+            partition,
             Time.now
           )
 
@@ -159,7 +207,7 @@ module Karafka
           # WaterDrop message attribute (Kafka assigns offsets, not producers)
           @_karafka_next_offset = metadata.delete(:offset)
 
-          topic = if metadata[:topic]
+          topic_name = if metadata[:topic]
             metadata[:topic]
           elsif defined?(@consumer)
             @consumer.topic.name
@@ -169,7 +217,7 @@ module Karafka
           end
           Karafka.producer.produce_sync(
             {
-              topic: topic,
+              topic: topic_name,
               payload: payload
             }.merge(metadata)
           )
@@ -254,21 +302,17 @@ module Karafka
         # @param topic [Karafka::Routing::Topic] topic for which we want to build the consumer
         # @return [Object] karafka consumer
         def _karafka_build_consumer_for(topic)
-          coordinators = Karafka::Processing::CoordinatorsBuffer.new(
-            Karafka::Routing::Topics.new([topic])
-          )
           processing_cfg = Karafka::App.config.internal.processing
           @consumer = topic.consumer.new
           @consumer.producer = Karafka::App.producer
           # Inject appropriate strategy so needed options and components are available
-          strategy = Karafka::App.config.internal.processing.strategy_selector.find(topic)
+          strategy = processing_cfg.consumer_groups.strategy_selector.find(topic)
           @consumer.singleton_class.include(strategy)
           @consumer.client = @_karafka_consumer_client
-          @consumer.coordinator = coordinators.find_or_create(topic.name, 0)
-          @consumer.coordinator.seek_offset = 0
+          @consumer.coordinator = Testing::Helpers.karafka_build_coordinator(topic, 0)
           # Indicate usage as for tests no direct enqueuing happens
           @consumer.instance_variable_set(:@used, true)
-          expansions = processing_cfg.expansions_selector.find(topic)
+          expansions = processing_cfg.consumer_groups.expansions_selector.find(topic)
           expansions.each { |expansion| @consumer.singleton_class.include(expansion) }
           @_karafka_consumer_mappings[topic.id] = @consumer
           @consumer

data/lib/karafka/testing/minitest/proxy.rb

@@ -12,16 +12,19 @@ module Karafka
         end
 
         # Forwards all arguments to `#_karafka_consumer_for`
+        # @return [Karafka::BaseConsumer] consumer instance
         def consumer_for(*)
           @minitest_example._karafka_consumer_for(*)
         end
 
         # Forwards all arguments to `#_karafka_produce`
+        # @return [Rdkafka::Producer::DeliveryReport] delivery report
         def produce(*)
           @minitest_example._karafka_produce(*)
         end
 
         # Forwards all arguments to `#_karafka_produce_to`
+        # @return [Rdkafka::Producer::DeliveryReport] delivery report
         def produce_to(*)
           @minitest_example._karafka_produce_to(*)
         end

data/lib/karafka/testing/rspec/helpers.rb

@@ -66,26 +66,30 @@ module Karafka
 
         # Creates a consumer instance for a given topic
         #
-        # @param requested_topic [String, Symbol] name of the topic for which we want to
-        #   create a consumer instance
+        # @param topic_name [String, Symbol] name of the topic for which we want to create a
+        #   consumer instance
         # @param requested_consumer_group [String, Symbol, nil] optional name of the consumer group
         #   if we have multiple consumer groups listening on the same topic
         # @return [Object] Karafka consumer instance
         # @raise [Karafka::Testing::Errors::TopicNotFoundError] raised when we're unable to find
         #   topic that was requested
+        # @raise [Karafka::Testing::Errors::TopicInManyConsumerGroupsError] raised when the topic is
+        #   in many consumer groups and no consumer group was requested
+        # @raise [Karafka::Testing::Errors::ConsumerGroupNotFoundError] raised when the requested
+        #   consumer group does not exist
         #
         # @example Creates a MyConsumer consumer instance with settings for `my_requested_topic`
         #   RSpec.describe MyConsumer do
         #     subject(:consumer) { karafka.consumer_for(:my_requested_topic) }
         #   end
-        def _karafka_consumer_for(requested_topic, requested_consumer_group = nil)
+        def _karafka_consumer_for(topic_name, requested_consumer_group = nil)
           selected_topics = Testing::Helpers.karafka_consumer_find_candidate_topics(
-            requested_topic.to_s,
+            topic_name.to_s,
             requested_consumer_group.to_s
           )
 
-          raise Errors::TopicInManyConsumerGroupsError, requested_topic if selected_topics.size > 1
-          raise Errors::TopicNotFoundError, requested_topic if selected_topics.empty?
+          raise Errors::TopicInManyConsumerGroupsError, topic_name if selected_topics.size > 1
+          raise Errors::TopicNotFoundError, topic_name if selected_topics.empty?
 
           _karafka_build_consumer_for(selected_topics.first)
         end
@@ -101,12 +105,12 @@ module Karafka
         #
         # @example Send a json message to consumer and simulate, that it is partition 6
         #   before do
-        #     karafka.produce({ 'hello' => 'world' }.to_json, 'partition' => 6)
+        #     karafka.produce({ 'hello' => 'world' }.to_json, partition: 6)
         #   end
         def _karafka_add_message_to_consumer_if_needed(message)
           consumer_obj = if defined?(consumer)
             consumer
-          elsif _karafka_described_class_is_consumer?
+          elsif _karafka_described_class_consumer?
             _karafka_resolve_subject_consumer || _karafka_find_consumer_for_message(message)
           else
             _karafka_find_consumer_for_message(message)
@@ -135,17 +139,34 @@ module Karafka
             metadata[key] = message.fetch(message_key)
           end
 
-          # Add this message to previously produced messages
-          _karafka_consumer_messages << Karafka::Messages::Message.new(
+          metadata[:raw_headers] ||= {}
+
+          metadata = Karafka::Messages::Metadata.new(metadata)
+          karafka_message = Karafka::Messages::Message.new(
             message[:payload],
-            Karafka::Messages::Metadata.new(metadata)
+            metadata
           )
+          metadata[:message] = karafka_message
+
+          # Add this message to previously produced messages
+          _karafka_consumer_messages << karafka_message
+
+          # A real batch never mixes partitions, so the first message of this consumer decides
+          # the partition
+          partition = (@_karafka_batch_partitions ||= {})[consumer_obj] ||= karafka_message.partition
+
+          if consumer_obj.partition != partition
+            consumer_obj.coordinator = Testing::Helpers.karafka_build_coordinator(
+              consumer_obj.topic,
+              partition
+            )
+          end
 
           # Update batch metadata
           batch_metadata = Karafka::Messages::Builders::BatchMetadata.call(
             _karafka_consumer_messages,
             consumer_obj.topic,
-            0,
+            partition,
             Time.now
           )
 
@@ -166,11 +187,11 @@ module Karafka
           # WaterDrop message attribute (Kafka assigns offsets, not producers)
           @_karafka_next_offset = metadata.delete(:offset)
 
-          topic = if metadata[:topic]
+          topic_name = if metadata[:topic]
             metadata[:topic]
           elsif defined?(consumer)
             consumer.topic.name
-          elsif _karafka_described_class_is_consumer? && (sub = _karafka_resolve_subject_consumer)
+          elsif _karafka_described_class_consumer? && (sub = _karafka_resolve_subject_consumer)
             sub.topic.name
           else
             last_consumer = @_karafka_consumer_mappings&.values&.last
@@ -178,7 +199,7 @@ module Karafka
           end
           Karafka.producer.produce_sync(
             {
-              topic: topic,
+              topic: topic_name,
               payload: payload
             }.merge(metadata)
           )
@@ -222,7 +243,7 @@ module Karafka
         # When true, the implicit RSpec subject can be used as a consumer fallback.
         #
         # @return [Boolean] true if described_class inherits from Karafka::BaseConsumer
-        def _karafka_described_class_is_consumer?
+        def _karafka_described_class_consumer?
           respond_to?(:described_class) &&
             described_class.is_a?(Class) &&
             described_class < Karafka::BaseConsumer
@@ -286,22 +307,17 @@ module Karafka
         # @param topic [Karafka::Routing::Topic] topic for which we want to build the consumer
         # @return [Object] karafka consumer
         def _karafka_build_consumer_for(topic)
-          coordinators = Karafka::Processing::CoordinatorsBuffer.new(
-            Karafka::Routing::Topics.new([topic])
-          )
-
           processing_cfg = Karafka::App.config.internal.processing
           consumer = topic.consumer.new
           consumer.producer = Karafka::App.producer
           # Inject appropriate strategy so needed options and components are available
-          strategy = processing_cfg.strategy_selector.find(topic)
+          strategy = processing_cfg.consumer_groups.strategy_selector.find(topic)
           consumer.singleton_class.include(strategy)
           consumer.client = _karafka_consumer_client
-          consumer.coordinator = coordinators.find_or_create(topic.name, 0)
-          consumer.coordinator.seek_offset = 0
+          consumer.coordinator = Testing::Helpers.karafka_build_coordinator(topic, 0)
           # Indicate usage as for tests no direct enqueuing happens
           consumer.instance_variable_set(:@used, true)
-          expansions = processing_cfg.expansions_selector.find(topic)
+          expansions = processing_cfg.consumer_groups.expansions_selector.find(topic)
           expansions.each { |expansion| consumer.singleton_class.include(expansion) }
 
           @_karafka_consumer_mappings[topic.id] = consumer

data/lib/karafka/testing/rspec/proxy.rb

@@ -12,16 +12,19 @@ module Karafka
         end
 
         # Forwards all arguments to `#_karafka_consumer_for`
+        # @return [Karafka::BaseConsumer] consumer instance
         def consumer_for(*)
           @rspec_example._karafka_consumer_for(*)
         end
 
         # Forwards all arguments to `#_karafka_produce`
+        # @return [Rdkafka::Producer::DeliveryReport] delivery report
         def produce(*)
           @rspec_example._karafka_produce(*)
         end
 
         # Forwards all arguments to `#_karafka_produce_to`
+        # @return [Rdkafka::Producer::DeliveryReport] delivery report
         def produce_to(*)
           @rspec_example._karafka_produce_to(*)
         end

data/lib/karafka/testing/spec_producer_client.rb

@@ -4,18 +4,18 @@ module Karafka
   module Testing
     # Spec producer client used to buffer messages that we send out in specs
     class SpecProducerClient < ::WaterDrop::Clients::Buffered
-      # @param rspec [RSpec::Core::ExampleGroup] rspec example we need to hold to trigger actions
-      #   on it that are rspec context aware
-      def initialize(rspec)
+      # @param example [RSpec::Core::ExampleGroup, Minitest::Test] test example we need to hold to
+      #   trigger actions on it that are test context aware
+      def initialize(example)
         super(nil)
-        @rspec = rspec
+        @example = example
       end
 
       # "Produces" message to Kafka. That is, it acknowledges it locally, adds it to the internal
       # buffer and adds it (if needed) into the current consumer messages buffer
       # @param message [Hash] `Karafka.producer.produce_sync` message hash
       def produce(message)
-        @rspec._karafka_add_message_to_consumer_if_needed(message)
+        @example._karafka_add_message_to_consumer_if_needed(message)
 
         super
       end

data/lib/karafka/testing/version.rb

@@ -3,6 +3,6 @@
 module Karafka
   module Testing
     # Current version of gem. It should match Karafka framework version
-    VERSION = "2.6.1"
+    VERSION = "2.6.3"
   end
 end

data/lib/karafka/testing.rb

@@ -8,6 +8,8 @@ module Karafka
       #
       # You should never use karafka-testing primitives when framework is not loaded because
       # testing lib stubs certain pieces of Karafka that need to be initialized.
+      #
+      # @raise [Karafka::Testing::Errors::KarafkaNotLoadedError] raised when Karafka is not loaded
       def ensure_karafka_loaded!
         return if ::Karafka.const_defined?("App", false)
 
@@ -19,6 +21,10 @@ module Karafka
 
       # If you do not initialize Karafka always within your specs, do not include/use this lib
       # in places where Karafka would not be loaded.
+      #
+      # @raise [Karafka::Testing::Errors::KarafkaNotLoadedError] raised when Karafka is not loaded
+      # @raise [Karafka::Testing::Errors::KarafkaNotInitializedError] raised when Karafka is still
+      #   initializing
       def ensure_karafka_initialized!
         ensure_karafka_loaded!
 

data/package-lock.json

@@ -103,9 +103,9 @@
       }
     },
     "node_modules/fastq": {
-      "version": "1.20.1",
-      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.20.1.tgz",
-      "integrity": "sha512-GGToxJ/w1x32s/D2EKND7kTil4n8OVk/9mycTc4VDza13lOvpUZTGX3mFSCtV9ksdGBVzvsyAVLM6mHFThxXxw==",
+      "version": "1.20.3",
+      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.20.3.tgz",
+      "integrity": "sha512-XKv5nnLs6nLF71NgiKJLIZFLkPyIEuOselLG7ujZnGrRfQK8HpvY+WqKhAJUAdLomwVHErVS4LfxFlPq0/FTAw==",
       "dev": true,
       "license": "ISC",
       "dependencies": {
@@ -286,9 +286,9 @@
       }
     },
     "node_modules/smol-toml": {
-      "version": "1.7.0",
-      "resolved": "https://registry.npmjs.org/smol-toml/-/smol-toml-1.7.0.tgz",
-      "integrity": "sha512-aqVvWoyO21L23mb+drl4RmMXbf6N7FdHjAhTRA9ZBL7apWBgfWC16KjrASI+1p9GAroljyMHj6fK67i0UiTNvQ==",
+      "version": "1.9.0",
+      "resolved": "https://registry.npmjs.org/smol-toml/-/smol-toml-1.9.0.tgz",
+      "integrity": "sha512-hpd+HLON7HdZXqYchMM/+LaTTbdK0AU3NngIJ4KVyWbY9bfQqdL9cD+4yf6dUoU2Ap4VsU0JkQi6FxAI1B2mXQ==",
       "dev": true,
       "license": "BSD-3-Clause",
       "engines": {
@@ -312,9 +312,9 @@
       }
     },
     "node_modules/yaml": {
-      "version": "2.9.0",
-      "resolved": "https://registry.npmjs.org/yaml/-/yaml-2.9.0.tgz",
-      "integrity": "sha512-2AvhNX3mb8zd6Zy7INTtSpl1F15HW6Wnqj0srWlkKLcpYl/gMIMJiyuGq2KeI2YFxUPjdlB+3Lc10seMLtL4cA==",
+      "version": "2.9.1",
+      "resolved": "https://registry.npmjs.org/yaml/-/yaml-2.9.1.tgz",
+      "integrity": "sha512-3NxN8+78OdzbT7C/WjGsyfPAtJaN3FNDsWxv7Y7mcDsT/oOmgW8BpyQQFFBnvZE3j9Y2Sdz1ULFLezL7Eb2yFw==",
       "dev": true,
       "license": "ISC",
       "bin": {

data/renovate.json

@@ -1,4 +1,7 @@
 {
+  "bundler": {
+    "managerFilePatterns": ["/(^|/)Gemfile\\.lint$/"]
+  },
   "$schema": "https://docs.renovatebot.com/renovate-schema.json",
   "extends": [
     "config:recommended"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: karafka-testing
 version: !ruby/object:Gem::Version
-  version: 2.6.1
+  version: 2.6.3
 platform: ruby
 authors:
 - Maciej Mensfeld
@@ -104,14 +104,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.2.0
+      version: 3.3.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.16
+rubygems_version: 4.0.20
 specification_version: 4
 summary: Library which provides helpers for easier Karafka consumers tests
 test_files: []