metadata/utils/find_tests.py

# Lint as: python3
# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import glob
import os

# Relative to .star root (in autotest/files).
OUTPUT_FILE = "metadata/tests.star"

HEADER = """# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

TESTS = []
def _append_tests(tests):
    for t in tests:
        TESTS.append(t)
"""

TEMPLATE = """
load("//{path}", {unique_name} = "TESTS")
_append_tests({unique_name})
"""

def main():
  """Main entry point of this script."""
  root_dir = os.path.realpath(
      os.path.join(os.path.realpath(__file__), "../../.."))
  os.chdir(root_dir)
  output = HEADER
  for test_type in ["client", "server"]:
    for path in glob.iglob(test_type + "/site_tests/*/*.star"):
      head, file_name = os.path.split(path)
      head, test_folder = os.path.split(head)
      unique_name = "_".join(
          [test_type, test_folder, file_name[:-len(".star")]])
      output += TEMPLATE.format(path=path, unique_name=unique_name)

  with open(OUTPUT_FILE, 'w') as fh:
    fh.write(output)
  print("Wrote output to", OUTPUT_FILE)


if __name__ == '__main__':
  main()