# 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