Add a pytest rules

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )

.bazelrc

@@ -3,7 +3,7 @@
 # This lets us glob() up all the files inside the examples to make them inputs to tests
 # (Note, we cannot use `common --deleted_packages` because the bazel version command doesn't support it)
 # To update these lines, run tools/bazel_integration_test/update_deleted_packages.sh
-build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
-query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import
+build --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
+query --deleted_packages=examples/legacy_pip_import/boto,examples/legacy_pip_import/extras,examples/legacy_pip_import/helloworld,examples/pip_install,examples/pip_parse,examples/py_import,examples/pytest,examples/wheel,examples/wheel/lib
 
 test --test_output=errors

examples/BUILD

@@ -36,3 +36,8 @@ bazel_integration_test(
  name = "py_import_example",
  timeout = "long",
 )
+
+bazel_integration_test(
+ name="pytest_example",
+ timeout = "long"
+)

examples/pytest/BUILD

@@ -0,0 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_binary",
+ "py_library", "py_test", "py_test_suite", "pytest_test")
+load("@pip//:requirements.bzl", "requirement")
+
+TEST_DEPS = [
+ requirement("pluggy"),
+ requirement("py"),
+ requirement("pytest"),
+ requirement("pytest-instafail"),
+ requirement("pytest-trio"),
+ requirement("pytest-mock"),
+]
+
+py_library(
+ name="lib",
+ srcs="main.py"
+)
+
+py_test_suite(
+ name="unit",
+ size="small",
+ srcs="main_tests.py",
+ args=[
+ "--instafail",
+ ],
+ deps=[
+ ":lib",
+ ] + TEST_DEPS,
+)
+
+

examples/pytest/WORKSPACE

@@ -0,0 +1,15 @@
+workspace(name="example_repo")
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+http_archive(
+ name="rules_python",
+ sha256="778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
+ url="https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
+)
+
+load("@rules_python//python:pip.bzl", "pip_install", "pip_repositories")
+
+pip_install(
+ requirements="//:requirements.txt",
+)

examples/pytest/main.py

@@ -0,0 +1,2 @@
+def add(first, second):
+ return first + second

examples/pytest/main_tests.py

@@ -0,0 +1,9 @@
+import pytest
+
+from .main import add
+
+
+@pytest.mark.parameterize("first,second,expected", [1, 2, 3])
+def test_add(first, second, expected):
+ result = add(first, second)
+ assert result == expected

python/BUILD

@@ -47,6 +47,8 @@ filegroup(
  "packaging.bzl",
  "pip.bzl",
  "private/reexports.bzl",
+ "pytest.bzl",
+ "suite.bzl",
  "whl.bzl",
  ],
  visibility = ["//:__pkg__"],
@@ -144,4 +146,6 @@ exports_files([
  "packaging.bzl",
  "pip.bzl",
  "whl.bzl",
+ "pytest.bzl",
+ "suite.bzl"
 ])

python/defs.bzl

@@ -22,7 +22,6 @@ migrated to Starlark, their implementations will be moved here.
 load("@bazel_tools//tools/python:srcs_version.bzl", _find_requirements = "find_requirements")
 load("@bazel_tools//tools/python:toolchain.bzl", _py_runtime_pair = "py_runtime_pair")
 load(":private/reexports.bzl", "internal_PyInfo", "internal_PyRuntimeInfo")
-
 # Exports of native-defined providers.
 
 PyInfo = internal_PyInfo

python/pytest.bzl

@@ -0,0 +1,81 @@
+load("@rules_python//python:defs.bzl", "PyInfo", "py_test")
+
+def _stringify(paths):
+ return repr(paths)
+
+def _pytest_runner_impl(ctx):
+ if len(ctx.attr.srcs) == 0:
+ fail("No test files specified.")
+
+ expanded_args = [ctx.expand_location(arg, ctx.attr.data) for arg in ctx.attr.args]
+
+ runner = ctx.actions.declare_file(ctx.attr.name)
+ ctx.actions.write(
+ runner,
+ """
+if __name__ == "__main__":
+ import sys
+ import pytest
+
+ args = ["-ra"] + %s + sys.argv[1:] + %s
+
+ sys.exit(pytest.main(args))""" % (_stringify(expanded_args), _stringify([src.path for src in ctx.files.srcs])),
+ is_executable = True,
+ )
+
+ return [
+ DefaultInfo(
+ files = depset([runner]),
+ runfiles = ctx.runfiles(ctx.files.data),
+ executable = runner,
+ ),
+ ]
+
+_pytest_runner = rule(
+ _pytest_runner_impl,
+ attrs = {
+ "srcs": attr.label_list(
+ allow_files = [".py"],
+ ),
+ "deps": attr.label_list(
+ providers = [
+ PyInfo,
+ ],
+ ),
+ "args": attr.string_list(
+ default = [],
+ ),
+ "data": attr.label_list(
+ allow_empty = True,
+ allow_files = True,
+ ),
+ "python_version": attr.string(
+ values = ["PY2", "PY3"],
+ default = "PY3",
+ ),
+ },
+)
+
+def pytest_test(name, srcs, deps = None, args = None, data = None, python_version = None, **kwargs):
+ runner_target = "%s-runner.py" % name
+
+ _pytest_runner(
+ name = runner_target,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ args = args,
+ data = data,
+ python_version = python_version,
+ )
+
+ py_test(
+ name = name,
+ python_version = python_version,
+ srcs = srcs + [runner_target],
+ deps = deps,
+ main = runner_target,
+ legacy_create_init = False,
+ imports = ["."],
+ **kwargs
+ )

python/suite.bzl

@@ -0,0 +1,37 @@
+# load("@rules_python//python:defs.bzl", "py_library")
+load(":pytest.bzl", "pytest_test")
+
+def _is_test(file):
+ return file.startswith("test_") or file.endswith("_tests.py")
+
+def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
+ library_name = "%s-test-lib" % name
+
+ py_library(
+ name = library_name,
+ testonly = True,
+ srcs = srcs,
+ deps = deps,
+ imports = imports,
+ )
+
+ tests = []
+ for src in srcs:
+ if _is_test(src):
+ test_name = "%s-%s" % (name, src)
+
+ tests.append(test_name)
+
+ pytest_test(
+ name = test_name,
+ size = size,
+ srcs = [src],
+ deps = [library_name],
+ python_version = python_version,
+ **kwargs
+ )
+ native.test_suite(
+ name = name,
+ tests = tests,
+ visibility = visibility,
+ )