[Findit] Add findit_for_client to do analysis based on client_id

appengine/findit/crash/test/findit_for_client_test.py

Index: appengine/findit/crash/test/findit_for_client_test.py
diff --git a/appengine/findit/crash/test/findit_for_client_test.py b/appengine/findit/crash/test/findit_for_client_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..1775a687bdcb2176eafca01d7b063e37894ad8d8
--- /dev/null
+++ b/appengine/findit/crash/test/findit_for_client_test.py
@@ -0,0 +1,214 @@
+# 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 copy
+import json
+
+from google.appengine.api import app_identity
+
+from crash import findit_for_client
+from crash import findit_for_chromecrash
+from crash.test.crash_testcase import CrashTestCase
+from crash.type_enums import CrashClient
+from model.crash.fracas_crash_analysis import FracasCrashAnalysis
+
+
+class FinditForClientTest(CrashTestCase):
+
+  def testCheckPolicyUnsupportedClient(self):
+    pass_check, _ = findit_for_client.CheckPolicyForClient(
+        {'signature': 'sig'}, '1', 'sig', 'unsupported_client',
+        'canary', 'stack_trace', {'channel': 'canary'})
+    self.assertFalse(pass_check)
+
+  def testCheckPolicyUnsupportedPlatform(self):
+    chrome_version = '1'
+    signature = 'signature'
+    platform = 'unsupported_platform'
+    crash_identifiers = {
+      'chrome_version': chrome_version,
+      'signature': signature,
+      'channel': 'canary',
+      'platform': platform,
+      'process_type': 'browser',
+    }
+
+    pass_check, _ = findit_for_client.CheckPolicyForClient(
+        crash_identifiers, chrome_version, signature,
+        CrashClient.FRACAS, platform, 'stack_trace', {'channel': 'canary'})
+    self.assertFalse(pass_check)
+
+  def testCheckPolicyBlacklistedSignature(self):
+    chrome_version = '1'
+    signature = 'Blacklist marker signature'
+    platform = 'win'
+    crash_identifiers = {
+      'chrome_version': chrome_version,
+      'signature': signature,
+      'channel': 'canary',
+      'platform': platform,
+      'process_type': 'browser',
+    }
+
+    pass_check, _ = findit_for_client.CheckPolicyForClient(
+        crash_identifiers, chrome_version, signature,
+        CrashClient.FRACAS, platform, 'stack_trace', {'channel': 'canary'})
+    self.assertFalse(pass_check)
+
+  def testCheckPolicyPlatformRename(self):
+    chrome_version = '1'
+    signature = 'signature'
+    platform = 'linux'
+    crash_identifiers = {
+      'chrome_version': chrome_version,
+      'signature': signature,
+      'channel': 'canary',
+      'platform': platform,
+      'process_type': 'browser',
+    }
+
+    pass_check, args = findit_for_client.CheckPolicyForClient(
+        crash_identifiers, chrome_version, signature,
+        CrashClient.FRACAS, platform, 'stack_trace', {'channel': 'canary'})
+    self.assertTrue(pass_check)
+    self.assertEqual(args[4], 'unix')
+
+  def testGetAnalysisForClient(self):
+    crash_identifiers = {'signature': 'sig'}
+    analysis = FracasCrashAnalysis.Create(crash_identifiers)
+    analysis.put()
+
+    self.assertEqual(findit_for_client.GetAnalysisForClient(
+        crash_identifiers, CrashClient.FRACAS), analysis)
+
+  def testGetAnalysisForUnsuportedClient(self):
+    crash_identifiers = {'signature': 'sig'}
+    analysis = FracasCrashAnalysis.Create(crash_identifiers)
+    analysis.put()
+
+    self.assertIsNone(findit_for_client.GetAnalysisForClient(
+        crash_identifiers, 'Unsupported_client'), analysis)
+
+  def testResetAnalysisForFracas(self):
+    chrome_version = '1'
+    signature = 'signature'
+    platform = 'linux'
+    stack_trace = 'stack_trace'
+    crash_identifiers = {
+      'chrome_version': chrome_version,
+      'signature': signature,
+      'channel': 'canary',
+      'platform': platform,
+      'process_type': 'browser',
+    }
+    customized_data = {'channel': 'canary'}
+
+    analysis = FracasCrashAnalysis.Create(crash_identifiers)
+
+    findit_for_client.ResetAnalysis(
+        analysis, chrome_version, signature, CrashClient.FRACAS, platform,
+        stack_trace, customized_data)
+
+    analysis = FracasCrashAnalysis.Get(crash_identifiers)
+    self.assertEqual(analysis.crashed_version, chrome_version)
+    self.assertEqual(analysis.signature, signature)
+    self.assertEqual(analysis.platform, platform)
+    self.assertEqual(analysis.stack_trace, stack_trace)
+    self.assertEqual(analysis.channel, customized_data['channel'])
+
+  def testCreateAnalysisForClient(self):
+    crash_identifiers = {'signature': 'sig'}
+    self.assertIsNotNone(findit_for_client.CreateAnalysisForClient(
+        crash_identifiers, CrashClient.FRACAS))
+
+  def testCreateAnalysisForUnsupportedClientId(self):
+    crash_identifiers = {'signature': 'sig'}
+    self.assertIsNone(findit_for_client.CreateAnalysisForClient(
+        crash_identifiers, 'unsupported_id'))
+
+  def testGetPublishResultFromAnalysisFoundTrue(self):
+    mock_host = 'https://host.com'
+    self.mock(app_identity, 'get_default_version_hostname', lambda: mock_host)
+
+    analysis_result = {
+        'found': True,
+        'suspected_cls': [
+            {'confidence': 0.21434,
+             'reason': ['reason1', 'reason2'],
+             'other': 'data'}
+        ],
+        'other_data': 'data',
+    }
+
+    processed_analysis_result = copy.deepcopy(analysis_result)
+    processed_analysis_result['feedback_url'] = (
+        mock_host + '/crash/fracas-result-feedback?'
+        'key=agx0ZXN0YmVkLXRlc3RyQQsSE0ZyYWNhc0NyYXNoQW5hbHlzaXMiKDMzNTY5MDU3'
+        'M2ZlYTFlZGZhMjViOTVjZmI4OGZhODFlNDk0YTEyODkM')
+
+    for cl in processed_analysis_result['suspected_cls']:
+      cl['confidence'] = round(cl['confidence'], 2)
+      cl.pop('reason', None)
+
+    crash_identifiers = {'signature': 'sig'}
+    expected_messages_data = {
+            'crash_identifiers': crash_identifiers,
+            'client_id': CrashClient.FRACAS,
+            'result': processed_analysis_result,
+    }
+
+    analysis = FracasCrashAnalysis.Create(crash_identifiers)
+    analysis.client_id = CrashClient.FRACAS
+    analysis.result = analysis_result
+
+    self.assertEqual(findit_for_client.GetPublishResultFromAnalysis(
+        analysis, crash_identifiers,
+        CrashClient.FRACAS), expected_messages_data)
+
+  def testGetPublishResultFromAnalysisFoundFalse(self):
+    mock_host = 'https://host.com'
+    self.mock(app_identity, 'get_default_version_hostname', lambda: mock_host)
+
+    analysis_result = {
+        'found': False,
+    }
+
+    processed_analysis_result = copy.deepcopy(analysis_result)
+    processed_analysis_result['feedback_url'] = (
+        mock_host + '/crash/fracas-result-feedback?'
+        'key=agx0ZXN0YmVkLXRlc3RyQQsSE0ZyYWNhc0NyYXNoQW5hbHlzaXMiKDMzNTY5MDU3'
+        'M2ZlYTFlZGZhMjViOTVjZmI4OGZhODFlNDk0YTEyODkM')
+
+    crash_identifiers = {'signature': 'sig'}
+    expected_messages_data = {
+            'crash_identifiers': crash_identifiers,
+            'client_id': CrashClient.FRACAS,
+            'result': processed_analysis_result,
+    }
+
+    analysis = FracasCrashAnalysis.Create(crash_identifiers)
+    analysis.client_id = CrashClient.FRACAS
+    analysis.result = analysis_result
+
+    self.assertEqual(findit_for_client.GetPublishResultFromAnalysis(
+        analysis, crash_identifiers,
+        CrashClient.FRACAS), expected_messages_data)
+
+  def testFindCulprit(self):
+    expected_result = {'found': False}
+    expected_tags = {'found_suspects': False,
+                     'has_regression_range': False}
+
+    def _MockFindCulpritForChromeCrash(*_):
+      return expected_result, expected_tags
+
+    self.mock(findit_for_chromecrash, 'FindCulpritForChromeCrash',
+              _MockFindCulpritForChromeCrash)
+
+    analysis = FracasCrashAnalysis.Create({'signature': 'sig'})
+    analysis.client_id = CrashClient.FRACAS
+
+    result, tags = findit_for_client.FindCulprit(analysis)
+    self.assertEqual(result, expected_result)
+    self.assertEqual(tags, expected_tags)