diff --git a/test/test_registration.rb b/test/test_registration.rb index 04c66a0e9d716bf1a933f9d01d7a0eb7cefd603b..462267ad1c9cfad3f3930c00eadd640d3784022f 100644 --- a/test/test_registration.rb +++ b/test/test_registration.rb @@ -631,22 +631,23 @@ class RegistrationTest < Minitest::Test RESPONSE - Command::COMMAND_MANAGER.expect( - :write, - EMPromise.reject(:test_result), - [Matching.new do |iq| - assert_equal :form, iq.form.type - assert_equal( - "The JMP number +15555550000 is no longer available.", - iq.form.instructions - ) - end] - ) + result = execute_command do + Command::COMMAND_MANAGER.expect( + :write, + EMPromise.reject(:test_result), + [Matching.new do |iq| + assert_equal :form, iq.form.type + assert_equal( + "The JMP number +15555550000 is no longer available.", + iq.form.instructions + ) + end] + ) - assert_equal( - :test_result, - execute_command { @finish.write.catch { |e| e } } - ) + @finish.write.catch { |e| e } + end + + assert_equal :test_result, result assert_mock Command::COMMAND_MANAGER assert_instance_of( TelSelections::ChooseTel,