lib/mutator/transition.rb in mutator-0.0.2 vs lib/mutator/transition.rb in mutator-0.1.0

@@ -1,25 +1,44 @@
 module Mutator
   class Transition
     attr_reader :to, :from, :machine
 
-    def initialize(to:, from:, machine:)
-      @to, @from, @machine = to, from, machine
+    def initialize(opts)
+      require_parameters!(opts)
+      @to, @from, @machine = opts[:to], opts[:from], opts[:machine]
     end
 
+    def call
+      stateholder.state = to if valid?
+    end
+
     def valid?
-      transitions.present?
+      transitions.length > 0
     end
 
     def stateholder
       machine.stateholder
     end
 
+    def ==(other)
+      to == other.to && from == other.from && machine == other.machine
+    end
+
+    def eql?(other)
+      public_send(:==, other)
+    end
+
   protected
 
     def transitions
-      machine.transitions.select do |t|
-        t[:to] == to && t[:from].include?(from)
+      machine.transitions.select do |transition|
+        transition[:to] == to && transition[:from].include?(from)
+      end
+    end
+
+    def require_parameters!(opts)
+      [:to, :from, :machine].each do |attr|
+        fail ArgumentError, "must provide #{attr}" unless opts[attr]
       end
     end
   end
 end