blob: 2aeb535a6417e41c1c77d8ee1e312f563793ac6e [file] [log] [blame]
# Copyright 2016 The Chromium Authors. All rights reserved.
# Use of this source code is govered by a BSD-style
# license that can be found in the LICENSE file or at
# https://developers.google.com/open-source/licenses/bsd
"""Unittests for monorail.tracker.spam."""
import unittest
from framework import permissions
from services import service_manager
from services import issue_svc
from testing import fake
from testing import testing_helpers
from tracker import spam
class FlagSpamFormTest(unittest.TestCase):
def setUp(self):
self.cnxn = 'fake cnxn'
self.services = service_manager.Services(
config=fake.ConfigService(),
issue=fake.IssueService(),
user=fake.UserService(),
project=fake.ProjectService(),
spam=fake.SpamService()
)
self.project = self.services.project.TestAddProject('proj', project_id=987)
self.servlet = spam.FlagSpamForm(
'req', 'res', services=self.services)
def testProcessFormData_Permission(self):
local_id_1 = self.services.issue.CreateIssue(
self.cnxn, self.services, self.project.project_id,
'summary_1', 'status', 111L, [], [], [], [], 111L, 'description_1')
# test owner case.
_, mr = testing_helpers.GetRequestObjects(
project=self.project,
perms=permissions.OWNER_ACTIVE_PERMISSIONSET)
mr.local_id = local_id_1
mr.auth_user_id = 222L
post_data = {
'id': local_id_1,
'spam': 'true'
}
res = self.servlet.ProcessFormData(mr, post_data)
self.assertEqual('http://127.0.0.1/p/None/issues/detail?id=1', res)
# test member case.
_, mr = testing_helpers.GetRequestObjects(
project=self.project,
perms=permissions.COMMITTER_ACTIVE_PERMISSIONSET)
mr.local_id = local_id_1
mr.auth_user_id = 222L
post_data = {
'id': local_id_1,
'spam': 'true'
}
res = self.servlet.ProcessFormData(mr, post_data)
self.assertEqual('http://127.0.0.1/p/None/issues/detail?id=1', res)
# test non-member case.
_, mr = testing_helpers.GetRequestObjects(
project=self.project,
perms=permissions.READ_ONLY_PERMISSIONSET)
mr.local_id = local_id_1
mr.auth_user_id = 222L
post_data = {
'id': local_id_1,
'spam': 'true'
}
with self.assertRaises(permissions.PermissionException):
res = self.servlet.ProcessFormData(mr, post_data)
def testProcessFormData_Comment(self):
local_id_1 = self.services.issue.CreateIssue(
self.cnxn, self.services, self.project.project_id,
'summary_1', 'status', 111L, [], [], [], [], 111L, 'description_1')
# test owner case, non-existent comment.
_, mr = testing_helpers.GetRequestObjects(
project=self.project,
perms=permissions.OWNER_ACTIVE_PERMISSIONSET)
mr.local_id = local_id_1
mr.auth_user_id = 222L
post_data = {
'id': local_id_1,
'comment_id': 123,
'spam': 'true'
}
with self.assertRaises(issue_svc.NoSuchCommentException):
res = self.servlet.ProcessFormData(mr, post_data)
comment = self.services.issue.CreateIssueComment(
self.cnxn, self.project.project_id, local_id_1, 111L, "Test comment")
_, mr = testing_helpers.GetRequestObjects(
project=self.project,
perms=permissions.OWNER_ACTIVE_PERMISSIONSET)
mr.local_id = local_id_1
mr.auth_user_id = 222L
post_data = {
'id': local_id_1,
'comment_id': comment.id,
'sequence_num': 2,
'spam': 'true'
}
res = self.servlet.ProcessFormData(mr, post_data)
self.assertEqual('http://127.0.0.1/p/None/issues/detail?id=1', res)