#!/usr/bin/env python3

# Copyright 2023 the V8 project 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 contextlib
import io
import os
import pathlib
import unittest

from tempfile import TemporaryDirectory
from unittest.mock import patch, mock_open

from download_profiles import main, parse_args, retrieve_version


def mock_version_file(major, minor, build, patch):
  return (
      f'#define V8_MAJOR_VERSION {major}\n'
      f'#define V8_MINOR_VERSION {minor}\n'
      f'#define V8_BUILD_NUMBER {build}\n'
      f'#define V8_PATCH_LEVEL {patch}\n'
  )


class TestDownloadProfiles(unittest.TestCase):

  def _test_cmd(self, cmd, exitcode):
    out = io.StringIO()
    err = io.StringIO()
    with self.assertRaises(SystemExit) as se, \
        contextlib.redirect_stdout(out), \
        contextlib.redirect_stderr(err):
      main(cmd)
    self.assertEqual(se.exception.code, exitcode)
    return out.getvalue(), err.getvalue()

  def test_validate_profiles(self):
    out, err = self._test_cmd(['validate', '--version', '11.1.0.0'], 0)
    self.assertEqual(len(out), 0)
    self.assertEqual(len(err), 0)

  def test_download_profiles(self):
    with TemporaryDirectory() as td, \
        patch('download_profiles.PGO_PROFILE_DIR', pathlib.Path(td)):
      out, err = self._test_cmd(['download', '--version', '11.1.0.0'], 0)
      self.assertEqual(len(out), 0)
      self.assertEqual(len(err), 0)
      self.assertGreater(
          len([f for f in os.listdir(td) if f.endswith('.profile')]), 0)
      with open(pathlib.Path(td) / 'profiles_version') as f:
        self.assertEqual(f.read(), '11.1.0.0')

      # A second download should not be started as profiles exist already
      with patch('download_profiles.call_gsutil') as gsutil:
        out, err = self._test_cmd(['download', '--version', '11.1.0.0'], 0)
        self.assertEqual(
            out,
            'Profiles already downloaded, use --force to overwrite.\n',
        )
        gsutil.assert_not_called()

      # A forced download should always trigger
      with patch('download_profiles.call_gsutil') as gsutil:
        cmd = ['download', '--version', '11.1.0.0', '--force']
        out, err = self._test_cmd(cmd, 0)
        self.assertEqual(len(out), 0)
        self.assertEqual(len(err), 0)
        gsutil.assert_called_once()

  def test_arg_quiet(self):
    with patch(
        'builtins.open',
        new=mock_open(read_data=mock_version_file(11, 9, 0, 0))):
      out, err = self._test_cmd(['download'], 0)
      self.assertGreater(len(out), 0)

      out, err = self._test_cmd(['download', '--quiet'], 0)
      self.assertEqual(len(out), 0)

  def test_invalid_args(self):
    out, err = self._test_cmd(['invalid-action'], 2)
    self.assertEqual(len(out), 0)
    self.assertGreater(len(err), 0)

  def test_invalid_depot_tools_path(self):
    out, err = self._test_cmd(
        ['validate', '--depot-tools', '/no-depot-tools-path'], 3)
    self.assertEqual(len(out), 0)
    self.assertGreater(len(err), 0)

  def test_missing_profiles(self):
    out, err = self._test_cmd(['download', '--version', '0.0.0.42'], 4)
    self.assertEqual(len(out), 0)
    self.assertGreater(len(err), 0)


class TestRetrieveVersion(unittest.TestCase):
  def test_retrieve_valid_version(self):
    with patch(
        'builtins.open',
        new=mock_open(read_data=mock_version_file(11, 4, 1, 0))):
      args = parse_args(['download'])
      version = retrieve_version(args)

    self.assertEqual(version, '11.4.1.0')

  def test_retrieve_parameter_version(self):
    args = parse_args(['download', '--version', '11.1.1.42'])
    version = retrieve_version(args)
    self.assertEqual(version, '11.1.1.42')

  def test_retrieve_untagged_version(self):
    out = io.StringIO()
    with patch(
        'builtins.open',
        new=mock_open(read_data=mock_version_file(11, 4, 0, 0))), \
        contextlib.redirect_stdout(out), \
        self.assertRaises(SystemExit) as se:
      args = parse_args(['download'])
      version = retrieve_version(args)

    self.assertEqual(se.exception.code, 0)
    self.assertEqual(out.getvalue(),
        'The version file specifies 11.4.0.0, which has no profiles.\n')


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