def setUp(self):
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
config=fake.ConfigService(),
issue=fake.IssueService(),
user=fake.UserService(),
user_star=fake.UserStarService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
project_star=fake.ProjectStarService(),
features=fake.FeaturesService())
self.project = self.services.project.TestAddProject('proj',
project_id=987)
self.user = self.services.user.TestAddUser('*****@*****.**', 111)
self.user_2 = self.services.user.TestAddUser('*****@*****.**', 222)
self.group1_id = self.services.usergroup.CreateGroup(
self.cnxn, self.services, '*****@*****.**', 'anyone')
self.group2_id = self.services.usergroup.CreateGroup(
self.cnxn, self.services, '*****@*****.**', 'anyone')
self.services.usergroup.UpdateMembers(self.cnxn, self.group1_id, [111],
'member')
self.services.usergroup.UpdateMembers(self.cnxn, self.group2_id,
[222, 111], 'owner')
self.users_svcr = users_servicer.UsersServicer(self.services,
make_rate_limiter=False)
self.prpc_context = context.ServicerContext()
self.prpc_context.set_code(codes.StatusCode.OK)
def setUp(self):
self.mox = mox.Mox()
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_memcache_stub()
self.testbed.init_datastore_v3_stub()
self.testbed.init_user_stub()
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
cache_manager=fake.CacheManager())
self.project = self.services.project.TestAddProject(
'proj', project_id=789, owner_ids=[111])
self.user = self.services.user.TestAddUser('*****@*****.**', 222)
self.svcr = TestableServicer(self.services)
self.token = xsrf.GenerateToken(222, xsrf.XHR_SERVLET_PATH)
self.request = UpdateSomethingRequest(exc_class=None)
self.prpc_context = context.ServicerContext()
self.prpc_context.set_code(codes.StatusCode.OK)
self.auth = authdata.AuthData(user_id=222, email='*****@*****.**')
self.oauth_patcher = mock.patch(
'google.appengine.api.oauth.get_current_user')
self.mock_oauth_gcu = self.oauth_patcher.start()
self.mock_oauth_gcu.return_value = None
def setUp(self):
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.services = service_manager.Services(user=fake.UserService())
self.mr = testing_helpers.MakeMonorailRequest(project=fake.Project(),
services=self.services)
self.mr.cnxn = fake.MonorailConnection()
self.errors = template_helpers.EZTError()
def setUpServlet(self, servlet_factory):
# pylint: disable=attribute-defined-outside-init
self.services = service_manager.Services(
project=fake.ProjectService(),
config=fake.ConfigService(),
user=fake.UserService(),
issue=fake.IssueService(),
template=Mock(spec=template_svc.TemplateService),
features=fake.FeaturesService())
self.servlet = servlet_factory('req', 'res', services=self.services)
self.project = self.services.project.TestAddProject('proj',
project_id=789,
contrib_ids=[333])
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.services.config.StoreConfig(None, self.config)
self.cnxn = fake.MonorailConnection()
self.mr = testing_helpers.MakeMonorailRequest(path='/p/proj/admin',
project=self.project)
self.mox = mox.Mox()
self.test_template = tracker_bizobj.MakeIssueTemplate(
'Test Template', 'sum', 'New', 111, 'content', [], [], [], [])
self.test_template.template_id = 12345
self.test_templates = testing_helpers.DefaultTemplates()
self.test_templates.append(self.test_template)
self.services.template.GetProjectTemplates\
.return_value = self.test_templates
self.services.template.GetTemplateSetForProject\
.return_value = [(12345, 'Test template', 0)]
def setUp(self):
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
usergroup=fake.UserGroupService(),
user=fake.UserService())
self.user_1 = self.services.user.TestAddUser('*****@*****.**', 111)
self.sitewide_svcr = sitewide_servicer.SitewideServicer(
self.services, make_rate_limiter=False)
def setUp(self):
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
user=fake.UserService(),
usergroup=fake.UserGroupService())
self.services.user.TestAddUser('*****@*****.**', 111)
self.services.user.TestAddUser('*****@*****.**', 888)
self.services.usergroup.TestAddGroupSettings(888, '*****@*****.**')
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_memcache_stub()
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.cache_manager = fake.CacheManager()
self.user_service = MakeUserService(self.cache_manager, self.mox)
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_taskqueue_stub()
self.testbed.init_memcache_stub()
self.testbed.init_datastore_v3_stub()
# Load queue.yaml.
self.taskqueue_stub = self.testbed.get_stub(testbed.TASKQUEUE_SERVICE_NAME)
self.taskqueue_stub._root_path = os.path.dirname(
os.path.dirname(os.path.dirname( __file__ )))
self.services = service_manager.Services(
config=fake.ConfigService(),
issue=fake.IssueService(),
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
template=Mock(spec=template_svc.TemplateService),
features=fake.FeaturesService())
self.project = self.services.project.TestAddProject('proj', project_id=987)
request = webapp2.Request.blank('/p/proj/issues/entry')
response = webapp2.Response()
self.servlet = issueentry.IssueEntry(
request, response, services=self.services)
self.user = self.services.user.TestAddUser('to_pass_tests', 0)
self.services.features.TestAddHotlist(
name='dontcare', summary='', owner_ids=[0])
self.template = testing_helpers.DefaultTemplates()[1]
self.services.template.GetTemplateByName = Mock(return_value=self.template)
self.services.template.GetTemplateSetForProject = Mock(
return_value=[(1, 'name', False)])
# Set-up for testing hotlist parsing.
# Scenario:
# Users: U1, U2, and U3
# Hotlists:
# H1: owned by U1 (private)
# H2: owned by U2, can be edited by U1 (private)
# H2: owned by U3, can be edited by U1 and U2 (public)
self.cnxn = fake.MonorailConnection()
self.U1 = self.services.user.TestAddUser('U1', 111)
self.U2 = self.services.user.TestAddUser('U2', 222)
self.U3 = self.services.user.TestAddUser('U3', 333)
self.H1 = self.services.features.TestAddHotlist(
name='H1', summary='', owner_ids=[111], is_private=True)
self.H2 = self.services.features.TestAddHotlist(
name='H2', summary='', owner_ids=[222], editor_ids=[111],
is_private=True)
self.H2_U3 = self.services.features.TestAddHotlist(
name='H2', summary='', owner_ids=[333], editor_ids=[111, 222],
is_private=False)
self.mox = mox.Mox()
def setUp(self):
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService())
self.project = self.services.project.TestAddProject('proj',
project_id=789,
owner_ids=[111])
self.user = self.services.user.TestAddUser('*****@*****.**', 111)
def setUp(self):
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.config.well_known_labels.append(
tracker_pb2.LabelDef(label='OldLabel',
label_docstring='Do not use any longer',
deprecated=True))
self.services = service_manager.Services(
usergroup=fake.UserGroupService(),
config=fake.ConfigService(),
user=fake.UserService())
self.mr = testing_helpers.MakeMonorailRequest(project=fake.Project(),
services=self.services)
self.mr.cnxn = fake.MonorailConnection()
self.errors = template_helpers.EZTError()
def setUpServlet(self, servlet_factory):
# pylint: disable=attribute-defined-outside-init
self.services = service_manager.Services(
project=fake.ProjectService(),
config=fake.ConfigService(),
user=fake.UserService(),
issue=fake.IssueService(),
features=fake.FeaturesService())
self.servlet = servlet_factory('req', 'res', services=self.services)
self.project = self.services.project.TestAddProject('proj', project_id=789)
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.services.config.StoreConfig(None, self.config)
self.cnxn = fake.MonorailConnection()
self.mr = testing_helpers.MakeMonorailRequest(
path='/p/proj/admin', project=self.project)
self.mox = mox.Mox()
def setUp(self):
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
config=fake.ConfigService(),
issue=fake.IssueService(),
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
project_star=fake.ProjectStarService(),
features=fake.FeaturesService())
self.admin = self.services.user.TestAddUser('*****@*****.**', 123)
self.admin.is_site_admin = True
self.services.user.TestAddUser('*****@*****.**', 111)
self.services.user.TestAddUser('*****@*****.**', 222)
self.services.user.TestAddUser('*****@*****.**', 333)
self.services.user.TestAddUser('*****@*****.**', 444)
self.services.user.TestAddUser('*****@*****.**', 666)
# User group 888 has members: user_555 and [email protected]
self.services.user.TestAddUser('*****@*****.**', 888)
self.services.usergroup.TestAddGroupSettings(
888, '*****@*****.**')
self.services.usergroup.TestAddMembers(888, [555, 1001])
# User group 999 has members: user_111 and user_444
self.services.user.TestAddUser('*****@*****.**', 999)
self.services.usergroup.TestAddGroupSettings(
999, '*****@*****.**')
self.services.usergroup.TestAddMembers(999, [111, 444])
# User group 777 has members: user_666 and group 999.
self.services.user.TestAddUser('*****@*****.**', 777)
self.services.usergroup.TestAddGroupSettings(
777, '*****@*****.**')
self.services.usergroup.TestAddMembers(777, [666, 999])
self.project = self.services.project.TestAddProject(
'proj', project_id=789)
self.project.owner_ids.extend([111])
self.project.committer_ids.extend([222])
self.project.contributor_ids.extend([333])
self.projects_svcr = projects_servicer.ProjectsServicer(
self.services, make_rate_limiter=False)
self.prpc_context = context.ServicerContext()
self.prpc_context.set_code(codes.StatusCode.OK)
def setUp(self):
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.cd1 = tracker_bizobj.MakeComponentDef(1, 789, 'FrontEnd', 'doc',
False, [], [111], 0, 0)
self.cd2 = tracker_bizobj.MakeComponentDef(2, 789, 'FrontEnd>Splash',
'doc', False, [], [222], 0,
0)
self.cd3 = tracker_bizobj.MakeComponentDef(3, 789, 'BackEnd', 'doc',
True, [], [111, 333], 0, 0)
self.config.component_defs = [self.cd1, self.cd2, self.cd3]
self.services = service_manager.Services(user=fake.UserService(),
config=fake.ConfigService())
self.services.user.TestAddUser('*****@*****.**', 111)
self.services.user.TestAddUser('*****@*****.**', 222)
self.services.user.TestAddUser('*****@*****.**', 333)
self.mr = fake.MonorailRequest(self.services)
self.mr.cnxn = fake.MonorailConnection()
def setUp(self):
self.services = service_manager.Services(
issue=fake.IssueService(),
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
features=fake.FeaturesService())
self.cnxn = fake.MonorailConnection()
# Set up for testing getBinnedHotlistViews.
# Project p1; issue i1 in p1; user u1 owns i1; ui1 is an *involved* user.
self.services.user.TestAddUser('u1', 111L)
project = self.services.project.TestAddProject('p1')
issue_local_id = self.services.issue.CreateIssue(
self.cnxn,
self.services,
project_id=project.project_id,
summary='summary',
status='Open',
owner_id=111L,
cc_ids=[],
labels=[],
field_values=[],
component_ids=[],
reporter_id=111L,
marked_description='marked description')
self.issue_id = self.services.issue.LookupIssueID(
self.cnxn, project_id=project.project_id, local_id=issue_local_id)
# ul1 is a *logged in* user.
self.services.user.TestAddUser('ul1', 222L)
# uo1 is an *other* user.
self.services.user.TestAddUser('uo1', 333L)
users_by_id = self.services.user.GetUsersByIDs(self.cnxn,
[111L, 222L, 333L])
self.userviews_by_id = {
k: framework_views.UserView(v)
for k, v in users_by_id.items()
}
self.user_auth = monorailrequest.AuthData.FromEmail(
self.cnxn, 'ul1', self.services)
self.hotlist_item_fields = [(self.issue_id, None, None, None, None)]
def setUp(self):
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
config=fake.ConfigService(),
issue=fake.IssueService(),
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService(),
features=fake.FeaturesService(),
hotlist_star=fake.HotlistStarService())
self.project = self.services.project.TestAddProject(
'proj', project_id=789, owner_ids=[111], contrib_ids=[222, 333])
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.user = self.services.user.TestAddUser('*****@*****.**', 111)
self.user = self.services.user.TestAddUser('*****@*****.**', 222)
self.user = self.services.user.TestAddUser('*****@*****.**', 333)
self.user = self.services.user.TestAddUser('*****@*****.**', 444)
self.features_svcr = features_servicer.FeaturesServicer(
self.services, make_rate_limiter=False)
self.prpc_context = context.ServicerContext()
self.prpc_context.set_code(codes.StatusCode.OK)
self.issue_1 = fake.MakeTestIssue(
789, 1, 'sum', 'New', 111, project_name='proj', issue_id=78901)
self.issue_2 = fake.MakeTestIssue(
789, 2, 'sum', 'New', 111, project_name='proj', issue_id=78902)
self.services.issue.TestAddIssue(self.issue_1)
self.services.issue.TestAddIssue(self.issue_2)
# For testing PredictComponent
self._ml_engine = component_helpers_test.FakeMLEngine(self)
self._top_words = None
self._components_by_index = None
mock.patch(
'services.ml_helpers.setup_ml_engine', lambda: self._ml_engine).start()
mock.patch(
'features.component_helpers._GetTopWords',
lambda _: self._top_words).start()
mock.patch('cloudstorage.open', self.cloudstorageOpen).start()
mock.patch('settings.component_features', 5).start()
self.addCleanup(mock.patch.stopall)
def setUp(self):
self.mox = mox.Mox()
self.cnxn = fake.MonorailConnection()
self.cache_manager = cachemanager_svc.CacheManager()
self.cache_manager.invalidate_tbl = self.mox.CreateMock(
sql.SQLTableManager)
def setUp(self):
self.cnxn = fake.MonorailConnection()
self.services = service_manager.Services(
user=fake.UserService(), usergroup=fake.UserGroupService())
self.services.user.TestAddUser('*****@*****.**', 111)
