| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| |
| # Pylint doesn't play well with fixtures and dependency injection from pytest |
| # pylint: disable=redefined-outer-name |
| |
| import os |
| |
| import pytest |
| |
| from buildstream import _yaml |
| |
| from buildstream._testing import cli_integration as cli # pylint: disable=unused-import |
| from buildstream._testing.integration import assert_contains |
| from buildstream._testing.integration import integration_cache # pylint: disable=unused-import |
| from buildstream._testing._utils.site import HAVE_SANDBOX |
| |
| from tests.testutils.python_repo import setup_pypi_repo # pylint: disable=unused-import |
| |
| |
| pytestmark = pytest.mark.integration |
| |
| |
| DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "pip") |
| |
| |
| @pytest.mark.datafiles(DATA_DIR) |
| @pytest.mark.skipif(not HAVE_SANDBOX, reason="Only available with a functioning sandbox") |
| def test_pip_build(cli, datafiles): |
| project = str(datafiles) |
| checkout = os.path.join(cli.directory, "checkout") |
| element_path = os.path.join(project, "elements") |
| element_name = "pip/hello.bst" |
| |
| element = { |
| "kind": "pip", |
| "variables": {"pip": "pip3"}, |
| "depends": [{"filename": "base.bst"}], |
| "sources": [ |
| { |
| "kind": "tar", |
| "url": "file://{}/files/piphello.tar.xz".format(project), |
| "ref": "ad96570b552498807abec33c06210bf68378d854ced6753b77916c5ed517610d", |
| } |
| ], |
| } |
| os.makedirs( |
| os.path.dirname(os.path.join(element_path, element_name)), |
| exist_ok=True, |
| ) |
| _yaml.roundtrip_dump(element, os.path.join(element_path, element_name)) |
| |
| result = cli.run(project=project, args=["build", element_name]) |
| assert result.exit_code == 0 |
| |
| result = cli.run( |
| project=project, |
| args=["artifact", "checkout", element_name, "--directory", checkout], |
| ) |
| assert result.exit_code == 0 |
| |
| assert_contains( |
| checkout, |
| [ |
| "/usr", |
| "/usr/lib", |
| "/usr/bin", |
| "/usr/bin/hello", |
| "/usr/lib/python3.6", |
| ], |
| ) |
| |
| |
| # Test running an executable built with pip |
| @pytest.mark.datafiles(DATA_DIR) |
| @pytest.mark.skipif(not HAVE_SANDBOX, reason="Only available with a functioning sandbox") |
| def test_pip_run(cli, datafiles): |
| # Create and build our test element |
| test_pip_build(cli, datafiles) |
| |
| project = str(datafiles) |
| element_name = "pip/hello.bst" |
| |
| result = cli.run(project=project, args=["shell", element_name, "/usr/bin/hello"]) |
| assert result.exit_code == 0 |
| assert result.output == "Hello, world!\n" |
| |
| |
| @pytest.mark.datafiles(DATA_DIR) |
| @pytest.mark.skipif(not HAVE_SANDBOX, reason="Only available with a functioning sandbox") |
| def test_pip_element_should_install_pip_deps(cli, datafiles, setup_pypi_repo): |
| project = str(datafiles) |
| elements_path = os.path.join(project, "elements") |
| element_name = "pip/hello.bst" |
| |
| # check that exotically named packages are imported correctly |
| myreqs_packages = "alohalib" |
| dependencies = [ |
| "app2", |
| "app.3", |
| "app-4", |
| "app_5", |
| "app.no.6", |
| "app-no-7", |
| "app_no_8", |
| ] |
| mock_packages = {myreqs_packages: {package: {} for package in dependencies}} |
| |
| # set up directories |
| pypi_repo = os.path.join(project, "files", "pypi-repo") |
| os.makedirs(pypi_repo, exist_ok=True) |
| os.makedirs( |
| os.path.dirname(os.path.join(elements_path, element_name)), |
| exist_ok=True, |
| ) |
| setup_pypi_repo(mock_packages, pypi_repo) |
| |
| # create pip element |
| element = { |
| "kind": "pip", |
| "variables": {"pip": "pip3"}, |
| "depends": [{"filename": "base.bst"}], |
| "sources": [ |
| { |
| "kind": "tar", |
| "url": "file://{}/files/piphello.tar.xz".format(project), |
| # FIXME: remove hardcoded ref once issue #1010 is closed |
| "ref": "ad96570b552498807abec33c06210bf68378d854ced6753b77916c5ed517610d", |
| }, |
| { |
| "kind": "pip", |
| "url": "file://{}".format(os.path.realpath(pypi_repo)), |
| "packages": [myreqs_packages], |
| }, |
| ], |
| } |
| _yaml.roundtrip_dump(element, os.path.join(elements_path, element_name)) |
| |
| result = cli.run(project=project, args=["source", "track", element_name]) |
| assert result.exit_code == 0 |
| |
| result = cli.run(project=project, args=["build", element_name]) |
| assert result.exit_code == 0 |
| |
| # get installed packages in sandbox |
| installed_packages = set( |
| cli.run(project=project, args=["shell", element_name, "pip3", "freeze"]).output.split("\n") |
| ) |
| # compare with packages that are expected to be installed |
| pip_source_packages = {package.replace("_", "-") + "==0.1" for package in dependencies + [myreqs_packages]} |
| assert pip_source_packages.issubset(installed_packages) |