#!/usr/bin/env vpython3
# coding=utf-8
# 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 git_find_releases.py."""

from io import StringIO
import logging
import os
import sys
import unittest
from unittest import mock

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

import gclient_utils
import git_find_releases


@unittest.skipIf(
    gclient_utils.IsEnvCog(), "not supported in non-git environment"
)
class TestGitFindReleases(unittest.TestCase):
    @mock.patch("sys.stdout", StringIO())
    @mock.patch("git_common.run", return_value="")
    def test_invalid_commit(self, git_run):
        result = git_find_releases.main(["foo"])
        self.assertEqual(1, result)
        self.assertEqual("foo not found", sys.stdout.getvalue().strip())
        git_run.assert_called_once_with(
            "name-rev", "--tags", "--name-only", "foo"
        )

    @mock.patch("sys.stdout", StringIO())
    @mock.patch("git_common.run")
    def test_no_merge(self, git_run):
        def git_run_function(*args):
            assert len(args) > 1
            if args[0] == "name-rev" and args[1] == "--tags":
                return "undefined"

            if args[0] == "name-rev" and args[1] == "--refs":
                return "1.0.0"

            if args[0] == "log":
                return ""
            assert False, "Unexpected arguments for git.run"

        git_run.side_effect = git_run_function
        result = git_find_releases.main(["foo"])
        self.assertEqual(0, result)
        stdout = sys.stdout.getvalue().strip()
        self.assertIn("commit foo was", stdout)
        self.assertIn("No merges found", stdout)
        self.assertEqual(3, git_run.call_count)


if __name__ == "__main__":
    logging.basicConfig(
        level=logging.DEBUG if "-v" in sys.argv else logging.ERROR
    )
    unittest.main()