framework/test/framework_views_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.

"""Unit tests for framework_views classes."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import

import time
import unittest

from framework import framework_constants
from framework import framework_views
from framework import monorailrequest
from mrproto import project_pb2
from mrproto import tracker_pb2
from mrproto import user_pb2
import settings
from services import service_manager
from testing import fake


LONG_STR = 'VeryLongStringThatCertainlyWillNotFit'
LONG_PART_STR = 'OnePartThatWillNotFit-OneShort'


class LabelViewTest(unittest.TestCase):

  def testLabelView(self):
    view = framework_views.LabelView('', None)
    self.assertEqual('', view.name)

    view = framework_views.LabelView('Priority-High', None)
    self.assertEqual('Priority-High', view.name)
    self.assertIsNone(view.is_restrict)
    self.assertEqual('', view.docstring)
    self.assertEqual('Priority', view.prefix)
    self.assertEqual('High', view.value)

    view = framework_views.LabelView('%s-%s' % (LONG_STR, LONG_STR), None)
    self.assertEqual('%s-%s' % (LONG_STR, LONG_STR), view.name)
    self.assertEqual('', view.docstring)
    self.assertEqual(LONG_STR, view.prefix)
    self.assertEqual(LONG_STR, view.value)

    view = framework_views.LabelView(LONG_PART_STR, None)
    self.assertEqual(LONG_PART_STR, view.name)
    self.assertEqual('', view.docstring)
    self.assertEqual('OnePartThatWillNotFit', view.prefix)
    self.assertEqual('OneShort', view.value)

    config = tracker_pb2.ProjectIssueConfig()
    config.well_known_labels.append(tracker_pb2.LabelDef(
        label='Priority-High', label_docstring='Must ship in this milestone'))

    view = framework_views.LabelView('Priority-High', config)
    self.assertEqual('Must ship in this milestone', view.docstring)

    view = framework_views.LabelView('Priority-Foo', config)
    self.assertEqual('', view.docstring)

    view = framework_views.LabelView('Restrict-View-Commit', None)
    self.assertTrue(view.is_restrict)


class StatusViewTest(unittest.TestCase):

  def testStatusView(self):
    view = framework_views.StatusView('', None)
    self.assertEqual('', view.name)

    view = framework_views.StatusView('Accepted', None)
    self.assertEqual('Accepted', view.name)
    self.assertEqual('', view.docstring)
    self.assertEqual('yes', view.means_open)

    view = framework_views.StatusView(LONG_STR, None)
    self.assertEqual(LONG_STR, view.name)
    self.assertEqual('', view.docstring)
    self.assertEqual('yes', view.means_open)

    config = tracker_pb2.ProjectIssueConfig()
    config.well_known_statuses.append(tracker_pb2.StatusDef(
        status='SlamDunk', status_docstring='Code fixed and taught a lesson',
        means_open=False))

    view = framework_views.StatusView('SlamDunk', config)
    self.assertEqual('Code fixed and taught a lesson', view.docstring)
    self.assertFalse(view.means_open)

    view = framework_views.StatusView('SlammedBack', config)
    self.assertEqual('', view.docstring)


class UserViewTest(unittest.TestCase):

  def setUp(self):
    self.user = user_pb2.User(user_id=111)

  def testGetAvailablity_Anon(self):
    self.user.user_id = 0
    user_view = framework_views.UserView(self.user)
    self.assertEqual(None, user_view.avail_message)
    self.assertEqual(None, user_view.avail_state)

  def testGetAvailablity_Banned(self):
    self.user.banned = 'spamming'
    user_view = framework_views.UserView(self.user)
    self.assertEqual('Banned', user_view.avail_message)
    self.assertEqual('banned', user_view.avail_state)

  def testGetAvailablity_Vacation(self):
    self.user.vacation_message = 'gone fishing'
    user_view = framework_views.UserView(self.user)
    self.assertEqual('gone fishing', user_view.avail_message)
    self.assertEqual('none', user_view.avail_state)

    self.user.vacation_message = (
      'Gone fishing as really long time with lots of friends and reading '
      'a long novel by a famous author.  I wont have internet access but '
      'If you urgently need anything you can call Alice or Bob for most '
      'things otherwise call Charlie.  Wish me luck! ')
    user_view = framework_views.UserView(self.user)
    self.assertTrue(len(user_view.avail_message) >= 50)
    self.assertTrue(len(user_view.avail_message_short) < 50)
    self.assertEqual('none', user_view.avail_state)

  def testGetAvailablity_Bouncing(self):
    self.user.email_bounce_timestamp = 1234567890
    user_view = framework_views.UserView(self.user)
    self.assertEqual('Email to this user bounced', user_view.avail_message)
    self.assertEqual(user_view.avail_message_short, user_view.avail_message)
    self.assertEqual('none', user_view.avail_state)

  def testGetAvailablity_Groups(self):
    user_view = framework_views.UserView(self.user, is_group=True)
    self.assertEqual(None, user_view.avail_message)
    self.assertEqual(None, user_view.avail_state)

    self.user.email = 'likely-user-group@example.com'
    user_view = framework_views.UserView(self.user)
    self.assertEqual(None, user_view.avail_message)
    self.assertEqual(None, user_view.avail_state)

  def testGetAvailablity_NeverVisitied(self):
    self.user.last_visit_timestamp = 0
    user_view = framework_views.UserView(self.user)
    self.assertEqual('User never visited', user_view.avail_message)
    self.assertEqual('never', user_view.avail_state)

  def testGetAvailablity_NotRecent(self):
    now = int(time.time())
    self.user.last_visit_timestamp = now - 20 * framework_constants.SECS_PER_DAY
    user_view = framework_views.UserView(self.user)
    self.assertEqual('Last visit 20 days ago', user_view.avail_message)
    self.assertEqual('unsure', user_view.avail_state)

  def testGetAvailablity_ReallyLongTime(self):
    now = int(time.time())
    self.user.last_visit_timestamp = now - 99 * framework_constants.SECS_PER_DAY
    user_view = framework_views.UserView(self.user)
    self.assertEqual('Last visit > 30 days ago', user_view.avail_message)
    self.assertEqual('none', user_view.avail_state)

  def testDeletedUser(self):
    deleted_user = user_pb2.User(user_id=1)
    user_view = framework_views.UserView(deleted_user)
    self.assertEqual(
        user_view.display_name, framework_constants.DELETED_USER_NAME)
    self.assertEqual(user_view.email, '')
    self.assertEqual(user_view.obscure_email, '')
    self.assertEqual(user_view.profile_url, '')

class RevealEmailsToMembersTest(unittest.TestCase):

  def setUp(self):
    self.cnxn = fake.MonorailConnection()
    self.services = service_manager.Services(
        project=fake.ProjectService(),
        user=fake.UserService(),
        usergroup=fake.UserGroupService())
    self.mr = monorailrequest.MonorailRequest(None)
    self.mr.project = self.services.project.TestAddProject(
        'proj',
        project_id=789,
        owner_ids=[111],
        committer_ids=[222],
        contrib_ids=[333, 888])
    user = self.services.user.TestAddUser('test@example.com', 1000)
    self.mr.auth.user_pb = user

  def CheckRevealAllToMember(
      self, logged_in_user_id, expected, viewed_user_id=333, group_id=None):
    user_view = framework_views.StuffUserView(
        viewed_user_id, 'user@example.com', True)

    if group_id:
      pass  # xxx re-implement groups

    users_by_id = {333: user_view}
    self.mr.auth.user_id = logged_in_user_id
    self.mr.auth.effective_ids = {logged_in_user_id}
    # Assert display name is obscured before the reveal.
    self.assertEqual('u...@example.com', user_view.display_name)
    # Assert profile url contains user ID before the reveal.
    self.assertEqual('/u/%s/' % viewed_user_id, user_view.profile_url)
    framework_views.RevealAllEmailsToMembers(
        self.cnxn, self.services, self.mr.auth, users_by_id)
    self.assertEqual(expected, not user_view.obscure_email)
    if expected:
      # Assert display name is now revealed.
      self.assertEqual('user@example.com', user_view.display_name)
      # Assert profile url contains the email.
      self.assertEqual('/u/user@example.com/', user_view.profile_url)
    else:
      # Assert display name is still hidden.
      self.assertEqual('u...@example.com', user_view.display_name)
      # Assert profile url still contains user ID.
      self.assertEqual('/u/%s/' % viewed_user_id, user_view.profile_url)

  # TODO(https://crbug.com/monorail/8192): Remove this method and related test.
  def DeprecatedCheckRevealAllToMember(
      self, logged_in_user_id, expected, viewed_user_id=333, group_id=None):
    user_view = framework_views.StuffUserView(
        viewed_user_id, 'user@example.com', True)

    if group_id:
      pass  # xxx re-implement groups

    users_by_id = {333: user_view}
    self.mr.auth.user_id = logged_in_user_id
    self.mr.auth.effective_ids = {logged_in_user_id}
    # Assert display name is obscured before the reveal.
    self.assertEqual('u...@example.com', user_view.display_name)
    # Assert profile url contains user ID before the reveal.
    self.assertEqual('/u/%s/' % viewed_user_id, user_view.profile_url)
    framework_views.RevealAllEmailsToMembers(
        self.cnxn, self.services, self.mr.auth, users_by_id, self.mr.project)
    self.assertEqual(expected, not user_view.obscure_email)
    if expected:
      # Assert display name is now revealed.
      self.assertEqual('user@example.com', user_view.display_name)
      # Assert profile url contains the email.
      self.assertEqual('/u/user@example.com/', user_view.profile_url)
    else:
      # Assert display name is still hidden.
      self.assertEqual('u...@example.com', user_view.display_name)
      # Assert profile url still contains user ID.
      self.assertEqual('/u/%s/' % viewed_user_id, user_view.profile_url)

  def testDontRevealEmailsToPriviledgedDomain(self):
    """We no longer give this advantage based on email address domain."""
    for priviledged_user_domain in settings.priviledged_user_domains:
      self.mr.auth.user_pb.email = 'test@' + priviledged_user_domain
      self.CheckRevealAllToMember(100001, False)

  def testRevealEmailToSelf(self):
    logged_in_user = self.services.user.TestAddUser('user@example.com', 333)
    self.mr.auth.user_pb = logged_in_user
    self.CheckRevealAllToMember(333, True)

  def testRevealAllEmailsToMembers_Collaborators(self):
    self.CheckRevealAllToMember(0, False)
    self.CheckRevealAllToMember(111, True)
    self.CheckRevealAllToMember(222, True)
    self.CheckRevealAllToMember(333, True)
    self.CheckRevealAllToMember(444, False)

    # Viewed user has indirect role in the project via a group.
    self.CheckRevealAllToMember(0, False, group_id=888)
    self.CheckRevealAllToMember(111, True, group_id=888)
    # xxx re-implement
    # self.CheckRevealAllToMember(
    #     111, True, viewed_user_id=444, group_id=888)

    # Logged in user has indirect role in the project via a group.
    self.CheckRevealAllToMember(888, True)

  def testDeprecatedRevealAllEmailsToMembers_Collaborators(self):
    self.DeprecatedCheckRevealAllToMember(0, False)
    self.DeprecatedCheckRevealAllToMember(111, True)
    self.DeprecatedCheckRevealAllToMember(222, True)
    self.DeprecatedCheckRevealAllToMember(333, True)
    self.DeprecatedCheckRevealAllToMember(444, False)

    # Viewed user has indirect role in the project via a group.
    self.DeprecatedCheckRevealAllToMember(0, False, group_id=888)
    self.DeprecatedCheckRevealAllToMember(111, True, group_id=888)

    # Logged in user has indirect role in the project via a group.
    self.DeprecatedCheckRevealAllToMember(888, True)

  def testRevealAllEmailsToMembers_Admins(self):
    self.CheckRevealAllToMember(555, False)
    self.mr.auth.user_pb.is_site_admin = True
    self.CheckRevealAllToMember(555, True)


class RevealAllEmailsTest(unittest.TestCase):

  def testRevealAllEmail(self):
    users_by_id = {
        111: framework_views.StuffUserView(111, 'a@a.com', True),
        222: framework_views.StuffUserView(222, 'b@b.com', True),
        333: framework_views.StuffUserView(333, 'c@c.com', True),
        999: framework_views.StuffUserView(999, 'z@z.com', True),
        }
    # Assert display names are obscured before the reveal.
    self.assertEqual('a...@a.com', users_by_id[111].display_name)
    self.assertEqual('b...@b.com', users_by_id[222].display_name)
    self.assertEqual('c...@c.com', users_by_id[333].display_name)
    self.assertEqual('z...@z.com', users_by_id[999].display_name)

    framework_views.RevealAllEmails(users_by_id)

    self.assertFalse(users_by_id[111].obscure_email)
    self.assertFalse(users_by_id[222].obscure_email)
    self.assertFalse(users_by_id[333].obscure_email)
    self.assertFalse(users_by_id[999].obscure_email)
    # Assert display names are now revealed.
    self.assertEqual('a@a.com', users_by_id[111].display_name)
    self.assertEqual('b@b.com', users_by_id[222].display_name)
    self.assertEqual('c@c.com', users_by_id[333].display_name)
    self.assertEqual('z@z.com', users_by_id[999].display_name)