Feng said:
Will I need to change my code or is there a way for me to write my unit
test that automatically enters some inputs?
It's somewhat difficult. This works for me:
my_program.rb:
--------------
def get_input
while true
print "Enter a non-zero integer greater than -5: "
num = gets
num_int = num.to_i
num_float = num.to_f
if num_int != num_float
puts "You entered a float. Try again."
elsif num_int <= -5
puts "You entered a number that was too small."
elsif num_int == 0
puts "You entered 0 or a word. That is not allowed."
else
return num_int
end
end
end
if __FILE__ == $0
result = get_input
puts "You entered #{result}. Thank you."
end
my_program_UnitTest.rb
--------------
require 'my_program'
require 'test/unit'
require 'stringio'
class TestMyProgram < Test::Unit::TestCase
def setup
$stdin = DATA #DATA is like a file containing the stuff after
__END__
trash_can = StringIO.new
$stdout = trash_can
#If you want to see the prompts shown to the user as part of
#the unit test output and/or be able to send output to your
#terminal for debugging, comment out the previous line.
end
def teardown
$stdin = STDIN
$stdout = STDOUT
end
def test_get_input
while not DATA.eof?
num = get_input
num_int = num.to_i
num_float = num.to_f
assert_not_equal(0, num_int)
assert(num_int > -5, "The number was not greater than -5.")
assert_equal(num_int, num_float)
end
end
end
__END__
-6
-4
0
2.5
6 #Must end the data with a good number--otherwise get_input will
#run out of data, causing gets in get_input to return nil, which
will
#cause get_input to go into an infinite loop.
Idealy, I'd like to do it so that when I run my unit test, It'll ask me
for the inputs.
After typing in 5 - 10 different numbers into the command line for hours
on end, you will probably think differently about testing code by hand.
In fact, if you are going to test code by hand, you don't need unit
testing. Unit testing is designed to enable you to test your program by
driving it with another program. That allows you to do tests relatively
quickly and thoroughly--and the tests are repeatable.
