deimos-ruby 2.2.0 → 2.2.1

data/spec/active_record_consumer_spec.rb

@@ -16,7 +16,7 @@ module ActiveRecordConsumerTest
       end
 
       # :nodoc:
-      class Widget < ActiveRecord::Base
+      class Widget < ActiveRecord::Base # rubocop:disable Lint/ConstantDefinitionInBlock
         default_scope -> { where(some_bool: false) }
       end
       Widget.reset_column_information
@@ -75,6 +75,7 @@ module ActiveRecordConsumerTest
                        some_int: nil)
           self.test_id = test_id
           self.some_int = some_int
+          super
         end
 
         def as_json(_opts={})
@@ -107,6 +108,7 @@ module ActiveRecordConsumerTest
                        updated_at: nil,
                        some_datetime_int: nil,
                        timestamp: nil)
+          super
           self.test_id = test_id
           self.some_int = some_int
           self.updated_at = updated_at
@@ -128,19 +130,19 @@ module ActiveRecordConsumerTest
       stub_const('Schemas::MySchemaWithDateTimes', schema_datetime_class)
 
       Karafka::App.routes.redraw do
-        topic "my-topic" do
+        topic 'my-topic' do
           consumer MyConsumer
           schema 'MySchemaWithDateTimes'
           namespace 'com.my-namespace'
           key_config plain: true
         end
-        topic "my-topic2" do
+        topic 'my-topic2' do
           consumer MyConsumerWithKey
           schema 'MySchemaWithDateTimes'
           namespace 'com.my-namespace'
           key_config schema: 'MySchemaId_key'
         end
-        topic "my-topic3" do
+        topic 'my-topic3' do
           consumer MyCustomFetchConsumer
           schema 'MySchema'
           namespace 'com.my-namespace'
@@ -180,7 +182,7 @@ module ActiveRecordConsumerTest
               expect(widget.test_id).to eq('abc')
               expect(widget.some_int).to eq(3)
               expect(widget.some_datetime_int).to eq(Time.zone.now)
-              expect(widget.some_bool).to eq(false)
+              expect(widget.some_bool).to be(false)
               expect(widget.updated_at).to eq(Time.zone.now)
 
               # test unscoped

data/spec/active_record_producer_spec.rb

@@ -6,10 +6,10 @@ describe Deimos::ActiveRecordProducer do
   include_context 'with widget_with_union_types'
 
   prepend_before(:each) do
-    producer_class = Class.new(Deimos::ActiveRecordProducer)
+    producer_class = Class.new(described_class)
     stub_const('MyProducer', producer_class)
 
-    producer_class = Class.new(Deimos::ActiveRecordProducer)
+    producer_class = Class.new(described_class)
     stub_const('MyBooleanProducer', producer_class)
 
     producer_class = Class.new(Deimos::ActiveRecordProducer) do
@@ -104,96 +104,106 @@ describe Deimos::ActiveRecordProducer do
           expect('my-topic').to have_sent(test_id: 'abc', some_int: 3)
         end
 
-        it 'should coerce values for a UnionSchema' do
+        it 'should coerce strings for a UnionSchema' do
           MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
-            test_union_type: %w(hello world)
-          ))
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: %w(hello world)
+                                             ))
 
           expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type: %w(hello world)
-                                                )
+            test_id: 'abc',
+            test_long: 399_999,
+            test_union_type: %w(hello world)
+          )
+        end
 
+        it 'should coerce maps for a UnionSchema' do
           MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: {
+                                                 record1_map: { a: 9999, b: 234 },
+                                                 record1_id: 567
+                                               }
+                                             ))
+
+          expect('my-topic-with-union-type').to have_sent(
+            test_id: 'abc',
+            test_long: 399_999,
             test_union_type: {
-              record1_map:{ a:9999, b:234 },
+              record1_map: { a: 9999, b: 234 },
               record1_id: 567
             }
-          ))
-
-          expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type:{
-                                                    record1_map:{ a:9999, b:234 },
-                                                    record1_id: 567
-                                                  }
-                                                )
+          )
+        end
 
+        it 'should coerce numbers for a UnionSchema' do
           MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
-            test_union_type: 1010101
-          ))
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: 1_010_101
+                                             ))
 
           expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type:1010101
-                                                )
+            test_id: 'abc',
+            test_long: 399_999,
+            test_union_type: 1_010_101
+          )
+        end
 
+        it 'should coerce records with strings for a UnionSchema' do
           MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
-            test_union_type: {
-              record2_id: "hello world"
-            }
-          ))
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: {
+                                                 record2_id: 'hello world'
+                                               }
+                                             ))
 
           expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type: {
-                                                    record2_id: "hello world"
-                                                  }
-                                                )
-
-          MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
+            test_id: 'abc',
+            test_long: 399_999,
             test_union_type: {
-              record3_id:10.1010
+              record2_id: 'hello world'
             }
-          ))
-
-          expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type: {
-                                                    record3_id:10.1010
-                                                  }
-                                                )
+          )
+        end
 
+        it 'should coerce records with floats for a UnionSchema' do
           MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
-            test_id: "abc",
-            test_long: 399999,
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: {
+                                                 record3_id: 10.1010
+                                               }
+                                             ))
+
+          expect('my-topic-with-union-type').to have_sent(
+            test_id: 'abc',
+            test_long: 399_999,
             test_union_type: {
-              record4_id:101010
+              record3_id: 10.1010
             }
-          ))
+          )
+        end
+
+        it 'should coerce records with ints for a UnionSchema' do
+          MyProducerWithUnionType.send_event(WidgetWithUnionType.new(
+                                               test_id: 'abc',
+                                               test_long: 399_999,
+                                               test_union_type: {
+                                                 record4_id: 101_010
+                                               }
+                                             ))
 
           expect('my-topic-with-union-type').to have_sent(
-                                                  test_id: "abc",
-                                                  test_long: 399999,
-                                                  test_union_type: {
-                                                    record4_id:101010
-                                                  }
-                                                )
+            test_id: 'abc',
+            test_long: 399_999,
+            test_union_type: {
+              record4_id: 101_010
+            }
+          )
         end
 
         it 'should coerce values' do
@@ -215,11 +225,11 @@ describe Deimos::ActiveRecordProducer do
           widget = Widget.create!(test_id: 'abc2', some_int: 3)
           MyProducerWithID.send_event({id: widget.id, test_id: 'abc2', some_int: 3})
           expect('my-topic-with-id').to have_sent(
-                                          test_id: 'abc2',
-                                          some_int: 3,
-                                          message_id: 'generated_id',
-                                          timestamp: anything
-                                        )
+            test_id: 'abc2',
+            some_int: 3,
+            message_id: 'generated_id',
+            timestamp: anything
+          )
         end
 
         it 'should post process the batch of records in #send_events' do

data/spec/backends/outbox_spec.rb

@@ -33,7 +33,7 @@ each_db_config(Deimos::Backends::Outbox) do
     described_class.publish(producer_class: MyProducer,
                             messages: [build_message(nil, 'my-topic', 'foo1')])
     expect(Deimos::KafkaMessage.count).to eq(1)
-    expect(Deimos::KafkaMessage.last.message).to eq(nil)
+    expect(Deimos::KafkaMessage.last.message).to be_nil
   end
 
   it 'should add to non-keyed messages' do

data/spec/batch_consumer_spec.rb

@@ -4,6 +4,15 @@
 module ConsumerTest
   describe Deimos::Consumer, 'Batch Consumer' do
     let(:schema) { 'MySchema' }
+    let(:batch) do
+      [
+        { 'test_id' => 'foo', 'some_int' => 123 },
+        { 'test_id' => 'bar', 'some_int' => 456 }
+      ]
+    end
+    let(:invalid_payloads) do
+      batch.push({ 'invalid' => 'key' })
+    end
     let(:use_schema_classes) { false }
     let(:should_reraise_errors) { false }
     let(:key_config) { { field: 'test_id' } }
@@ -14,6 +23,7 @@ module ConsumerTest
         end
       end
     end
+
     before(:each) do
       # :nodoc:
       stub_const('MyBatchConsumer', consumer_class)
@@ -35,19 +45,8 @@ module ConsumerTest
       end
     end
 
-    let(:batch) do
-      [
-        { 'test_id' => 'foo', 'some_int' => 123 },
-        { 'test_id' => 'bar', 'some_int' => 456 }
-      ]
-    end
-
-    let(:invalid_payloads) do
-      batch.concat([{ 'invalid' => 'key' }])
-    end
-
     describe 'consume_batch' do
-      SCHEMA_CLASS_SETTINGS.each do |setting, use_schema_classes|
+      SCHEMA_CLASS_SETTINGS.each_key do |setting|
         context "with Schema Class consumption #{setting}" do
 
           let(:schema_class_batch) do
@@ -114,9 +113,10 @@ module ConsumerTest
 
       context 'with plain keys' do
         let(:key_config) { { plain: true } }
+
         it 'should decode plain keys for all messages in the batch' do
           test_consume_batch('my-topic', batch, keys: [1, 2]) do |received|
-            expect(received.map(&:key)).to eq(["1", "2"])
+            expect(received.map(&:key)).to eq(%w(1 2))
           end
         end
       end
@@ -124,11 +124,6 @@ module ConsumerTest
 
     describe 'timestamps' do
       let(:schema) { 'MySchemaWithDateTimes' }
-      let(:key_config) { { none: true } }
-      before(:each) do
-        allow(Deimos.config.metrics).to receive(:histogram)
-      end
-
       let(:batch_with_time) do
         [
           {
@@ -145,7 +140,6 @@ module ConsumerTest
           }
         ]
       end
-
       let(:invalid_times) do
         [
           {
@@ -168,6 +162,11 @@ module ConsumerTest
           }
         ]
       end
+      let(:key_config) { { none: true } }
+
+      before(:each) do
+        allow(Deimos.config.metrics).to receive(:histogram)
+      end
 
       it 'should consume a batch' do
         # expect(Deimos.config.metrics).
@@ -197,6 +196,7 @@ module ConsumerTest
     describe 'logging' do
       let(:schema) { 'MySchemaWithUniqueId' }
       let(:key_config) { { plain: true } }
+
       before(:each) do
         allow(Deimos.config.metrics).to receive(:histogram)
         set_karafka_config(:payload_log, :keys)
@@ -214,7 +214,7 @@ module ConsumerTest
 
         expect(Deimos::Logging).
           to receive(:log_info).
-          with(hash_including(payload_keys: ["1", "2"]))
+          with(hash_including(payload_keys: %w(1 2)))
 
         test_consume_batch('my-topic', batch_with_message_id, keys: [1, 2])
       end

data/spec/consumer_spec.rb

@@ -8,6 +8,7 @@ module ConsumerTest
   describe Deimos::Consumer, 'Message Consumer' do
     let(:use_schema_classes) { false }
     let(:reraise_errors) { false }
+
     prepend_before(:each) do
       # :nodoc:
       consumer_class = Class.new(described_class) do
@@ -96,12 +97,14 @@ module ConsumerTest
               end
             end
 
-            test_consume_message(MyConsumer, { 'test_id' => 'foo',
+            expect {
+              test_consume_message(MyConsumer, { 'test_id' => 'foo',
                                  'some_int' => 123 }, key: 'a key')
+            }.not_to raise_error
           end
 
           it 'should fail if reraise is false but fatal_error is true' do
-            expect { test_consume_message(MyConsumer, {test_id: 'fatal'}) }.
+            expect { test_consume_message(MyConsumer, { test_id: 'fatal' }) }.
               to raise_error(Avro::SchemaValidator::ValidationError)
           end
 
@@ -114,10 +117,12 @@ module ConsumerTest
           it 'should fail on message with extra fields' do
             allow_any_instance_of(Deimos::SchemaBackends::AvroValidation).
               to receive(:coerce) { |_, m| m.with_indifferent_access }
-            expect { test_consume_message(MyConsumer,
-                                         { 'test_id' => 'foo',
-                                         'some_int' => 123,
-                                         'extra_field' => 'field name' }) }.
+            expect {
+              test_consume_message(MyConsumer,
+                                   { 'test_id' => 'foo',
+                                   'some_int' => 123,
+                                   'extra_field' => 'field name' })
+            }.
               to raise_error(Avro::SchemaValidator::ValidationError)
           end
 
@@ -127,17 +132,22 @@ module ConsumerTest
               test_consume_message(
                 MyConsumer,
                 { 'test_id' => 'foo',
-                  'some_int' => 123 }) { raise 'OH NOES' }
+                  'some_int' => 123 }
+              ) { raise 'OH NOES' }
             }.not_to raise_error
           end
 
           it 'should not fail when consume fails without reraising errors' do
             set_karafka_config(:reraise_errors, false)
-            allow(Deimos::ProducerMiddleware).to receive(:call) { |m| m[:payload] = m[:payload].to_json; m }
+            allow(Deimos::ProducerMiddleware).to receive(:call) do |m|
+              m[:payload] = m[:payload].to_json
+              m
+            end
             expect {
               test_consume_message(
                 MyConsumer,
-                { 'invalid' => 'key' })
+                { 'invalid' => 'key' }
+              )
             }.not_to raise_error
           end
         end
@@ -171,6 +181,7 @@ module ConsumerTest
             end
           end
         end
+
         after(:each) do
           Karafka::App.routes.clear
         end
@@ -227,10 +238,10 @@ module ConsumerTest
           test_consume_message('my_consume_topic',
                                { 'test_id' => 'foo',
                                  'test_float' => 4.0,
-                                 'test_array' => ["1", "2"],
+                                 'test_array' => %w(1 2),
                                  'additional_field' => 'bar',
                                  'some_nested_record' => {
-                                    'some_int' => 1,
+                                   'some_int' => 1,
                                     'some_float' => 10.0,
                                     'some_string' => 'hi mom',
                                     'additional_field' => 'baz'
@@ -240,7 +251,7 @@ module ConsumerTest
             expect(payload['some_nested_record']['some_int']).to eq(1)
             expect(payload.to_h).not_to have_key('additional_field')
             expect(payload.to_h['some_nested_record']).not_to have_key('additional_field')
-          end
+                                 end
 
         end
       end

data/spec/gen/sample/v1/sample_pb.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: sample/v1/sample.proto
 
@@ -6,7 +7,6 @@ require 'google/protobuf'
 
 require 'google/protobuf/timestamp_pb'
 
-
 descriptor_data = "\n\x16sample/v1/sample.proto\x12\tsample.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"7\n\rNestedMessage\x12\x12\n\nnested_str\x18\x01 \x01(\t\x12\x12\n\nnested_num\x18\x02 \x01(\x05\"\xdb\x02\n\rSampleMessage\x12\x0b\n\x03str\x18\x01 \x01(\t\x12\x0b\n\x03num\x18\x02 \x01(\x05\x12\x0f\n\x07str_arr\x18\x03 \x03(\t\x12\x0c\n\x04\x66lag\x18\x04 \x01(\x08\x12-\n\ttimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12*\n\x06nested\x18\x06 \x01(\x0b\x32\x18.sample.v1.NestedMessageH\x00\x12\x13\n\tunion_str\x18\x07 \x01(\tH\x00\x12\x32\n\x10non_union_nested\x18\x08 \x01(\x0b\x32\x18.sample.v1.NestedMessage\x12\x35\n\x07str_map\x18\t \x03(\x0b\x32$.sample.v1.SampleMessage.StrMapEntry\x1a-\n\x0bStrMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x07\n\x05unionb\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
@@ -14,7 +14,7 @@ pool.add_serialized_file(descriptor_data)
 
 module Sample
   module V1
-    NestedMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("sample.v1.NestedMessage").msgclass
-    SampleMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("sample.v1.SampleMessage").msgclass
+    NestedMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('sample.v1.NestedMessage').msgclass
+    SampleMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup('sample.v1.SampleMessage').msgclass
   end
 end

data/spec/generators/active_record_generator_spec.rb

@@ -5,8 +5,8 @@ require 'generators/deimos/active_record_generator'
 RSpec.describe Deimos::Generators::ActiveRecordGenerator do
 
   after(:each) do
-    FileUtils.rm_rf('db') if File.exist?('db')
-    FileUtils.rm_rf('app') if File.exist?('app')
+    FileUtils.rm_rf('db')
+    FileUtils.rm_rf('app')
   end
 
   it 'should generate a migration' do
@@ -39,12 +39,12 @@ RSpec.describe Deimos::Generators::ActiveRecordGenerator do
             # TODO add indexes as necessary
           end
         end
-      
+
         def down
           return unless table_exists?(:generated_table)
           drop_table :generated_table
         end
-      
+
       end
     MIGRATION
     expect(File.read(files[0])).to eq(results)

data/spec/generators/schema_class/my_schema_with_circular_reference_spec.rb

@@ -90,7 +90,8 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithCircularReference do
     end
 
     it 'should return a JSON string of the payload' do
-      s = '{"properties":{"a_boolean":{"property":true},"an_integer":{"property":1},"a_float":{"property":4.5},"a_string":{"property":"string"},"an_array":{"property":[1,2,3]},"an_hash":{"property":{"a_key":"a_value"}}}}'
+      s = '{"properties":{"a_boolean":{"property":true},"an_integer":{"property":1},"a_float":{"property":4.5},"a_str' \
+          'ing":{"property":"string"},"an_array":{"property":[1,2,3]},"an_hash":{"property":{"a_key":"a_value"}}}}'
       expect(klass.to_json).to eq(s)
     end
   end

data/spec/generators/schema_class/my_schema_with_complex_types_spec.rb

@@ -109,7 +109,14 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithComplexType do
     end
 
     it 'should return a JSON string of the payload' do
-      s = '{"test_id":"test id","union_string":"","test_float":1.2,"test_string_array":["abc","def"],"test_int_array":[123,456],"test_optional_int":123,"some_integer_map":{"int_1":1,"int_2":2},"some_record":{"a_record_field":"field 1"},"some_optional_record":{"a_record_field":"field 2"},"some_record_array":[{"a_record_field":"field 3"},{"a_record_field":"field 4"}],"some_record_map":{"record_1":{"a_record_field":"field 5"},"record_2":{"a_record_field":"field 6"}},"some_enum_array":["sym1","sym2"],"some_optional_enum":null,"some_enum_with_default":"sym6"}'
+      s = '{"test_id":"test ' \
+          'id","union_string":"","test_float":1.2,"test_string_array":["abc","def"],"test_int_array":[123,456],"test_' \
+          'optional_i' \
+          'nt":123,"some_integer_map":{"int_1":1,"int_2":2},"some_record":{"a_record_field":"field ' \
+          '1"},"some_optional_record":{"a_record_field":"field 2"},"some_record_array":[{"a_record_field":"field ' \
+          '3"},{"a_record_field":"field 4"}],"some_record_map":{"record_1":{"a_record_field":"field ' \
+          '5"},"record_2":{"a_record_field":"field ' \
+          '6"}},"some_enum_array":["sym1","sym2"],"some_optional_enum":null,"some_enum_with_default":"sym6"}'
       expect(klass.to_json).to eq(s)
     end
   end
@@ -129,7 +136,7 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithComplexType do
     end
 
     it 'should set some_record to nil' do
-      klass = described_class.new(**payload_hash.merge(some_record: nil))
+      klass = described_class.new(**payload_hash, some_record: nil)
       expect(klass.some_record).to be_nil
     end
   end

data/spec/generators/schema_class_generator_spec.rb

@@ -4,15 +4,15 @@ require 'generators/deimos/schema_class_generator'
 require 'fileutils'
 
 class MultiFileSerializer
-  def process_string(s)
+  def process_string(str)
     # Ruby 3.4 changes how hashes are printed
     if Gem::Version.new(RUBY_VERSION) > Gem::Version.new('3.4.0')
-      s.gsub(/{"(.*)" => /, '{"\1"=>')
+      str.gsub(/{"(.*)" => /, '{"\1"=>')
     else
-      s
+      str
     end
-
   end
+
   def dump(value)
     value.keys.sort.map { |k| "#{k}:\n#{process_string(value[k])}\n" }.join("\n")
   end
@@ -32,7 +32,7 @@ RSpec.describe Deimos::Generators::SchemaClassGenerator do
   end
 
   after(:each) do
-    FileUtils.rm_rf('spec/app') if File.exist?('spec/app')
+    FileUtils.rm_rf('spec/app')
   end
 
   context 'with a Consumers Schema' do

data/spec/kafka_source_spec.rb

@@ -5,6 +5,7 @@ require 'activerecord-import'
 # Wrap in a module so our classes don't leak out afterwards
 module KafkaSourceSpec
   RSpec.describe Deimos::KafkaSource do
+    # rubocop:disable Lint/ConstantDefinitionInBlock
     before(:all) do
       ActiveRecord::Base.connection.create_table(:widgets, force: true) do |t|
         t.integer(:widget_id)
@@ -40,8 +41,8 @@ module KafkaSourceSpec
         end
       end
       Widget.reset_column_information
-
     end
+    # rubocop:enable Lint/ConstantDefinitionInBlock
 
     after(:all) do
       ActiveRecord::Base.connection.drop_table(:widgets)
@@ -111,22 +112,24 @@ module KafkaSourceSpec
       expect('my-topic-the-second').to have_sent(nil, widget.id)
     end
 
-    context 'multi-producer model using two different key fields' do
+    describe 'multi-producer model using two different key fields' do
       before(:each) do
-        class MultiKeyWidget < ActiveRecord::Base
+        klass = Class.new(ActiveRecord::Base) do
           include Deimos::KafkaSource
+
           self.table_name = 'widgets'
 
           def self.kafka_producers
             [WidgetProducer, WidgetStringKeyProducer]
           end
         end
+        stub_const('MultiKeyWidget', klass)
 
-        class WidgetStringKeyProducer < Deimos::ActiveRecordProducer
+        klass = Class.new(Deimos::ActiveRecordProducer) do
           class << self
 
             def generate_payload(attributes, record)
-              payload = super(attributes, record)
+              payload = super
               payload.merge('id' => record.model_id)
             end
 
@@ -135,6 +138,7 @@ module KafkaSourceSpec
             end
           end
         end
+        stub_const('WidgetStringKeyProducer', klass)
 
         Karafka::App.routes.redraw do
           topic 'my-topic' do
@@ -365,8 +369,9 @@ module KafkaSourceSpec
       before(:each) do
         # Dummy class we can include the mixin in. Has a backing table created
         # earlier and has the import hook disabled
-        class WidgetNoImportHook < ActiveRecord::Base
+        klass = Class.new(ActiveRecord::Base) do
           include Deimos::KafkaSource
+
           self.table_name = 'widgets'
 
           # :nodoc:
@@ -385,6 +390,7 @@ module KafkaSourceSpec
           end
         end
 
+        stub_const('WidgetNoImportHook', klass)
         WidgetNoImportHook.reset_column_information
       end
 
@@ -451,6 +457,7 @@ module KafkaSourceSpec
         # earlier and has the import hook disabled
         buggy_class = Class.new(ActiveRecord::Base) do
           include Deimos::KafkaSource
+
           self.table_name = 'widgets'
 
           # :nodoc: