deimos-ruby 1.20.1 → 1.22

data/spec/active_record_batch_consumer_mysql_spec.rb

@@ -1,244 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecordBatchConsumerTest
-  describe Deimos::ActiveRecordConsumer,
-           'Batch Consumer with MySQL handling associations',
-           :integration,
-           db_config: DbConfigs::DB_OPTIONS.second do
-    include_context('with DB')
-
-    before(:all) do
-      ActiveRecord::Base.connection.create_table(:widgets, force: true) do |t|
-        t.string(:test_id)
-        t.string(:part_one)
-        t.string(:part_two)
-        t.integer(:some_int)
-        t.boolean(:deleted, default: false)
-        t.timestamps
-
-        t.index(%i(part_one part_two), unique: true)
-      end
-
-      # create one-to-one association -- Details
-      ActiveRecord::Base.connection.create_table(:details, force: true) do |t|
-        t.string(:title)
-        t.belongs_to(:widget)
-
-        t.index(%i(title), unique: true)
-      end
-
-      # Create one-to-many association Locales
-      ActiveRecord::Base.connection.create_table(:locales, force: true) do |t|
-        t.string(:title)
-        t.string(:language)
-        t.belongs_to(:widget)
-
-        t.index(%i(title language), unique: true)
-      end
-
-      class Detail < ActiveRecord::Base
-        validates :title, presence: true
-      end
-
-      class Locale < ActiveRecord::Base
-        validates :title, presence: true
-        validates :language, presence: true
-      end
-
-      # Sample model
-      class Widget < ActiveRecord::Base
-        has_one :detail
-        has_many :locales
-        validates :test_id, presence: true
-
-        default_scope -> { where(deleted: false) }
-      end
-
-      Widget.reset_column_information
-      Detail.reset_column_information
-      Locale.reset_column_information
-    end
-
-    after(:all) do
-      ActiveRecord::Base.connection.drop_table(:widgets)
-      ActiveRecord::Base.connection.drop_table(:details)
-      ActiveRecord::Base.connection.drop_table(:locales)
-    end
-
-    before(:each) do
-      ActiveRecord::Base.connection.truncate_tables(%i(widgets details locales))
-    end
-
-    prepend_before(:each) do
-      stub_const('MyBatchConsumer', consumer_class)
-    end
-
-    # Helper to publish a list of messages and call the consumer
-    def publish_batch(messages)
-      keys = messages.map { |m| m[:key] }
-      payloads = messages.map { |m| m[:payload] }
-
-      test_consume_batch(MyBatchConsumer, payloads, keys: keys, call_original: true)
-    end
-
-    context 'when association_list configured in consumer without model changes' do
-      let(:consumer_class) do
-        Class.new(described_class) do
-          schema 'MySchema'
-          namespace 'com.my-namespace'
-          key_config plain: true
-          record_class Widget
-          association_list :detail
-
-          def build_records(messages)
-            messages.map do |m|
-              payload = m.payload
-              w = Widget.new(test_id: payload['test_id'], some_int: payload['some_int'])
-              d = Detail.new(title: payload['title'])
-              w.detail = d
-              w
-            end
-          end
-        end
-      end
-
-      it 'should raise error when bulk_import_id is not found' do
-        stub_const('MyBatchConsumer', consumer_class)
-        expect {
-          publish_batch([{ key: 2,
-                           payload: { test_id: 'xyz', some_int: 5, title: 'Widget Title' } }])
-        }.to raise_error('Create bulk_import_id on ActiveRecordBatchConsumerTest::Widget'\
-        ' and set it in `build_records` for associations. Run rails g deimos:bulk_import_id {table}'\
-        ' to create the migration.')
-      end
-    end
-
-    context 'with one-to-one relation in association_list and custom bulk_import_id' do
-      let(:consumer_class) do
-        Class.new(described_class) do
-          schema 'MySchema'
-          namespace 'com.my-namespace'
-          key_config plain: true
-          record_class Widget
-          association_list :detail
-          bulk_import_id_column :custom_id
-
-          def build_records(messages)
-            messages.map do |m|
-              payload = m.payload
-              w = Widget.new(test_id: payload['test_id'],
-                             some_int: payload['some_int'],
-                             custom_id: SecureRandom.uuid)
-              d = Detail.new(title: payload['title'])
-              w.detail = d
-              w
-            end
-          end
-
-          def key_columns(messages, klass)
-            case klass.to_s
-            when Widget.to_s
-              super
-            when Detail.to_s
-              %w(title widget_id)
-            else
-              []
-            end
-          end
-
-          def columns(record_class)
-            all_cols = record_class.columns.map(&:name)
-
-            case record_class.to_s
-            when Widget.to_s
-              super
-            when Detail.to_s
-              all_cols - ['id']
-            else
-              []
-            end
-          end
-        end
-      end
-
-      before(:all) do
-        ActiveRecord::Base.connection.add_column(:widgets, :custom_id, :string, if_not_exists: true)
-        Widget.reset_column_information
-      end
-
-      it 'should save item to widget and associated detail' do
-        stub_const('MyBatchConsumer', consumer_class)
-        publish_batch([{ key: 2,
-                         payload: { test_id: 'xyz', some_int: 5, title: 'Widget Title' } }])
-        expect(Widget.count).to eq(1)
-        expect(Detail.count).to eq(1)
-        expect(Widget.first.id).to eq(Detail.first.widget_id)
-      end
-    end
-
-    context 'with one-to-many relationship in association_list and default bulk_import_id' do
-      let(:consumer_class) do
-        Class.new(described_class) do
-          schema 'MySchema'
-          namespace 'com.my-namespace'
-          key_config plain: true
-          record_class Widget
-          association_list :locales
-
-          def build_records(messages)
-            messages.map do |m|
-              payload = m.payload
-              w = Widget.new(test_id: payload['test_id'],
-                             some_int: payload['some_int'],
-                             bulk_import_id: SecureRandom.uuid)
-              w.locales << Locale.new(title: payload['title'], language: 'en')
-              w.locales << Locale.new(title: payload['title'], language: 'fr')
-              w
-            end
-          end
-
-          def key_columns(messages, klass)
-            case klass.to_s
-            when Widget.to_s
-              super
-            when Detail.to_s
-              %w(title widget_id)
-            when Locale.to_s
-              %w(title language)
-            else
-              []
-            end
-          end
-
-          def columns(record_class)
-            all_cols = record_class.columns.map(&:name)
-
-            case record_class.to_s
-            when Widget.to_s
-              super
-            when Detail.to_s, Locale.to_s
-              all_cols - ['id']
-            else
-              []
-            end
-          end
-        end
-      end
-
-      before(:all) do
-        ActiveRecord::Base.connection.add_column(:widgets, :bulk_import_id, :string, if_not_exists: true)
-        Widget.reset_column_information
-      end
-
-      it 'should save item to widget and associated details' do
-        stub_const('MyBatchConsumer', consumer_class)
-        publish_batch([{ key: 2,
-                         payload: { test_id: 'xyz', some_int: 5, title: 'Widget Title' } }])
-        expect(Widget.count).to eq(1)
-        expect(Locale.count).to eq(2)
-        expect(Widget.first.id).to eq(Locale.first.widget_id)
-        expect(Widget.first.id).to eq(Locale.second.widget_id)
-      end
-    end
-           end
-end