#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys
import unittest

_THIS_DIR = os.path.abspath(os.path.dirname(__file__))
# The repo's root directory.
_ROOT_DIR = os.path.abspath(os.path.join(_THIS_DIR, "..", ".."))

# Add the repo's root directory for clearer imports.
sys.path.insert(0, _ROOT_DIR)

import metadata.validate


class ValidateTest(unittest.TestCase):
  def test_validate_file(self):
    # Validate a valid file (no errors or warnings).
    test_filepath = os.path.join(_THIS_DIR, "data",
                                 "README.chromium.test.multi-valid")
    results = metadata.validate.validate_file(
        filepath=test_filepath,
        repo_root_dir=_THIS_DIR,
    )
    self.assertEqual(len(results), 0)

    # Validate an invalid file (both errors and warnings).
    test_filepath = os.path.join(_THIS_DIR, "data",
                                 "README.chromium.test.multi-invalid")
    results = metadata.validate.validate_file(
        filepath=test_filepath,
        repo_root_dir=_THIS_DIR,
    )
    self.assertEqual(len(results), 11)
    error_count = 0
    warning_count = 0
    for result in results:
      if result.is_fatal():
        error_count += 1
      else:
        warning_count += 1
    self.assertEqual(error_count, 9)
    self.assertEqual(warning_count, 2)

  def test_check_file(self):
    # Check a valid file (no errors or warnings).
    test_filepath = os.path.join(_THIS_DIR, "data",
                                 "README.chromium.test.multi-valid")
    errors, warnings = metadata.validate.check_file(
        filepath=test_filepath,
        repo_root_dir=_THIS_DIR,
    )
    self.assertEqual(len(errors), 0)
    self.assertEqual(len(warnings), 0)

    # Check an invalid file (both errors and warnings).
    test_filepath = os.path.join(_THIS_DIR, "data",
                                 "README.chromium.test.multi-invalid")
    errors, warnings = metadata.validate.check_file(
        filepath=test_filepath,
        repo_root_dir=_THIS_DIR,
    )
    # TODO(aredulla): update this test once validation errors can be returned
    #                 as errors.
    # self.assertEqual(len(errors), 9)
    # self.assertEqual(len(warnings), 2)
    self.assertEqual(len(errors), 0)
    self.assertEqual(len(warnings), 11)


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