Hide delegation on state names

aasm · Feb 16, 2023 · 47a6265 · 47a6265
1 parent 2e952ff
commit 47a6265
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/aasm/instance_base.rb b/lib/aasm/instance_base.rb
@@ -57,6 +57,10 @@ def states(options={}, *args)
       end
     end
 
+    def state_names
+      self.states.map(&:name)
+    end
+
     def events(options={}, *args)
       state = options[:state] || current_state
       events = @instance.class.aasm(@name).events.select {|e| e.transitions_from_state?(state) }

diff --git a/spec/unit/states_on_one_line_example_spec.rb b/spec/unit/states_on_one_line_example_spec.rb
@@ -13,4 +13,10 @@
       expect(example.aasm(:one_line).states.map(&:name)).to eq [:initial, :first, :second]
     end
   end
+
+  describe 'states_names' do
+    it 'should have all 3 states defined' do
+      expect(example.aasm(:one_line).state_names).to eq [:initial, :first, :second]
+    end
+  end
 end