Make mutator helper methods call 'save!' if object responds to it

lib/enumerate_it/class_methods.rb

@@ -67,6 +67,7 @@ def create_mutator_methods(klass, attribute_name, helpers)
         klass.enumeration.each_pair do |key, values|
           define_method "#{prefix_name}#{key}!" do
             send "#{attribute_name}=", values.first
+            save! if respond_to?(:save!)
           end
         end
       end

spec/enumerate_it_spec.rb

@@ -163,6 +163,15 @@ def initialize(foobar)
       expect(target.foobar).to eq(TestEnumeration::VALUE_3)
     end
 
+    context 'when class responds to save! method' do
+      it 'calls save!' do
+        target = test_class_with_helper.new(TestEnumeration::VALUE_2)
+        allow(target).to receive(:save!)
+        target.value_3!
+        expect(target).to have_received(:save!)
+      end
+    end
+
     context 'with :prefix option' do
       let :test_class_with_prefixed_helper do
         Class.new do
@@ -193,6 +202,15 @@ def initialize(foobar)
         target.foobar_value_3!
         expect(target.foobar).to eq(TestEnumeration::VALUE_3)
       end
+
+      context 'when class responds to save! method' do
+        it 'calls save!' do
+          target = test_class_with_prefixed_helper.new(TestEnumeration::VALUE_2)
+          allow(target).to receive(:save!)
+          target.foobar_value_3!
+          expect(target).to have_received(:save!)
+        end
+      end
     end
 
     context 'with :polymorphic option' do