blob: 5bde7c6dd038c34a7b81d50c185a623353908360 [file] [log] [blame]
# 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)