diff --git a/app.rb b/app.rb index d1c71372..f042de8d 100644 --- a/app.rb +++ b/app.rb @@ -256,8 +256,13 @@ get '/?' do end def generate_question - question_first_number = rand 5 - question_last_number = rand 5 + if ENV['RACK_ENV'] == 'test' + question_first_number = 1 + question_last_number = 1 + else + question_first_number = rand 5 + question_last_number = rand 5 + end session[:question_answer] = (question_first_number + question_last_number).to_s [question_first_number, question_last_number] end