Add tests for pip source plugin

87c4fb2a · Chandan Singh · 13c13eca · 87c4fb2a · 87c4fb2a · 87c4fb2a
Commit 87c4fb2a authored 6 years ago by Chandan Singh
--- a/tests/integration/pip.py
+++ b/tests/integration/pip.py
--- a/tests/integration/pip_source.py
+++ b/tests/integration/pip_source.py
+import os
+import pytest
+
+from buildstream import _yaml
+
+from tests.testutils import cli_integration as cli
+from tests.testutils.integration import assert_contains
+
+
+pytestmark = pytest.mark.integration
+
+
+DATA_DIR = os.path.join(
+    os.path.dirname(os.path.realpath(__file__)),
+    "project"
+)
+
+
+@pytest.mark.datafiles(DATA_DIR)
+def test_pip_source(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    checkout = os.path.join(cli.directory, 'checkout')
+    element_path = os.path.join(project, 'elements')
+    element_name = 'pip/hello.bst'
+
+    element = {
+        'kind': 'import',
+        'sources': [
+            {
+                'kind': 'local',
+                'path': 'files/pip-source'
+            },
+            {
+                'kind': 'pip',
+                'python-exe': 'python3',
+                'index-url': 'file://{}'.format(os.path.realpath(os.path.join(project, 'files', 'pypi-repo'))),
+                'requirements-files': ['myreqs.txt'],
+                'packages': ['app2']
+            }
+        ]
+    }
+    os.makedirs(os.path.dirname(os.path.join(element_path, element_name)), exist_ok=True)
+    _yaml.dump(element, os.path.join(element_path, element_name))
+
+    result = cli.run(project=project, args=['track', element_name])
+    assert result.exit_code == 0
+
+    result = cli.run(project=project, args=['build', element_name])
+    assert result.exit_code == 0
+
+    result = cli.run(project=project, args=['checkout', element_name, checkout])
+    assert result.exit_code == 0
+
+    assert_contains(checkout, ['/bin', '/bin/app1', '/bin/app2'])
--- a/tests/integration/project/files/pip-source/myreqs.txt
+++ b/tests/integration/project/files/pip-source/myreqs.txt
+app1
--- a/tests/integration/project/files/pypi-repo/app1/App1-0.1.tar.gz
+++ b/tests/integration/project/files/pypi-repo/app1/App1-0.1.tar.gz
--- a/tests/integration/project/files/pypi-repo/app1/index.html
+++ b/tests/integration/project/files/pypi-repo/app1/index.html
+<html>
+  <head>
+    <title>Links for app1</title>
+  </head>
+  <body>
+    <a href='App1-0.1.tar.gz'>App1-0.1.tar.gz</a><br />
+  </body>
+</html>
--- a/tests/integration/project/files/pypi-repo/app2/App2-0.1.tar.gz
+++ b/tests/integration/project/files/pypi-repo/app2/App2-0.1.tar.gz
--- a/tests/integration/project/files/pypi-repo/app2/index.html
+++ b/tests/integration/project/files/pypi-repo/app2/index.html
+<html>
+  <head>
+    <title>Links for app1</title>
+  </head>
+  <body>
+    <a href='App2-0.1.tar.gz'>App2-0.1.tar.gz</a><br />
+  </body>
+</html>
--- a/tests/sources/pip.py
+++ b/tests/sources/pip.py
+import os
+import pytest
+
+from buildstream._exceptions import ErrorDomain
+from buildstream import _yaml
+from tests.testutils import cli
+
+DATA_DIR = os.path.join(
+    os.path.dirname(os.path.realpath(__file__)),
+    'pip',
+)
+
+
+def generate_project(project_dir, tmpdir):
+    project_file = os.path.join(project_dir, "project.conf")
+    _yaml.dump({'name': 'foo'}, project_file)
+
+
+# Test that without ref, consistency is set appropriately.
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'no-ref'))
+def test_no_ref(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    generate_project(project, tmpdir)
+    assert cli.get_element_state(project, 'target.bst') == 'no reference'
+
+
+# Test that pip is not allowed to be the first source
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'first-source-pip'))
+def test_first_source(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    generate_project(project, tmpdir)
+    result = cli.run(project=project, args=[
+        'show', 'target.bst'
+    ])
+    result.assert_main_error(ErrorDomain.ELEMENT, None)
+
+
+# Test that error is raised when neither packges nor requirements files
+# have been specified
+@pytest.mark.datafiles(os.path.join(DATA_DIR, 'no-packages'))
+def test_no_packages(cli, tmpdir, datafiles):
+    project = os.path.join(datafiles.dirname, datafiles.basename)
+    generate_project(project, tmpdir)
+    result = cli.run(project=project, args=[
+        'show', 'target.bst'
+    ])
+    result.assert_main_error(ErrorDomain.SOURCE, None)
--- a/tests/sources/pip/first-source-pip/target.bst
+++ b/tests/sources/pip/first-source-pip/target.bst
+kind: import
+description: pip should not be allowed to be the first source
+sources:
+- kind: pip
+  python-exe: python3
+  packages:
+  - flake8
--- a/tests/sources/pip/no-packages/file
+++ b/tests/sources/pip/no-packages/file
+Hello World!
--- a/tests/sources/pip/no-packages/target.bst
+++ b/tests/sources/pip/no-packages/target.bst
+kind: import
+description: The kind of this element is irrelevant.
+sources:
+- kind: local
+  path: file
+- kind: pip
+  python-exe: python3
--- a/tests/sources/pip/no-ref/file
+++ b/tests/sources/pip/no-ref/file
+Hello World!
--- a/tests/sources/pip/no-ref/target.bst
+++ b/tests/sources/pip/no-ref/target.bst
+kind: import
+description: The kind of this element is irrelevant.
+sources:
+- kind: local
+  path: file
+- kind: pip
+  python-exe: python3
+  packages:
+  - flake8