diff --git a/lib/sequencer/unit/import/common/model/reset_primary_key_sequence.rb b/lib/sequencer/unit/import/common/model/reset_primary_key_sequence.rb index b5e3eaee6..7eb07d5df 100644 --- a/lib/sequencer/unit/import/common/model/reset_primary_key_sequence.rb +++ b/lib/sequencer/unit/import/common/model/reset_primary_key_sequence.rb @@ -4,6 +4,7 @@ class Sequencer module Common module Model class ResetPrimaryKeySequence < Sequencer::Unit::Base + extend Forwardable uses :model_class diff --git a/spec/sequencer/unit/import/common/model/reset_primary_key_sequence_spec.rb b/spec/sequencer/unit/import/common/model/reset_primary_key_sequence_spec.rb new file mode 100644 index 000000000..9305c1490 --- /dev/null +++ b/spec/sequencer/unit/import/common/model/reset_primary_key_sequence_spec.rb @@ -0,0 +1,15 @@ +require 'rails_helper' + +RSpec.describe Sequencer::Unit::Import::Common::Model::ResetPrimaryKeySequence, sequencer: :unit do + + it 'calls DbHelper.import_post for given model_class' do + + model_class = ::User + + expect(DbHelper).to receive(:import_post).with(model_class.table_name) + + provided = process( + model_class: model_class, + ) + end +end