Project import generated by Copybara.
GitOrigin-RevId: d9e9e3fb4e31372ec1fb43b178994ca78fa8fe70
diff --git a/tracker/test/component_helpers_test.py b/tracker/test/component_helpers_test.py
new file mode 100644
index 0000000..ee7f56c
--- /dev/null
+++ b/tracker/test/component_helpers_test.py
@@ -0,0 +1,113 @@
+# 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 or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Unit tests for the component_helpers module."""
+from __future__ import print_function
+from __future__ import division
+from __future__ import absolute_import
+
+import unittest
+
+from proto import tracker_pb2
+from services import service_manager
+from testing import fake
+from tracker import component_helpers
+from tracker import tracker_bizobj
+
+
+class ComponentHelpersTest(unittest.TestCase):
+
+ def setUp(self):
+ self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
+ self.cd1 = tracker_bizobj.MakeComponentDef(
+ 1, 789, 'FrontEnd', 'doc', False, [], [111], 0, 0)
+ self.cd2 = tracker_bizobj.MakeComponentDef(
+ 2, 789, 'FrontEnd>Splash', 'doc', False, [], [222], 0, 0)
+ self.cd3 = tracker_bizobj.MakeComponentDef(
+ 3, 789, 'BackEnd', 'doc', True, [], [111, 333], 0, 0)
+ self.config.component_defs = [self.cd1, self.cd2, self.cd3]
+ self.services = service_manager.Services(
+ user=fake.UserService(),
+ config=fake.ConfigService())
+ self.services.user.TestAddUser('a@example.com', 111)
+ self.services.user.TestAddUser('b@example.com', 222)
+ self.services.user.TestAddUser('c@example.com', 333)
+ self.mr = fake.MonorailRequest(self.services)
+ self.mr.cnxn = fake.MonorailConnection()
+
+ def testParseComponentRequest_Empty(self):
+ post_data = fake.PostData(admins=[''], cc=[''], labels=[''])
+ parsed = component_helpers.ParseComponentRequest(
+ self.mr, post_data, self.services)
+ self.assertEqual('', parsed.leaf_name)
+ self.assertEqual('', parsed.docstring)
+ self.assertEqual([], parsed.admin_usernames)
+ self.assertEqual([], parsed.cc_usernames)
+ self.assertEqual([], parsed.admin_ids)
+ self.assertEqual([], parsed.cc_ids)
+ self.assertFalse(self.mr.errors.AnyErrors())
+
+ def testParseComponentRequest_Normal(self):
+ post_data = fake.PostData(
+ leaf_name=['FrontEnd'],
+ docstring=['The server-side app that serves pages'],
+ deprecated=[False],
+ admins=['a@example.com'],
+ cc=['b@example.com, c@example.com'],
+ labels=['Hot, Cold'])
+ parsed = component_helpers.ParseComponentRequest(
+ self.mr, post_data, self.services)
+ self.assertEqual('FrontEnd', parsed.leaf_name)
+ self.assertEqual('The server-side app that serves pages', parsed.docstring)
+ self.assertEqual(['a@example.com'], parsed.admin_usernames)
+ self.assertEqual(['b@example.com', 'c@example.com'], parsed.cc_usernames)
+ self.assertEqual(['Hot', 'Cold'], parsed.label_strs)
+ self.assertEqual([111], parsed.admin_ids)
+ self.assertEqual([222, 333], parsed.cc_ids)
+ self.assertEqual([0, 1], parsed.label_ids)
+ self.assertFalse(self.mr.errors.AnyErrors())
+
+ def testParseComponentRequest_InvalidUser(self):
+ post_data = fake.PostData(
+ leaf_name=['FrontEnd'],
+ docstring=['The server-side app that serves pages'],
+ deprecated=[False],
+ admins=['a@example.com, invalid_user'],
+ cc=['b@example.com, c@example.com'],
+ labels=[''])
+ parsed = component_helpers.ParseComponentRequest(
+ self.mr, post_data, self.services)
+ self.assertEqual('FrontEnd', parsed.leaf_name)
+ self.assertEqual('The server-side app that serves pages', parsed.docstring)
+ self.assertEqual(['a@example.com', 'invalid_user'], parsed.admin_usernames)
+ self.assertEqual(['b@example.com', 'c@example.com'], parsed.cc_usernames)
+ self.assertEqual([111], parsed.admin_ids)
+ self.assertEqual([222, 333], parsed.cc_ids)
+ self.assertTrue(self.mr.errors.AnyErrors())
+ self.assertEqual('invalid_user unrecognized', self.mr.errors.member_admins)
+
+ def testGetComponentCcIDs(self):
+ issue = tracker_pb2.Issue()
+ issues_components_cc_ids = component_helpers.GetComponentCcIDs(
+ issue, self.config)
+ self.assertEqual(set(), issues_components_cc_ids)
+
+ issue.component_ids = [1, 2]
+ issues_components_cc_ids = component_helpers.GetComponentCcIDs(
+ issue, self.config)
+ self.assertEqual({111, 222}, issues_components_cc_ids)
+
+ def testGetCcIDsForComponentAndAncestors(self):
+ components_cc_ids = component_helpers.GetCcIDsForComponentAndAncestors(
+ self.config, self.cd1)
+ self.assertEqual({111}, components_cc_ids)
+
+ components_cc_ids = component_helpers.GetCcIDsForComponentAndAncestors(
+ self.config, self.cd2)
+ self.assertEqual({111, 222}, components_cc_ids)
+
+ components_cc_ids = component_helpers.GetCcIDsForComponentAndAncestors(
+ self.config, self.cd3)
+ self.assertEqual({111, 333}, components_cc_ids)