diff --git a/spec/unit/persistence/active_record_persistence_multiple_spec.rb b/spec/unit/persistence/active_record_persistence_multiple_spec.rb index dff3658a..f4fd492d 100644 --- a/spec/unit/persistence/active_record_persistence_multiple_spec.rb +++ b/spec/unit/persistence/active_record_persistence_multiple_spec.rb @@ -122,8 +122,12 @@ let(:multiple_with_enum_without_column) {MultipleWithEnumWithoutColumn.new} - it "should raise NoMethodError for transitions" do - expect{multiple_with_enum_without_column.send(:view, :left)}.to raise_error(NoMethodError, /undefined method .status./) + it "should raise an error for transitions" do + if ActiveRecord.gem_version >= Gem::Version.new('7.1.0') + expect{multiple_with_enum_without_column.send(:view, :left)}.to raise_error(RuntimeError, /Unknown enum attribute 'status'/) + else + expect{multiple_with_enum_without_column.send(:view, :left)}.to raise_error(NoMethodError, /undefined method .status./) + end end end diff --git a/spec/unit/persistence/active_record_persistence_spec.rb b/spec/unit/persistence/active_record_persistence_spec.rb index 089e378e..ce0900d5 100644 --- a/spec/unit/persistence/active_record_persistence_spec.rb +++ b/spec/unit/persistence/active_record_persistence_spec.rb @@ -122,8 +122,12 @@ let(:with_enum_without_column) {WithEnumWithoutColumn.new} - it "should raise NoMethodError for transitions" do - expect{with_enum_without_column.send(:view)}.to raise_error(NoMethodError, /undefined method .status./) + it "should raise an error for transitions" do + if ActiveRecord.gem_version >= Gem::Version.new('7.1.0') + expect{with_enum_without_column.send(:view)}.to raise_error(RuntimeError, /Unknown enum attribute 'status'/) + else + expect{with_enum_without_column.send(:view)}.to raise_error(NoMethodError, /undefined method .status./) + end end end