#!/usr/bin/env vpython3

# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests of BundledTestRunner."""

import unittest

from bundled_test_runner import _BundledTestRunner, TestCase

# Test names should be self-explained, no point of adding function docstring.
# pylint: disable=missing-function-docstring, protected-access


class BundledTestRunnerTests(unittest.TestCase):
    """Test class."""

    @staticmethod
    def _default_test_case() -> TestCase:
        return TestCase(
            package='fuchsia-pkg://fuchsia.com/dart_runner_tests#meta/' +
            'dart_runner_tests.cm')

    def test_resolve_test_package_into_far(self):
        runner = _BundledTestRunner(
            'out/put', [BundledTestRunnerTests._default_test_case()],
            'target-id', None, '/tmp/log', [], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': 'out/put/dart_runner_tests.far'})

    def test_dedupe_fars(self):
        runner = _BundledTestRunner(
            'out/put', [BundledTestRunnerTests._default_test_case()],
            'target-id', None, '/tmp/log', ['dart_runner_tests.far'], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': 'out/put/dart_runner_tests.far'})

    def test_absolute_path(self):
        runner = _BundledTestRunner(
            'out/put', [BundledTestRunnerTests._default_test_case()],
            'target-id', None, '/tmp/log',
            ['/tmp/packages/dart_runner_tests.far'], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': '/tmp/packages/dart_runner_tests.far'})

    def test_different_cm(self):
        runner = _BundledTestRunner('out/put', [
            TestCase(
                package=
                'fuchsia-pkg://fuchsia.com/dart_runner_tests#meta/meta.cm')
        ], 'target-id', None, '/tmp/log', [], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': 'out/put/dart_runner_tests.far'})

    def test_join_relative_path(self):
        runner = _BundledTestRunner(
            'out/default', [BundledTestRunnerTests._default_test_case()],
            'target-id', None, '/tmp/log', ['gen/dart_runner_tests.far'], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': 'out/default/gen/dart_runner_tests.far'})

    def test_current_dir_as_out_dir(self):
        runner = _BundledTestRunner(
            '.', [BundledTestRunnerTests._default_test_case()], 'target-id',
            None, '/tmp/log', ['out/default/bin/dart_runner_tests.far'], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': './out/default/bin/dart_runner_tests.far'})

    def test_empty_out_dir(self):
        runner = _BundledTestRunner(
            '', [BundledTestRunnerTests._default_test_case()], 'target-id',
            None, '/tmp/log', ['out/default/bin/dart_runner_tests.far'], None)
        self.assertEqual(
            runner._package_deps,
            {'dart_runner_tests': 'out/default/bin/dart_runner_tests.far'})

    def test_none_out_dir(self):
        with self.assertRaises(TypeError):
            _BundledTestRunner(None,
                               [BundledTestRunnerTests._default_test_case()],
                               'target-id', None, '/tmp/log',
                               ['out/default/bin/dart_runner_tests.far'], None)

    def test_conflict_fars(self):
        with self.assertRaises(AssertionError):
            _BundledTestRunner('',
                               [BundledTestRunnerTests._default_test_case()],
                               'target-id', None, '/tmp/log', [
                                   'out/default/bin/dart_runner_tests.far',
                                   'not/out/default/bin/dart_runner_tests.far'
                               ], None)


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