lib/mathy/operations/operation.rb in mathy-0.0.1 vs lib/mathy/operations/operation.rb in mathy-0.0.2

@@ -1,20 +1,26 @@
 module Mathy
   module Operations
     class Operation
       attr_reader :key
 
-      def initialize(verification, key)
-        @verifier = verification
+      def initialize(difficulty, key, verification)
+        @difficulty = difficulty
         @key = key
+        @verifier = verification
       end
 
-      def play_turn(operands)
+      def play_turn
+        operands = @difficulty.next_operands
         @verifier.check_answer("#{operands.join(" #{key} ")} = ", calculate(operands))
       end
 
       def matches?(other_key)
         key == other_key
+      end
+
+      def calculate(operands)
+        operands.inject { |result, x| result.send(key.to_sym, x) }
       end
     end
   end
 end