from unittest import mock

import pytest

from mlflow.utils import PYTHON_VERSION
from mlflow.utils.environment import _PythonEnv


def test_constructor_argument_validation():
    with pytest.raises(TypeError, match="`python` must be a string"):
        _PythonEnv(python=1)

    with pytest.raises(TypeError, match="`build_dependencies` must be a list"):
        _PythonEnv(build_dependencies=0)

    with pytest.raises(TypeError, match="`dependencies` must be a list"):
        _PythonEnv(dependencies=0)


def test_to_yaml(tmp_path):
    yaml_path = tmp_path / "python_env.yaml"
    _PythonEnv(PYTHON_VERSION, ["a"], ["b"]).to_yaml(yaml_path)
    expected_content = f"""
python: {PYTHON_VERSION}
build_dependencies:
- a
dependencies:
- b
""".lstrip()
    assert yaml_path.read_text() == expected_content


def test_from_yaml(tmp_path):
    content = f"""
python: {PYTHON_VERSION}
build_dependencies:
- a
- b
dependencies:
- c
- d
"""
    yaml_path = tmp_path / "test.yaml"
    yaml_path.write_text(content)
    python_env = _PythonEnv.from_yaml(yaml_path)
    assert python_env.python == PYTHON_VERSION
    assert python_env.build_dependencies == ["a", "b"]
    assert python_env.dependencies == ["c", "d"]


def test_from_conda_yaml(tmp_path):
    content = f"""
name: example
channels:
  - conda-forge
dependencies:
  - python={PYTHON_VERSION}
  - pip
  - pip:
    - a
    - b
"""
    yaml_path = tmp_path / "conda.yaml"
    yaml_path.write_text(content)
    python_env = _PythonEnv.from_conda_yaml(yaml_path)
    assert python_env.python == PYTHON_VERSION
    assert python_env.build_dependencies == ["pip"]
    assert python_env.dependencies == ["a", "b"]


def test_from_conda_yaml_build_dependencies(tmp_path):
    content = f"""
name: example
channels:
  - conda-forge
dependencies:
  - python={PYTHON_VERSION}
  - pip=1.2.3
  - wheel==4.5.6
  - setuptools<=7.8.9
  - pip:
    - a
    - b
"""
    yaml_path = tmp_path / "conda.yaml"
    yaml_path.write_text(content)
    python_env = _PythonEnv.from_conda_yaml(yaml_path)
    assert python_env.python == PYTHON_VERSION
    assert python_env.build_dependencies == ["pip==1.2.3", "wheel==4.5.6", "setuptools<=7.8.9"]
    assert python_env.dependencies == ["a", "b"]


def test_from_conda_yaml_use_current_python_version_when_no_python_spec_in_conda_yaml(tmp_path):
    content = """
name: example
channels:
  - conda-forge
dependencies:
  - pip
  - pip:
    - a
    - b
"""
    yaml_path = tmp_path / "conda.yaml"
    yaml_path.write_text(content)
    assert _PythonEnv.from_conda_yaml(yaml_path).python == PYTHON_VERSION


def test_from_conda_yaml_invalid_python_comparator(tmp_path):
    content = f"""
name: example
channels:
  - conda-forge
dependencies:
  - python<{PYTHON_VERSION}
  - pip:
    - a
    - b
"""
    yaml_path = tmp_path / "conda.yaml"
    yaml_path.write_text(content)
    with pytest.raises(Exception, match="Invalid version comparator for python"):
        _PythonEnv.from_conda_yaml(yaml_path)


def test_from_conda_yaml_conda_dependencies_warning(tmp_path):
    content = f"""
name: example
channels:
  - conda-forge
dependencies:
  - python={PYTHON_VERSION}
  - foo
  - bar
  - pip:
    - a
"""
    yaml_path = tmp_path / "conda.yaml"
    yaml_path.write_text(content)
    with mock.patch("mlflow.utils.environment._logger.warning") as mock_warning:
        _PythonEnv.from_conda_yaml(yaml_path)
        mock_warning.assert_called_with(
            "The following conda dependencies will not be installed "
            "in the resulting environment: %s",
            ["foo", "bar"],
        )