blob: 8392481d96d93b44dcb44ee093a34edd05882aa2 [file] [log] [blame]
# -*- coding: utf-8 -*-
# Copyright 2018 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 chart_svc module."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import datetime
try:
from mox3 import mox
except ImportError:
import mox
import re
import settings
import unittest
from google.appengine.ext import testbed
from services import chart_svc
from services import config_svc
from services import service_manager
from framework import permissions
from framework import sql
from mrproto import ast_pb2
from mrproto import tracker_pb2
from search import ast2select
from search import search_helpers
from testing import fake
from tracker import tracker_bizobj
def MakeChartService(my_mox, config):
chart_service = chart_svc.ChartService(config)
for table_var in ['issuesnapshot_tbl', 'issuesnapshot2label_tbl',
'issuesnapshot2component_tbl', 'issuesnapshot2cctbl', 'labeldef_tbl']:
setattr(chart_service, table_var, my_mox.CreateMock(sql.SQLTableManager))
return chart_service
class ChartServiceTest(unittest.TestCase):
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_memcache_stub()
self.mox = mox.Mox()
self.cnxn = self.mox.CreateMock(sql.MonorailConnection)
self.services = service_manager.Services()
self.config_service = fake.ConfigService()
self.services.config = self.config_service
self.services.chart = MakeChartService(self.mox, self.config_service)
self.services.issue = fake.IssueService()
self.mox.StubOutWithMock(self.services.chart, '_QueryToWhere')
self.mox.StubOutWithMock(search_helpers, 'GetPersonalAtRiskLabelIDs')
self.mox.StubOutWithMock(settings, 'num_logical_shards')
settings.num_logical_shards = 1
self.mox.StubOutWithMock(self.services.chart, '_currentTime')
self.defaultLeftJoins = [
('Issue ON IssueSnapshot.issue_id = Issue.id', []),
('Issue2Label AS Forbidden_label'
' ON Issue.id = Forbidden_label.issue_id'
' AND Forbidden_label.label_id IN (%s,%s)', [91, 81]),
('Issue2Cc AS I2cc'
' ON Issue.id = I2cc.issue_id'
' AND I2cc.cc_id IN (%s,%s)', [10, 20]),
]
self.defaultWheres = [
('IssueSnapshot.period_start <= %s', [1514764800]),
('IssueSnapshot.period_end > %s', [1514764800]),
('Issue.is_spam = %s', [False]),
('Issue.deleted = %s', [False]),
('IssueSnapshot.project_id IN (%s)', [789]),
('(Issue.reporter_id IN (%s,%s)'
' OR Issue.owner_id IN (%s,%s)'
' OR I2cc.cc_id IS NOT NULL'
' OR Forbidden_label.label_id IS NULL)',
[10, 20, 10, 20]
),
]
def tearDown(self):
self.testbed.deactivate()
self.mox.UnsetStubs()
self.mox.ResetAll()
def _verifySQL(self, cols, left_joins, where, group_by=None):
for col in cols:
self.assertTrue(sql._IsValidColumnName(col))
for join_str, _ in left_joins:
self.assertTrue(sql._IsValidJoin(join_str))
for where_str, _ in where:
self.assertTrue(sql._IsValidWhereCond(where_str))
if group_by:
for groupby_str in group_by:
self.assertTrue(sql._IsValidGroupByTerm(groupby_str))
def testQueryIssueSnapshots_InvalidGroupBy(self):
"""Make sure the `group_by` argument is checked."""
project = fake.Project(project_id=789)
perms = permissions.USER_PERMISSIONSET
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.mox.ReplayAll()
with self.assertRaises(ValueError):
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='rutabaga', label_prefix='rutabaga')
self.mox.VerifyAll()
def testQueryIssueSnapshots_NoLabelPrefix(self):
"""Make sure the `label_prefix` argument is required."""
project = fake.Project(project_id=789)
perms = permissions.USER_PERMISSIONSET
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.mox.ReplayAll()
with self.assertRaises(ValueError):
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='label')
self.mox.VerifyAll()
def testQueryIssueSnapshots_Impossible(self):
"""We give an error message when a query could never have results."""
project = fake.Project(project_id=789)
perms = permissions.USER_PERMISSIONSET
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndRaise(ast2select.NoPossibleResults())
self.mox.ReplayAll()
total, errors, limit_reached = self.services.chart.QueryIssueSnapshots(
self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, query='prefix=')
self.mox.VerifyAll()
self.assertEqual({}, total)
self.assertEqual(['Invalid query.'], errors)
self.assertFalse(limit_reached)
def testQueryIssueSnapshots_Components(self):
"""Test a burndown query from a regular user grouping by component."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'Comp.path',
'COUNT(IssueSnapshot.issue_id)'
]
left_joins = self.defaultLeftJoins + [
('IssueSnapshot2Component AS Is2c'
' ON Is2c.issuesnapshot_id = IssueSnapshot.id', []),
('ComponentDef AS Comp ON Comp.id = Is2c.component_id', [])
]
where = self.defaultWheres
group_by = ['Comp.path']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='component')
self.mox.VerifyAll()
def testQueryIssueSnapshots_Labels(self):
"""Test a burndown query from a regular user grouping by label."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'Lab.label',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = self.defaultLeftJoins + [
('IssueSnapshot2Label AS Is2l'
' ON Is2l.issuesnapshot_id = IssueSnapshot.id', []),
('LabelDef AS Lab ON Lab.id = Is2l.label_id', [])
]
where = self.defaultWheres + [
('LOWER(Lab.label) LIKE %s', ['foo-%']),
]
group_by = ['Lab.label']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='label', label_prefix='Foo')
self.mox.VerifyAll()
def testQueryIssueSnapshots_Open(self):
"""Test a burndown query from a regular user grouping
by status is open or closed."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'IssueSnapshot.is_open',
'COUNT(IssueSnapshot.issue_id) AS issue_count',
]
left_joins = self.defaultLeftJoins
where = self.defaultWheres
group_by = ['IssueSnapshot.is_open']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='open')
self.mox.VerifyAll()
def testQueryIssueSnapshots_Status(self):
"""Test a burndown query from a regular user grouping by open status."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'Stats.status',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = self.defaultLeftJoins + [
('StatusDef AS Stats ON ' \
'Stats.id = IssueSnapshot.status_id', [])
]
where = self.defaultWheres
group_by = ['Stats.status']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='status')
self.mox.VerifyAll()
def testQueryIssueSnapshots_Hotlist(self):
"""Test a QueryIssueSnapshots when a hotlist is passed."""
hotlist = fake.Hotlist('hotlist_rutabaga', 19191)
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'IssueSnapshot.issue_id',
]
left_joins = self.defaultLeftJoins + [
(('IssueSnapshot2Hotlist AS Is2h'
' ON Is2h.issuesnapshot_id = IssueSnapshot.id'
' AND Is2h.hotlist_id = %s'), [hotlist.hotlist_id]),
]
where = self.defaultWheres + [
('Is2h.hotlist_id = %s', [hotlist.hotlist_id]),
]
group_by = []
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, hotlist=hotlist)
self.mox.VerifyAll()
def testQueryIssueSnapshots_Owner(self):
"""Test a burndown query from a regular user grouping by owner."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'IssueSnapshot.owner_id',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = self.defaultLeftJoins
where = self.defaultWheres
group_by = ['IssueSnapshot.owner_id']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='owner')
self.mox.VerifyAll()
def testQueryIssueSnapshots_NoGroupBy(self):
"""Test a burndown query from a regular user with no grouping."""
project = fake.Project(project_id=789)
perms = permissions.PermissionSet(['BarPerm'])
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([91, 81])
cols = [
'IssueSnapshot.issue_id',
]
left_joins = self.defaultLeftJoins
where = self.defaultWheres
group_by = None
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by=None, label_prefix='Foo')
self.mox.VerifyAll()
def testQueryIssueSnapshots_LabelsNotLoggedInUser(self):
"""Tests fetching burndown snapshot counts grouped by labels
for a user who is not logged in. Also no restricted labels are
present.
"""
project = fake.Project(project_id=789)
perms = permissions.READ_ONLY_PERMISSIONSET
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, set([]), project,
perms).AndReturn([91, 81])
cols = [
'Lab.label',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = [
('Issue ON IssueSnapshot.issue_id = Issue.id', []),
('Issue2Label AS Forbidden_label'
' ON Issue.id = Forbidden_label.issue_id'
' AND Forbidden_label.label_id IN (%s,%s)', [91, 81]),
('IssueSnapshot2Label AS Is2l'
' ON Is2l.issuesnapshot_id = IssueSnapshot.id', []),
('LabelDef AS Lab ON Lab.id = Is2l.label_id', []),
]
where = [
('IssueSnapshot.period_start <= %s', [1514764800]),
('IssueSnapshot.period_end > %s', [1514764800]),
('Issue.is_spam = %s', [False]),
('Issue.deleted = %s', [False]),
('IssueSnapshot.project_id IN (%s)', [789]),
('Forbidden_label.label_id IS NULL', []),
('LOWER(Lab.label) LIKE %s', ['foo-%']),
]
group_by = ['Lab.label']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=set([]), project=project,
perms=perms, group_by='label', label_prefix='Foo')
self.mox.VerifyAll()
def testQueryIssueSnapshots_NoRestrictedLabels(self):
"""Test a label burndown query when the project has no restricted labels."""
project = fake.Project(project_id=789)
perms = permissions.USER_PERMISSIONSET
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project,
perms).AndReturn([])
cols = [
'Lab.label',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = [
('Issue ON IssueSnapshot.issue_id = Issue.id', []),
('Issue2Cc AS I2cc'
' ON Issue.id = I2cc.issue_id'
' AND I2cc.cc_id IN (%s,%s)', [10, 20]),
('IssueSnapshot2Label AS Is2l'
' ON Is2l.issuesnapshot_id = IssueSnapshot.id', []),
('LabelDef AS Lab ON Lab.id = Is2l.label_id', []),
]
where = [
('IssueSnapshot.period_start <= %s', [1514764800]),
('IssueSnapshot.period_end > %s', [1514764800]),
('Issue.is_spam = %s', [False]),
('Issue.deleted = %s', [False]),
('IssueSnapshot.project_id IN (%s)', [789]),
('(Issue.reporter_id IN (%s,%s)'
' OR Issue.owner_id IN (%s,%s)'
' OR I2cc.cc_id IS NOT NULL)',
[10, 20, 10, 20]
),
('LOWER(Lab.label) LIKE %s', ['foo-%']),
]
group_by = ['Lab.label']
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn(([], [], []))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
self.services.chart.QueryIssueSnapshots(self.cnxn, self.services,
unixtime=1514764800, effective_ids=[10, 20], project=project,
perms=perms, group_by='label', label_prefix='Foo')
self.mox.VerifyAll()
def SetUpStoreIssueSnapshots(self, replace_now=None,
project_id=789, owner_id=111,
component_ids=None, cc_rows=None):
"""Set up all calls to mocks that StoreIssueSnapshots will call."""
now = self.services.chart._currentTime().AndReturn(replace_now or 12345678)
self.services.chart.issuesnapshot_tbl.Update(self.cnxn,
delta={'period_end': now},
where=[('IssueSnapshot.issue_id = %s', [78901]),
('IssueSnapshot.period_end = %s',
[settings.maximum_snapshot_period_end])],
commit=False)
# Shard is 0 because len(shards) = 1 and 1 % 1 = 0.
shard = 0
self.services.chart.issuesnapshot_tbl.InsertRows(self.cnxn,
chart_svc.ISSUESNAPSHOT_COLS[1:],
[(78901, shard, project_id, 1, 111, owner_id, 1,
now, 4294967295, True)],
replace=True, commit=False, return_generated_ids=True).AndReturn([5678])
label_rows = [(5678, 1)]
self.services.chart.issuesnapshot2label_tbl.InsertRows(self.cnxn,
chart_svc.ISSUESNAPSHOT2LABEL_COLS,
label_rows,
replace=True, commit=False)
self.services.chart.issuesnapshot2cc_tbl.InsertRows(
self.cnxn, chart_svc.ISSUESNAPSHOT2CC_COLS,
[(5678, row[1]) for row in cc_rows],
replace=True, commit=False)
component_rows = [(5678, component_id) for component_id in component_ids]
self.services.chart.issuesnapshot2component_tbl.InsertRows(
self.cnxn, chart_svc.ISSUESNAPSHOT2COMPONENT_COLS,
component_rows,
replace=True, commit=False)
# Spacing of string must match.
self.cnxn.Execute((
'\n INSERT INTO IssueSnapshot2Hotlist '
'(issuesnapshot_id, hotlist_id)\n '
'SELECT %s, hotlist_id FROM Hotlist2Issue '
'WHERE issue_id = %s\n '
), [5678, 78901])
def testStoreIssueSnapshots_NoChange(self):
"""Test that StoreIssueSnapshots inserts and updates previous
issue snapshots correctly."""
now_1 = 1517599888
now_2 = 1517599999
issue = fake.MakeTestIssue(issue_id=78901,
project_id=789, local_id=1, reporter_id=111, owner_id=111,
summary='sum', status='Status1',
labels=['Type-Defect'],
component_ids=[11], assume_stale=False,
opened_timestamp=123456789, modified_timestamp=123456789,
star_count=12, cc_ids=[222, 333], derived_cc_ids=[888])
# Snapshot #1
cc_rows = [(5678, 222), (5678, 333), (5678, 888)]
self.SetUpStoreIssueSnapshots(replace_now=now_1,
component_ids=[11], cc_rows=cc_rows)
# Snapshot #2
self.SetUpStoreIssueSnapshots(replace_now=now_2,
component_ids=[11], cc_rows=cc_rows)
self.mox.ReplayAll()
self.services.chart.StoreIssueSnapshots(self.cnxn, [issue], commit=False)
self.services.chart.StoreIssueSnapshots(self.cnxn, [issue], commit=False)
self.mox.VerifyAll()
def testStoreIssueSnapshots_AllFieldsChanged(self):
"""Test that StoreIssueSnapshots inserts and updates previous
issue snapshots correctly. This tests that all relations (labels,
CCs, and components) are updated."""
now_1 = 1517599888
now_2 = 1517599999
issue_1 = fake.MakeTestIssue(issue_id=78901,
project_id=789, local_id=1, reporter_id=111, owner_id=111,
summary='sum', status='Status1',
labels=['Type-Defect'],
component_ids=[11, 12], assume_stale=False,
opened_timestamp=123456789, modified_timestamp=123456789,
star_count=12, cc_ids=[222, 333], derived_cc_ids=[888])
issue_2 = fake.MakeTestIssue(issue_id=78901,
project_id=123, local_id=1, reporter_id=111, owner_id=222,
summary='sum', status='Status2',
labels=['Type-Enhancement'],
component_ids=[13], assume_stale=False,
opened_timestamp=123456789, modified_timestamp=123456789,
star_count=12, cc_ids=[222, 444], derived_cc_ids=[888, 999])
# Snapshot #1
cc_rows_1 = [(5678, 222), (5678, 333), (5678, 888)]
self.SetUpStoreIssueSnapshots(replace_now=now_1,
component_ids=[11, 12], cc_rows=cc_rows_1)
# Snapshot #2
cc_rows_2 = [(5678, 222), (5678, 444), (5678, 888), (5678, 999)]
self.SetUpStoreIssueSnapshots(replace_now=now_2,
project_id=123, owner_id=222, component_ids=[13],
cc_rows=cc_rows_2)
self.mox.ReplayAll()
self.services.chart.StoreIssueSnapshots(self.cnxn, [issue_1], commit=False)
self.services.chart.StoreIssueSnapshots(self.cnxn, [issue_2], commit=False)
self.mox.VerifyAll()
def testQueryIssueSnapshots_WithQueryStringAndCannedQuery(self):
"""Test the query param is parsed and used."""
project = fake.Project(project_id=789)
perms = permissions.USER_PERMISSIONSET
search_helpers.GetPersonalAtRiskLabelIDs(self.cnxn, None,
self.config_service, [10, 20], project, perms).AndReturn([])
cols = [
'Lab.label',
'COUNT(IssueSnapshot.issue_id)',
]
left_joins = [
('Issue ON IssueSnapshot.issue_id = Issue.id', []),
('Issue2Cc AS I2cc'
' ON Issue.id = I2cc.issue_id'
' AND I2cc.cc_id IN (%s,%s)', [10, 20]),
('IssueSnapshot2Label AS Is2l'
' ON Is2l.issuesnapshot_id = IssueSnapshot.id', []),
('LabelDef AS Lab ON Lab.id = Is2l.label_id', []),
('IssueSnapshot2Label AS Cond0 '
'ON IssueSnapshot.id = Cond0.issuesnapshot_id '
'AND Cond0.label_id = %s', [15]),
]
where = [
('IssueSnapshot.period_start <= %s', [1514764800]),
('IssueSnapshot.period_end > %s', [1514764800]),
('Issue.is_spam = %s', [False]),
('Issue.deleted = %s', [False]),
('IssueSnapshot.project_id IN (%s)', [789]),
('(Issue.reporter_id IN (%s,%s)'
' OR Issue.owner_id IN (%s,%s)'
' OR I2cc.cc_id IS NOT NULL)',
[10, 20, 10, 20]
),
('LOWER(Lab.label) LIKE %s', ['foo-%']),
('Cond0.label_id IS NULL', []),
('IssueSnapshot.is_open = %s', [True]),
]
group_by = ['Lab.label']
query_left_joins = [(
'IssueSnapshot2Label AS Cond0 '
'ON IssueSnapshot.id = Cond0.issuesnapshot_id '
'AND Cond0.label_id = %s', [15])]
query_where = [
('Cond0.label_id IS NULL', []),
('IssueSnapshot.is_open = %s', [True]),
]
unsupported_field_names = ['ownerbouncing']
unsupported_conds = [
ast_pb2.Condition(op=ast_pb2.QueryOp(1), field_defs=[
tracker_pb2.FieldDef(field_name='ownerbouncing',
field_type=tracker_pb2.FieldTypes.BOOL_TYPE),
])
]
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols, where,
left_joins, group_by, shard_id=0)
self.services.chart._QueryToWhere(mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg(),
mox.IgnoreArg()).AndReturn((query_left_joins, query_where,
unsupported_conds))
self.cnxn.Execute(stmt, stmt_args, shard_id=0).AndReturn([])
self._verifySQL(cols, left_joins, where, group_by)
self.mox.ReplayAll()
_, unsupported, limit_reached = self.services.chart.QueryIssueSnapshots(
self.cnxn, self.services, unixtime=1514764800,
effective_ids=[10, 20], project=project, perms=perms,
group_by='label', label_prefix='Foo',
query='-label:Performance%20is:ownerbouncing', canned_query='is:open')
self.mox.VerifyAll()
self.assertEqual(unsupported_field_names, unsupported)
self.assertFalse(limit_reached)
def testQueryToWhere_AddsShardId(self):
"""Test that shards are handled correctly."""
cols = []
where = []
joins = []
group_by = []
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols=cols,
where=where, joins=joins, group_by=group_by, shard_id=9)
self.assertEqual(stmt, ('SELECT COUNT(results.issue_id) '
'FROM (SELECT DISTINCT FROM IssueSnapshot\n'
'WHERE IssueSnapshot.shard = %s\nLIMIT 10000) AS results'))
self.assertEqual(stmt_args, [9])
# Test that shard_id is still correct on second invocation.
stmt, stmt_args = self.services.chart._BuildSnapshotQuery(cols=cols,
where=where, joins=joins, group_by=group_by, shard_id=8)
self.assertEqual(stmt, ('SELECT COUNT(results.issue_id) '
'FROM (SELECT DISTINCT FROM IssueSnapshot\n'
'WHERE IssueSnapshot.shard = %s\nLIMIT 10000) AS results'))
self.assertEqual(stmt_args, [8])
# Test no parameters were modified.
self.assertEqual(cols, [])
self.assertEqual(where, [])
self.assertEqual(joins, [])
self.assertEqual(group_by, [])