import os
import utils
from .template_projects.c import spam_c_template
from .template_projects import TemplateProject

subdir_package_project = TemplateProject()

subdir_package_project.files['src/spam/spam.c'] = spam_c_template

subdir_package_project.files['src/spam/setup.py'] = r'''
from setuptools import Extension, setup

setup(
    name="spam",
    ext_modules=[Extension('spam', sources=['spam.c'])],
    version="0.1.0",
)
'''

subdir_package_project.files['src/spam/test/run_tests.py'] = r'''
print('run_tests.py executed!')
'''

subdir_package_project.files['bin/before_build.py'] = r'''
print('before_build.py executed!')
'''


def test(capfd, tmpdir):
    project_dir = str(tmpdir)
    subdir_package_project.generate(project_dir)

    package_dir = os.path.join(project_dir, 'src', 'spam')
    # build the wheels
    actual_wheels = utils.cibuildwheel_run(project_dir, package_dir=package_dir, add_env={
        'CIBW_BEFORE_BUILD': 'python {project}/bin/before_build.py',
        'CIBW_TEST_COMMAND': 'python {package}/test/run_tests.py',
        # this shouldn't depend on the version of python, so build only
        # CPython 3.6
        'CIBW_BUILD': 'cp36-*',
    })

    # check that the expected wheels are produced
    expected_wheels = [w for w in utils.expected_wheels('spam', '0.1.0')
                       if 'cp36' in w]
    assert set(actual_wheels) == set(expected_wheels)

    captured = capfd.readouterr()
    assert "before_build.py executed!" in captured.out
    assert "run_tests.py executed!" in captured.out