80 lines
2.0 KiB
Python
80 lines
2.0 KiB
Python
"""Defines a rule for runtime test targets."""
|
|
|
|
load("//tools:defs.bzl", "go_test")
|
|
|
|
def _runtime_test_impl(ctx):
|
|
# Construct arguments.
|
|
args = [
|
|
"--lang",
|
|
ctx.attr.lang,
|
|
"--image",
|
|
ctx.attr.image,
|
|
]
|
|
if ctx.attr.blacklist_file:
|
|
args += [
|
|
"--blacklist_file",
|
|
ctx.files.blacklist_file[0].short_path,
|
|
]
|
|
|
|
# Build a runner.
|
|
runner = ctx.actions.declare_file("%s-executer" % ctx.label.name)
|
|
runner_content = "\n".join([
|
|
"#!/bin/bash",
|
|
"%s %s\n" % (ctx.files._runner[0].short_path, " ".join(args)),
|
|
])
|
|
ctx.actions.write(runner, runner_content, is_executable = True)
|
|
|
|
# Return the runner.
|
|
return [DefaultInfo(
|
|
executable = runner,
|
|
runfiles = ctx.runfiles(
|
|
files = ctx.files._runner + ctx.files.blacklist_file + ctx.files._proctor,
|
|
collect_default = True,
|
|
collect_data = True,
|
|
),
|
|
)]
|
|
|
|
_runtime_test = rule(
|
|
implementation = _runtime_test_impl,
|
|
attrs = {
|
|
"image": attr.string(
|
|
mandatory = False,
|
|
),
|
|
"lang": attr.string(
|
|
mandatory = True,
|
|
),
|
|
"blacklist_file": attr.label(
|
|
mandatory = False,
|
|
allow_single_file = True,
|
|
),
|
|
"_runner": attr.label(
|
|
default = "//test/runtimes/runner:runner",
|
|
),
|
|
"_proctor": attr.label(
|
|
default = "//test/runtimes/proctor:proctor",
|
|
),
|
|
},
|
|
test = True,
|
|
)
|
|
|
|
def runtime_test(name, **kwargs):
|
|
_runtime_test(
|
|
name = name,
|
|
image = name, # Resolved as images/runtimes/%s.
|
|
tags = [
|
|
"local",
|
|
"manual",
|
|
],
|
|
**kwargs
|
|
)
|
|
|
|
def blacklist_test(name, blacklist_file):
|
|
"""Test that a blacklist parses correctly."""
|
|
go_test(
|
|
name = name + "_blacklist_test",
|
|
library = ":runner",
|
|
srcs = ["blacklist_test.go"],
|
|
args = ["--blacklist_file", "test/runtimes/" + blacklist_file],
|
|
data = [blacklist_file],
|
|
)
|