# Copyright (c) Microsoft Corporation. All rights reserved.

# Licensed under the MIT License.

import importlib

import sys

import create_venv

import pytest

def test_venv_not_installed():

importlib.reload(create_venv)

create_venv.is_installed = lambda module: module != "venv"

with pytest.raises(create_venv.VenvError) as e:

create_venv.main()

assert str(e.value) == "CREATE_VENV.VENV_NOT_FOUND"

def test_pip_not_installed():

importlib.reload(create_venv)

create_venv.venv_exists = lambda _n: True

create_venv.is_installed = lambda module: module != "pip"

create_venv.run_process = lambda _args, _error_message: None

with pytest.raises(create_venv.VenvError) as e:

create_venv.main(["--install"])

assert str(e.value) == "CREATE_VENV.PIP_NOT_FOUND"

@pytest.mark.parametrize("env_exists", [True, False])

@pytest.mark.parametrize("git_ignore", [True, False])

@pytest.mark.parametrize("install", [True, False])

def test_create_env(env_exists, git_ignore, install):

importlib.reload(create_venv)

create_venv.is_installed = lambda _x: True

create_venv.venv_exists = lambda _n: env_exists

install_packages_called = False

def install_packages(_name):

nonlocal install_packages_called

install_packages_called = True

create_venv.install_packages = install_packages

run_process_called = False

def run_process(args, error_message):

nonlocal run_process_called

run_process_called = True

if not env_exists:

assert args == [sys.executable, "-m", "venv", create_venv.VENV_NAME]

assert error_message == "CREATE_VENV.VENV_FAILED_CREATION"

create_venv.run_process = run_process

add_gitignore_called = False

def add_gitignore(_name):

nonlocal add_gitignore_called

add_gitignore_called = True

create_venv.add_gitignore = add_gitignore

args = []

if git_ignore:

args.append("--git-ignore")

if install:

args.append("--install")

create_venv.main(args)

assert install_packages_called == install

# run_process is called when the venv does not exist

assert run_process_called != env_exists

# add_gitignore is called when new venv is created and git_ignore is True

assert add_gitignore_called == (not env_exists and git_ignore)

@pytest.mark.parametrize("install_type", ["requirements", "pyproject"])

def test_install_packages(install_type):

importlib.reload(create_venv)

create_venv.is_installed = lambda _x: True

create_venv.file_exists = lambda x: install_type in x

pip_upgraded = False

installing = None

def run_process(args, error_message):

nonlocal pip_upgraded, installing

if args[1:] == ["-m", "pip", "install", "--upgrade", "pip"]:

pip_upgraded = True

assert error_message == "CREATE_VENV.PIP_UPGRADE_FAILED"

elif args[1:-1] == ["-m", "pip", "install", "-r"]:

installing = "requirements"

assert error_message == "CREATE_VENV.PIP_FAILED_INSTALL_REQUIREMENTS"

elif args[1:] == ["-m", "pip", "install", "-e", ".[extras]"]:

installing = "pyproject"

assert error_message == "CREATE_VENV.PIP_FAILED_INSTALL_PYPROJECT"

create_venv.run_process = run_process

create_venv.main(["--install"])

assert pip_upgraded

assert installing == install_type