From fdcaaf01774415fd0be69dfe637dc31c6f938a27 Mon Sep 17 00:00:00 2001
From: Kyle Drake <kyle@kyledrake.net>
Date: Mon, 27 Oct 2014 17:17:12 -0700
Subject: [PATCH] fix question_answer for testing

---
 app.rb | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

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