Python RunRequest Examples

Example #1

def main(argv):
  """Main entry point."""
  if len(argv) > 1:
    raise app.UsageError("Unknown arguments: '{}'.".format(" ".join(argv[1:])))

  channel = grpc.insecure_channel(FLAGS.ledger)
  stub = alice_pb2_grpc.LedgerStub(channel)

  phd_path = pathlib.Path(FLAGS.phd)
  repo = git_repo.PhdRepo(phd_path)

  # TODO(cec): Sanity check values.
  target = FLAGS.target
  bazel_args = FLAGS.bazel_args
  bin_args = FLAGS.bin_args
  timeout_seconds = FLAGS.timeout_seconds

  job_id = stub.Add(
    alice_pb2.RunRequest(
      repo_state=repo.ToRepoState(),
      target=target,
      bazel_args=bazel_args,
      bin_args=bin_args,
      timeout_seconds=timeout_seconds,
    )
  )

  print(f"Started job {job_id.id}")
  while True:
    status = stub.Get(job_id)
    SummarizeJobStatus(status)
    time.sleep(1)
    if status.job_status == alice_pb2.LedgerEntry.COMPLETE:
      sys.exit(status.returncode)

Example #2

def test_BazelClient_Run_stdout(workspace: pathlib.Path, tempdir: pathlib.Path):
  """Check stdout of test target."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  assert process.stdout == "Hello, stdout!\n"

Example #3

def test_BazelClient_Run_returncode(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Check returncode of test target."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  assert process.returncode == 0

Example #4

def test_BazelClient_Run_stderr(workspace: pathlib.Path, tempdir: pathlib.Path):
  """Check stderr of test target."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  # Stderr starts with bazel build log.
  print(process.stderr)
  assert "Hello, stderr!\n" in process.stderr

Example #5

def test_BazelClient_Run_process_id(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Check that process ID is set."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  assert process.pid
  assert process.pid != system.PID

Example #6

def test_BazeClient_run_missing_target(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Check error for missing target."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(
    alice_pb2.RunRequest(target="//:not_a_target", ledger_id=1,)
  )

  process.join()
  assert process.returncode

Example #7

def test_BazelClient_Run_workdir_files(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Check that output files are generated."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  assert (process.workdir / "stdout.txt").is_file()
  assert (process.workdir / "stderr.txt").is_file()
  assert (process.workdir / "returncode.txt").is_file()

Example #8

def test_BazelClient_Run_process_isnt_running(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Check that process isn't running after completed."""
  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(alice_pb2.RunRequest(target="//:hello", ledger_id=1,))

  process.join()
  try:
    os.kill(process.pid, 0)
    test.Fail(
      "os.kill() didn't fail, that means the process is still " "running"
    )
  except ProcessLookupError:
    pass

Example #9

def test_BazelClient_kill_process(
  workspace: pathlib.Path, tempdir: pathlib.Path
):
  """Test that process can be killed."""
  # Create a binary which will never terminate once launched.
  with open(workspace / "BUILD", "a") as f:
    f.write(
      """
cc_binary(
    name = "nonterminating",
    srcs = ["nonterminating.cc"],
)
"""
    )

  with open(workspace / "nonterminating.cc", "w") as f:
    f.write(
      """
int main() {
  while (1) {}
}
"""
    )

  client = bazel.BazelClient(workspace, tempdir)
  process = client.Run(
    alice_pb2.RunRequest(target="//:nonterminating", ledger_id=1,)
  )

  # Sleep for luck ;-)
  time.sleep(3)

  # Send the non-terminating process a kill signal.
  os.kill(process.pid, signal.SIGTERM)
  process.join()
  try:
    os.kill(process.pid, 0)
    test.Fail(
      "os.kill() didn't fail, that means the process is still " "running"
    )
  except ProcessLookupError:
    pass

Example #10

def mock_run_request() -> alice_pb2.RunRequest:
    return alice_pb2.RunRequest(
        repo_state=None,
        target="//:foo",
    )