[Findit] Added algorithm to analysis

appengine/findit/handlers/flake/test/check_flake_test.py

 # Copyright 2016 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import webapp2
 
 from handlers.flake import check_flake
 from model.flake.master_flake_analysis import MasterFlakeAnalysis
 from model import analysis_status
+from model.analysis_status import STATUS_TO_DESCRIPTION
 from waterfall.test import wf_testcase
 
 
 class CheckFlakeTest(wf_testcase.WaterfallTestCase):
   app_module = webapp2.WSGIApplication([
       ('/waterfall/check-flake', check_flake.CheckFlake),
   ], debug=True)
 
   def _CreateAndSaveMasterFlakeAnalysis(
       self, master_name, builder_name, build_number,
@@ skipping 43 matching lines @@
     response = self.test_app.get('/waterfall/check-flake', params={
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'step_name': step_name,
         'test_name': test_name,
         'format': 'json'})
 
     self.assertEquals(200, response.status_int)
     expected_check_flake_result ={
-        'success_rates': [[int(build_number), success_rate]]
+        'success_rates': [[int(build_number), success_rate]],
+        'analysis_status': STATUS_TO_DESCRIPTION.get(
+            master_flake_analysis.status),
+        'suspected_flake_build_number': None
     }
     self.assertEqual(expected_check_flake_result, response.json_body)