blob: d54353aca4495a06c2bf1fd2844b44a02b709b55 [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 grid_view_helpers classes and functions."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import unittest
from framework import framework_constants
from framework import framework_views
from framework import grid_view_helpers
from mrproto import tracker_pb2
from testing import fake
from tracker import tracker_bizobj
class GridViewHelpersTest(unittest.TestCase):
def setUp(self):
self.default_cols = 'a b c'
self.builtin_cols = 'a b x y z'
self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789)
self.art1 = fake.MakeTestIssue(
789, 1, 'a summary', '', 0, derived_owner_id=111, star_count=12,
derived_labels='Priority-Medium Hot Mstone-1 Mstone-2',
derived_status='Overdue')
self.art2 = fake.MakeTestIssue(
789, 1, 'a summary', 'New', 111, star_count=12, merged_into=200001,
labels='Priority-Medium Type-DEFECT Hot Mstone-1 Mstone-2')
self.users_by_id = {
111: framework_views.StuffUserView(111, 'foo@example.com', True),
}
def testSortGridHeadings(self):
config = fake.MakeTestConfig(
789, labels=('Priority-High Priority-Medium Priority-Low Hot Cold '
'Milestone-Near Milestone-Far '
'Day-Sun Day-Mon Day-Tue Day-Wed Day-Thu Day-Fri Day-Sat'),
statuses=('New Accepted Started Fixed WontFix Invalid Duplicate'))
config.field_defs = [
tracker_pb2.FieldDef(field_id=1, project_id=789, field_name='Day',
field_type=tracker_pb2.FieldTypes.ENUM_TYPE)]
asc_accessors = {
'id': 'some function that is not called',
'reporter': 'some function that is not called',
'opened': 'some function that is not called',
'modified': 'some function that is not called',
}
# Verify that status headings are sorted according to the status
# values defined in the config.
col_name = 'status'
headings = ['Duplicate', 'Limbo', 'New', 'OnHold', 'Accepted', 'Fixed']
sorted_headings = grid_view_helpers.SortGridHeadings(
col_name, headings, self.users_by_id, config, asc_accessors)
self.assertEqual(
sorted_headings,
['New', 'Accepted', 'Fixed', 'Duplicate', 'Limbo', 'OnHold'])
# Verify that special columns are sorted alphabetically or numerically.
col_name = 'id'
headings = [1, 2, 5, 3, 4]
sorted_headings = grid_view_helpers.SortGridHeadings(
col_name, headings, self.users_by_id, config, asc_accessors)
self.assertEqual(sorted_headings,
[1, 2, 3, 4, 5])
# Verify that label value headings are sorted according to the labels
# values defined in the config.
col_name = 'priority'
headings = ['Medium', 'High', 'Low', 'dont-care']
sorted_headings = grid_view_helpers.SortGridHeadings(
col_name, headings, self.users_by_id, config, asc_accessors)
self.assertEqual(sorted_headings,
['High', 'Medium', 'Low', 'dont-care'])
# Verify that enum headings are sorted according to the labels
# values defined in the config.
col_name = 'day'
headings = ['Tue', 'Fri', 'Sun', 'Dogday', 'Wed', 'Caturday', 'Low']
sorted_headings = grid_view_helpers.SortGridHeadings(
col_name, headings, self.users_by_id, config, asc_accessors)
self.assertEqual(sorted_headings,
['Sun', 'Tue', 'Wed', 'Fri',
'Caturday', 'Dogday', 'Low'])
def testGetArtifactAttr_Explicit(self):
label_values = grid_view_helpers.MakeLabelValuesDict(self.art2)
id_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'id', self.users_by_id, label_values, self.config, {})
self.assertEqual([1], id_vals)
summary_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'summary', self.users_by_id, label_values, self.config, {})
self.assertEqual(['a summary'], summary_vals)
status_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'status', self.users_by_id, label_values, self.config, {})
self.assertEqual(['New'], status_vals)
stars_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'stars', self.users_by_id, label_values, self.config, {})
self.assertEqual([12], stars_vals)
owner_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'owner', self.users_by_id, label_values, self.config, {})
self.assertEqual(['f...@example.com'], owner_vals)
priority_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'priority', self.users_by_id, label_values, self.config, {})
self.assertEqual(['Medium'], priority_vals)
mstone_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'mstone', self.users_by_id, label_values, self.config, {})
self.assertEqual(['1', '2'], mstone_vals)
foo_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'foo', self.users_by_id, label_values, self.config, {})
self.assertEqual([framework_constants.NO_VALUES], foo_vals)
art3 = fake.MakeTestIssue(
987, 5, 'unecessary summary', 'New', 111, star_count=12,
issue_id=200001, project_name='other-project')
related_issues = {200001: art3}
merged_into_vals = grid_view_helpers.GetArtifactAttr(
self.art2, 'mergedinto', self.users_by_id, label_values,
self.config, related_issues)
self.assertEqual(['other-project:5'], merged_into_vals)
def testGetArtifactAttr_Derived(self):
label_values = grid_view_helpers.MakeLabelValuesDict(self.art1)
status_vals = grid_view_helpers.GetArtifactAttr(
self.art1, 'status', self.users_by_id, label_values, self.config, {})
self.assertEqual(['Overdue'], status_vals)
owner_vals = grid_view_helpers.GetArtifactAttr(
self.art1, 'owner', self.users_by_id, label_values, self.config, {})
self.assertEqual(['f...@example.com'], owner_vals)
priority_vals = grid_view_helpers.GetArtifactAttr(
self.art1, 'priority', self.users_by_id, label_values, self.config, {})
self.assertEqual(['Medium'], priority_vals)
mstone_vals = grid_view_helpers.GetArtifactAttr(
self.art1, 'mstone', self.users_by_id, label_values, self.config, {})
self.assertEqual(['1', '2'], mstone_vals)
def testMakeLabelValuesDict_Empty(self):
art = fake.MakeTestIssue(
789, 1, 'a summary', '', 0, derived_owner_id=111, star_count=12)
label_values = grid_view_helpers.MakeLabelValuesDict(art)
self.assertEqual({}, label_values)
def testMakeLabelValuesDict(self):
art = fake.MakeTestIssue(
789, 1, 'a summary', '', 0, derived_owner_id=111, star_count=12,
labels=['Priority-Medium', 'Hot', 'Mstone-1', 'Mstone-2'])
label_values = grid_view_helpers.MakeLabelValuesDict(art)
self.assertEqual(
{'priority': ['Medium'], 'mstone': ['1', '2']},
label_values)
art = fake.MakeTestIssue(
789, 1, 'a summary', '', 0, derived_owner_id=111, star_count=12,
labels='Priority-Medium Hot Mstone-1'.split(),
derived_labels=['Mstone-2'])
label_values = grid_view_helpers.MakeLabelValuesDict(art)
self.assertEqual(
{'priority': ['Medium'], 'mstone': ['1', '2']},
label_values)
def testMakeDrillDownSearch(self):
self.assertEqual('-has:milestone ',
grid_view_helpers.MakeDrillDownSearch('milestone', '----'))
self.assertEqual('milestone=22 ',
grid_view_helpers.MakeDrillDownSearch('milestone', '22'))
self.assertEqual(
'owner=a@example.com ',
grid_view_helpers.MakeDrillDownSearch('owner', 'a@example.com'))
def testAnyArtifactHasNoAttr_Empty(self):
artifacts = []
all_label_values = {}
self.assertFalse(grid_view_helpers.AnyArtifactHasNoAttr(
artifacts, 'milestone', self.users_by_id, all_label_values,
self.config, {}))
def testAnyArtifactHasNoAttr(self):
artifacts = [self.art1]
all_label_values = {
self.art1.local_id: grid_view_helpers.MakeLabelValuesDict(self.art1),
}
self.assertFalse(grid_view_helpers.AnyArtifactHasNoAttr(
artifacts, 'mstone', self.users_by_id, all_label_values,
self.config, {}))
self.assertTrue(grid_view_helpers.AnyArtifactHasNoAttr(
artifacts, 'milestone', self.users_by_id, all_label_values,
self.config, {}))
def testGetGridViewData(self):
# TODO(jojwang): write this test
pass
def testPrepareForMakeGridData(self):
# TODO(jojwang): write this test
pass