[Findit] Extending versioned_model.py to support versioning multiple entities of the same class.

appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py

Index: appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py
diff --git a/appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py b/appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py
index 0eb8285f6e4e80b21d659998dbdb4780eb171b8c..8918370538c56ff524c1e2fa0ac7592ab5d64563 100644
--- a/appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py
+++ b/appengine/findit/waterfall/process_flake_swarming_task_result_pipeline.py
@@ -59,21 +59,29 @@ class ProcessFlakeSwarmingTaskResultPipeline(
     successes = tests_statuses.get(test_name, {}).get('SUCCESS', 0)
 
     if tries > 0:
-      success_rate = successes * 1.0 / tries
+      pass_rate = successes * 1.0 / tries
     else:
-      success_rate = -1 # Special value to indicate test is not existing.
+      pass_rate = -1 # Special value to indicate test is not existing.
+
+    master_flake_analysis = MasterFlakeAnalysis.Get(
+        master_name, builder_name, master_build_number, step_name, test_name)
+    master_flake_analysis.build_numbers.append(build_number)
+    master_flake_analysis.pass_rates.append(pass_rate)
 
-    master_flake_analysis = MasterFlakeAnalysis.Get(master_name, builder_name,
-                                                    master_build_number,
-                                                    step_name, test_name)
     flake_swarming_task = FlakeSwarmingTask.Get(
         master_name, builder_name, build_number, step_name, test_name)
-
-    master_flake_analysis.build_numbers.append(build_number)
-    master_flake_analysis.success_rates.append(success_rate)
     flake_swarming_task.tries = tries
     flake_swarming_task.successes = successes
     flake_swarming_task.put()
+
+    results = flake_swarming_task.ResultsToDict()
+    # TODO (lijeffrey): Determine whether or not this flake swarming task
+    # was a cache hit (already ran results for more iterations than were
+    # requested) and update results['cache_hit'] accordingly.
+    results['cache_hit'] = False
+
+    task_id = flake_swarming_task.task_id
+    master_flake_analysis.swarming_rerun_results[task_id] = results
     master_flake_analysis.put()
     return tests_statuses