make then signature more explicit, simplify impl

freerange · Nov 25, 2019 · 4c888da · 4c888da
1 parent d128e6a
commit 4c888da
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/lib/mocha/expectation.rb b/lib/mocha/expectation.rb
@@ -443,11 +443,8 @@ def throws(tag, object = nil)
     #   radio.expects(:select_channel).with('BBC World Service').when(power.is('on'))
     #   radio.expects(:adjust_volume).with(-5).when(power.is('on'))
     #   radio.expects(:switch_off).then(power.is('off'))
-    def then(*parameters)
-      if parameters.length == 1
-        state = parameters.first
-        add_side_effect(ChangeStateSideEffect.new(state))
-      end
+    def then(state = nil)
+      add_side_effect(ChangeStateSideEffect.new(state)) if state
       self
     end