#!/usr/bin/env vpython3
# Copyright 2020 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 update_annotations_doc.py
"""

import os
import sys
import unittest
from mock import MagicMock

# Mock some imports which aren't necessary during testing.
sys.modules["infra_libs"] = MagicMock()
import update_annotations_doc
import generator_utils

# Absolute path to chrome/src.
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
SRC_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, "../../.."))
TESTS_DIR = os.path.join(SCRIPT_DIR, "test_data")


class UpdateAnnotationsDocTest(unittest.TestCase):
  network_doc_obj = update_annotations_doc.NetworkTrafficAnnotationsDoc(
    "", "", "", "", ""
  )

  def test_create_group_request(self):
    text = "TestGroup"
    req, index = self.network_doc_obj._create_group_or_sender_request(
      text, 0, generator_utils.Placeholder.GROUP
    )

    self.assertEqual(len(text) + 1, index)
    expected_req = [
      {"insertText": {"text": "TestGroup\n", "location": {"index": 0}}},
      {
        "updateParagraphStyle": {
          "fields": "*",
          "range": {"endIndex": 10, "startIndex": 0},
          "paragraphStyle": {
            "spacingMode": "NEVER_COLLAPSE",
            "direction": "LEFT_TO_RIGHT",
            "namedStyleType": "HEADING_1",
            "spaceAbove": {"unit": "PT"},
          },
        }
      },
      {
        "updateTextStyle": {
          "textStyle": {
            "fontSize": {"magnitude": 20, "unit": "PT"},
            "bold": False,
            "weightedFontFamily": {"fontFamily": "Roboto", "weight": 400},
          },
          "range": {"endIndex": 10, "startIndex": 0},
          "fields": "*",
        }
      },
    ]
    self.assertEqual(expected_req, req)

  def test_create_sender_request(self):
    text = "TestSender"
    print(text)
    req, index = self.network_doc_obj._create_group_or_sender_request(
      text, 0, generator_utils.Placeholder.SENDER
    )

    self.assertEqual(len(text) + 1, index)
    expected_req = [
      {"insertText": {"text": "TestSender\n", "location": {"index": 0}}},
      {
        "updateParagraphStyle": {
          "fields": "*",
          "range": {"endIndex": 11, "startIndex": 0},
          "paragraphStyle": {
            "spacingMode": "NEVER_COLLAPSE",
            "direction": "LEFT_TO_RIGHT",
            "namedStyleType": "HEADING_2",
            "spaceAbove": {"unit": "PT"},
          },
        }
      },
      {
        "updateTextStyle": {
          "textStyle": {
            "fontSize": {"magnitude": 14, "unit": "PT"},
            "bold": True,
            "weightedFontFamily": {"fontFamily": "Roboto", "weight": 400},
          },
          "range": {"endIndex": 11, "startIndex": 0},
          "fields": "*",
        }
      },
    ]
    self.assertEqual(expected_req, req)

  def test_create_annotation_request(self):
    traffic_annotation = generator_utils.TrafficAnnotation(
      **{
        "unique_id": "unique_id_A",
        "description": "description_A",
        "trigger": "trigger_A",
        "data": "data_A",
        "settings": "settings_A",
        "policy": "chrome_policy_A",
      }
    )

    req, index = self.network_doc_obj._create_annotation_request(
      traffic_annotation, 0
    )

    self.assertEqual(109, index)
    expected_req = [
      {'insertText': {'text': '\n', 'location': {'index': 0}}},
      {'insertTable': {'rows': 1, 'location': {'index': 0}, 'columns': 2}},
      {'insertText': {'text': 'unique_id_A', 'location': {'index': 4}}},
      {
        'insertText': {
          'text': "description_A\nTrigger: trigger_A\nData: data_A\nSettings: "
          "settings_A\nPolicy: chrome_policy_A",
          'location': {'index': 17},
        }
      },
      {
        'updateTableColumnProperties': {
          'columnIndices': [0],
          'fields': '*',
          'tableColumnProperties': {
            'width': {'magnitude': 153, 'unit': 'PT'},
            'widthType': 'FIXED_WIDTH',
          },
          'tableStartLocation': {'index': 1},
        }
      },
      {
        'updateTableColumnProperties': {
          'columnIndices': [1],
          'fields': '*',
          'tableColumnProperties': {
            'width': {'magnitude': 534, 'unit': 'PT'},
            'widthType': 'FIXED_WIDTH',
          },
          'tableStartLocation': {'index': 1},
        }
      },
      {
        'updateTableCellStyle': {
          'fields': '*',
          'tableCellStyle': {
            'rowSpan': 1,
            'borderBottom': {
              'color': {
                'color': {'rgbColor': {'blue': 1.0, 'green': 1.0, 'red': 1.0}}
              },
              'width': {'unit': 'PT'},
              'dashStyle': 'SOLID',
            },
            'paddingBottom': {'magnitude': 1.44, 'unit': 'PT'},
            'paddingLeft': {'magnitude': 1.44, 'unit': 'PT'},
            'paddingTop': {'magnitude': 1.44, 'unit': 'PT'},
            'borderLeft': {
              'color': {
                'color': {'rgbColor': {'blue': 1.0, 'green': 1.0, 'red': 1.0}}
              },
              'width': {'unit': 'PT'},
              'dashStyle': 'SOLID',
            },
            'columnSpan': 1,
            'backgroundColor': {
              'color': {'rgbColor': {'blue': 1.0, 'green': 1.0, 'red': 1.0}}
            },
            'borderRight': {
              'color': {
                'color': {'rgbColor': {'blue': 1.0, 'green': 1.0, 'red': 1.0}}
              },
              'width': {'unit': 'PT'},
              'dashStyle': 'SOLID',
            },
            'borderTop': {
              'color': {
                'color': {'rgbColor': {'blue': 1.0, 'green': 1.0, 'red': 1.0}}
              },
              'width': {'unit': 'PT'},
              'dashStyle': 'SOLID',
            },
            'paddingRight': {'magnitude': 1.44, 'unit': 'PT'},
          },
          'tableStartLocation': {'index': 1},
        }
      },
      {
        'updateParagraphStyle': {
          'fields': '*',
          'range': {'endIndex': 108, 'startIndex': 4},
          'paragraphStyle': {
            'spacingMode': 'NEVER_COLLAPSE',
            'direction': 'LEFT_TO_RIGHT',
            'spaceBelow': {'magnitude': 4, 'unit': 'PT'},
            'lineSpacing': 100,
            'avoidWidowAndOrphan': False,
            'namedStyleType': 'NORMAL_TEXT',
          },
        }
      },
      {
        'updateTextStyle': {
          'textStyle': {
            'fontSize': {'magnitude': 9, 'unit': 'PT'},
            'bold': False,
            'weightedFontFamily': {'fontFamily': 'Roboto', 'weight': 400},
          },
          'range': {'endIndex': 108, 'startIndex': 4},
          'fields': '*',
        }
      },
    ]
    self.assertEqual(expected_req, req)


if __name__ == "__main__":
  unittest.main()