mrproto/test/usergroup_pb2_test.py

# Copyright 2016 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Tests for usergroup_pb2 functions."""

from __future__ import print_function
from __future__ import division
from __future__ import absolute_import

import unittest

from mrproto import usergroup_pb2


class UserGroupPb2Test(unittest.TestCase):

  def testMakeSettings_Defaults(self):
    usergroup = usergroup_pb2.MakeSettings('anyone')
    self.assertEqual(
        usergroup_pb2.MemberVisibility.ANYONE,
        usergroup.who_can_view_members)
    self.assertIsNone(usergroup.ext_group_type)
    self.assertEqual(0, usergroup.last_sync_time)
    self.assertEqual([], usergroup.friend_projects)

  def testMakeSettings_Everything(self):
    usergroup = usergroup_pb2.MakeSettings(
        'Members', ext_group_type_str='mdb',
        last_sync_time=1234567890, friend_projects=[789])
    self.assertEqual(
        usergroup_pb2.MemberVisibility.MEMBERS,
        usergroup.who_can_view_members)
    self.assertEqual(usergroup_pb2.GroupType.MDB, usergroup.ext_group_type)
    self.assertEqual(1234567890, usergroup.last_sync_time)
    self.assertEqual([789], usergroup.friend_projects)