features/test/savedqueries_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 savedqueries feature."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import

import unittest

from features import savedqueries
from framework import monorailrequest
from framework import permissions
from services import service_manager
from testing import fake


class SavedQueriesTest(unittest.TestCase):

  def setUp(self):
    self.services = service_manager.Services(
        user=fake.UserService())
    self.servlet = savedqueries.SavedQueries(services=self.services)
    self.services.user.TestAddUser('a@example.com', 111)

  def testAssertBasePermission(self):
    """Only permit site admins and users viewing themselves."""
    mr = monorailrequest.MonorailRequest(self.services)
    mr.viewed_user_auth.user_id = 111
    mr.auth.user_id = 222

    self.assertRaises(permissions.PermissionException,
                      self.servlet.AssertBasePermission, mr)

    mr.auth.user_id = 111
    self.servlet.AssertBasePermission(mr)

    mr.auth.user_id = 222
    mr.auth.user_pb.is_site_admin = True
    self.servlet.AssertBasePermission(mr)