[Findit] Show build analysis references in UI for Findit Cross-platform auto-triage

appengine/findit/model/test/wf_analysis_test.py

 # Copyright 2015 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.
 
 from datetime import datetime
 
 import unittest
 
 from common.waterfall import failure_type
-from model.wf_analysis import WfAnalysis
 from model import analysis_status
 from model import result_status
+from model.wf_analysis import WfAnalysis
 
 
 class WfAnalysisTest(unittest.TestCase):
+
   def testWfAnalysisStatusIsCompleted(self):
     for status in (analysis_status.COMPLETED, analysis_status.ERROR):
       analysis = WfAnalysis.Create('m', 'b', 123)
       analysis.status = status
       self.assertTrue(analysis.completed)
 
   def testWfAnalysisStatusIsNotCompleted(self):
     for status in (analysis_status.PENDING, analysis_status.RUNNING):
       analysis = WfAnalysis.Create('m', 'b', 123)
       analysis.status = status
@@ skipping 31 matching lines @@
   def testWfAnalysisStatusIsNotFailed(self):
     for status in (analysis_status.PENDING, analysis_status.RUNNING,
                    analysis_status.COMPLETED):
       analysis = WfAnalysis.Create('m', 'b', 123)
       analysis.status = status
       self.assertFalse(analysis.failed)
 
   def testWfAnalysisStatusDescription(self):
     analysis = WfAnalysis.Create('m', 'b', 123)
     analysis.status = analysis_status.PENDING
-    self.assertEqual("Pending", analysis.status_description)
+    self.assertEqual('Pending', analysis.status_description)
     analysis.status = analysis_status.RUNNING
-    self.assertEqual("Running", analysis.status_description)
+    self.assertEqual('Running', analysis.status_description)
     analysis.status = analysis_status.COMPLETED
-    self.assertEqual("Completed", analysis.status_description)
+    self.assertEqual('Completed', analysis.status_description)
     analysis.status = analysis_status.ERROR
-    self.assertEqual("Error", analysis.status_description)
+    self.assertEqual('Error', analysis.status_description)
 
   def testWfAnalysisResultStatusDescription(self):
     analysis = WfAnalysis.Create('m', 'b', 123)
 
     self.assertEqual('', analysis.result_status_description)
     analysis.result_status = result_status.FOUND_CORRECT
-    self.assertEqual("Correct - Found", analysis.result_status_description)
+    self.assertEqual('Correct - Found', analysis.result_status_description)
 
   def testWfAnalysisCorrectnessIsUnknownIfIncompletedOrFailed(self):
     for status in (analysis_status.PENDING, analysis_status.RUNNING,
                    analysis_status.ERROR):
       analysis = WfAnalysis.Create('m', 'b', 123)
       analysis.status = status
       self.assertIsNone(analysis.correct)
 
   def testWfAnalysisCorrectnessIsUnknownIfUntriaged(self):
     for status in (result_status.FOUND_UNTRIAGED,
-                          result_status.NOT_FOUND_UNTRIAGED):
+                   result_status.NOT_FOUND_UNTRIAGED):
       analysis = WfAnalysis.Create('m', 'b', 123)
       analysis.status = analysis_status.COMPLETED
       analysis.result_status = status
       self.assertIsNone(analysis.correct)
 
   def testWfAnalysisHasCorrectResult(self):
     for status in (
         result_status.FOUND_CORRECT,
         result_status.NOT_FOUND_CORRECT,
         result_status.FOUND_CORRECT_DUPLICATE):
@@ skipping 35 matching lines @@
 
   def testFailureTypeForLegacyDataWithFailedTestStep(self):
     analysis = WfAnalysis.Create('m', 'b', 123)
     analysis.result = {'failures': [{'step_name': 'browser_tests'}]}
     self.assertEqual(failure_type.TEST, analysis.failure_type)
 
   def testFailureTypeStr(self):
     analysis = WfAnalysis.Create('m', 'b', 123)
     analysis.result = {'failures': [{'step_name': 'browser_tests'}]}
     self.assertEqual('test', analysis.failure_type_str)
+
+  def testWfAnalysisIsDuplicate(self):
+    for status in (
+        result_status.FOUND_CORRECT_DUPLICATE,
+        result_status.FOUND_INCORRECT_DUPLICATE):
+      analysis = WfAnalysis.Create('m', 'b', 123)
+      analysis.result_status = status
+      self.assertTrue(analysis.is_duplicate)
+
+  def testWfAnalysisIsNotDuplicate(self):
+    for status in (
+        result_status.FOUND_CORRECT,
+        result_status.FOUND_INCORRECT,
+        result_status.NOT_FOUND_INCORRECT,
+        result_status.FOUND_UNTRIAGED,
+        result_status.NOT_FOUND_UNTRIAGED,
+        result_status.NOT_FOUND_CORRECT,
+        None):
+      analysis = WfAnalysis.Create('m', 'b', 123)
+      analysis.result_status = status
+      self.assertFalse(analysis.is_duplicate)