[Findit] Enable project classifier and component classifier

appengine/findit/model/crash/test/crash_config_test.py

Index: appengine/findit/model/crash/test/crash_config_test.py
diff --git a/appengine/findit/model/crash/test/crash_config_test.py b/appengine/findit/model/crash/test/crash_config_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..9882850636cd9a61f68dae51adb570e0626724d1
--- /dev/null
+++ b/appengine/findit/model/crash/test/crash_config_test.py
@@ -0,0 +1,77 @@
+# 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 re
+import pickle
+import zlib
+
+from google.appengine.api import users
+
+from common.findit_testcase import FinditTestCase
+from model.crash.crash_config import CrashConfig
+
+
+DUMMY_COMPONENT_PATTERNS = {
+    "path_function_component": [
+        [
+          re.compile("src/comp1.*"),
+          None,
+          "Comp1>Dummy"
+        ],
+        [
+          re.compile("src/comp2.*"),
+          re.compile("func2.*"),
+          "Comp2>Dummy"
+        ],
+    ],
+    "top_n": 4
+}
+
+DUMMY_STR_COMPONENT_PATTERNS = {
+    "path_function_component": [
+        [
+          "src/comp1.*",
+          "",
+          "Comp1>Dummy"
+        ],
+        [
+          "src/comp2.*",
+          "func2.*",
+          "Comp2>Dummy"
+        ],
+    ],
+    "top_n": 4
+}
+
+CONFIG_DATA = {
+    'compressed_component_classifier': zlib.compress(pickle.dumps(
+        DUMMY_COMPONENT_PATTERNS)),
+}
+
+
+class CrashAnalysisTest(FinditTestCase):
+
+  def testGetDecompressedComponentClassifierSetting(self):
+    CrashConfig.Get().Update(
+        users.User(email='admin@chromium.org'), True, **{})
+
+    self.assertEqual(CrashConfig.Get().component_classifier, {})
+
+    CrashConfig.Get().Update(
+        users.User(email='admin@chromium.org'), True, **CONFIG_DATA)
+
+    self.assertEqual(CrashConfig.Get().component_classifier,
+                      DUMMY_COMPONENT_PATTERNS)
+
+  def testGetComponentClassifierSettingString(self):
+    CrashConfig.Get().Update(
+        users.User(email='admin@chromium.org'), True, **{})
+
+    self.assertEqual(CrashConfig.Get().str_component_classifier, {})
+
+    CrashConfig.Get().Update(
+        users.User(email='admin@chromium.org'), True, **CONFIG_DATA)
+
+    self.assertEqual(CrashConfig.Get().str_component_classifier,
+                     DUMMY_STR_COMPONENT_PATTERNS)