OLD | NEW |
(Empty) | |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is govered by a BSD-style |
| 3 # license that can be found in the LICENSE file or at |
| 4 # https://developers.google.com/open-source/licenses/bsd |
| 5 |
| 6 """Unit test for User Group helpers.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 from proto import usergroup_pb2 |
| 11 from sitewide import group_helpers |
| 12 |
| 13 |
| 14 class GroupHelpersTest(unittest.TestCase): |
| 15 |
| 16 def testGroupVisibilityView(self): |
| 17 gvv_anyone = group_helpers.GroupVisibilityView( |
| 18 usergroup_pb2.MemberVisibility.ANYONE) |
| 19 gvv_members = group_helpers.GroupVisibilityView( |
| 20 usergroup_pb2.MemberVisibility.MEMBERS) |
| 21 gvv_owners = group_helpers.GroupVisibilityView( |
| 22 usergroup_pb2.MemberVisibility.OWNERS) |
| 23 self.assertEqual('Anyone on the Internet', gvv_anyone.name) |
| 24 self.assertEqual('Group Members', gvv_members.name) |
| 25 self.assertEqual('Group Owners', gvv_owners.name) |
| 26 |
| 27 def testGroupMemberView(self): |
| 28 gmv = group_helpers.GroupMemberView( |
| 29 1L, 'test@example.com', 't...@example.com', 888L, 'member') |
| 30 self.assertEqual(888L, gmv.group_id) |
| 31 self.assertEqual('member', gmv.role) |
| 32 |
| 33 def testBuildUserGroupVisibilityOptions(self): |
| 34 vis_views = group_helpers.BuildUserGroupVisibilityOptions() |
| 35 self.assertEqual(3, len(vis_views)) |
| 36 |
| 37 def testGroupTypeView(self): |
| 38 gt_cia = group_helpers.GroupTypeView( |
| 39 usergroup_pb2.GroupType.CHROME_INFRA_AUTH) |
| 40 gt_mdb = group_helpers.GroupTypeView( |
| 41 usergroup_pb2.GroupType.MDB) |
| 42 self.assertEqual('Chrome-infra-auth', gt_cia.name) |
| 43 self.assertEqual('MDB', gt_mdb.name) |
| 44 |
| 45 def testBuildUserGroupTypeOptions(self): |
| 46 group_types = group_helpers.BuildUserGroupTypeOptions() |
| 47 self.assertEqual(3, len(group_types)) |
| 48 |
OLD | NEW |