<commit_before>#!/usr/bin/python
"""Run tests in parallel."""

import argparse
import glob
import itertools
import multiprocessing
import sys
import time

import jobset
import watch_dirs

# SimpleConfig: just compile with CONFIG=config, and run the binary to test
class SimpleConfig(object):
  def __init__(self, config):
    self.build_config = config

  def run_command(self, binary):
    return [binary]


# ValgrindConfig: compile with some CONFIG=config, but use valgrind to run
class ValgrindConfig(object):
  def __init__(self, config):
    self.build_config = config

  def run_command(self, binary):
    return ['valgrind', binary]


# SanConfig: compile with CONFIG=config, filter out incompatible binaries
class SanConfig(object):
  def __init__(self, config):
    self.build_config = config

  def run_command(self, binary):
    if '_ssl_' in binary:
      return None
    return [binary]

# different configurations we can run under
_CONFIGS = {
  'dbg': SimpleConfig('dbg'),
  'opt': SimpleConfig('opt'),
  'tsan': SanConfig('tsan'),
  'msan': SanConfig('msan'),
  'asan': SanConfig('asan'),
  'gcov': SimpleConfig('gcov'),
  'valgrind': ValgrindConfig('dbg'),
  }


_DEFAULT = ['dbg', 'opt']
_MAKE_TEST_TARGETS = ['buildtests_c', 'buildtests_cxx']

# parse command line
argp = argparse.ArgumentParser(description='Run grpc tests.')
argp.add_argument('-c', '--config',
                  choices=['all'] + sorted(_CONFIGS.keys()),
                  nargs='+',
                  default=_DEFAULT)
argp.add_argument('-t', '--test-filter', nargs='*', default=['*'])
argp.add_argument('-n', '--runs_per_test', default=1, type=int)
argp.add_argument('-f', '--forever',
                  default=False,
                  action='store_const',
                  const=True)
args = argp.parse_args()

# grab config
run_configs = set(_CONFIGS[cfg]
                  for cfg in itertools.chain.from_iterable(
                      _CONFIGS.iterkeys() if x == 'all' else [x]
                      for x in args.config))
build_configs = set(cfg.build_config for cfg in run_configs)
filters = args.test_filter
runs_per_test = args.runs_per_test
forever = args.forever


def _build_and_run(check_cancelled):
  """Do one pass of building & running tests."""
  # build latest, sharing cpu between the various makes
  if not jobset.run(
      (['make',
        '-j', '%d' % (multiprocessing.cpu_count() + 1),
        'CONFIG=%s' % cfg] + _MAKE_TEST_TARGETS
       for cfg in build_configs),
      check_cancelled, maxjobs=1):
    return 1

  # run all the tests
  if not jobset.run(itertools.ifilter(
      lambda x: x is not None, (
          config.run_command(x)
          for config in run_configs
          for filt in filters
          for x in itertools.chain.from_iterable(itertools.repeat(
              glob.glob('bins/%s/%s_test' % (
                  config.build_config, filt)),
              runs_per_test)))), check_cancelled):
    return 2

  return 0


if forever:
  while True:
    dw = watch_dirs.DirWatcher(['src', 'include', 'test'])
    initial_time = dw.most_recent_change()<commit_msg>Helgrind support for run_tests.py<commit_after>#!/usr/bin/python
"""Run tests in parallel."""

import argparse
import glob
import itertools
import multiprocessing
import sys
import time

import jobset
import watch_dirs

# SimpleConfig: just compile with CONFIG=config, and run the binary to test
class SimpleConfig(object):
  def __init__(self, config):
    self.build_config = config
    self.maxjobs = 32 * multiprocessing.cpu_count()

  def run_command(self, binary):
    return [binary]


# ValgrindConfig: compile with some CONFIG=config, but use valgrind to run
class ValgrindConfig(object):
  def __init__(self, config, tool):
    self.build_config = config
    self.tool = tool
    self.maxjobs = 4 * multiprocessing.cpu_count()

  def run_command(self, binary):
    return ['valgrind', binary, '--tool=%s' % self.tool]


# SanConfig: compile with CONFIG=config, filter out incompatible binaries
class SanConfig(object):
  def __init__(self, config):
    self.build_config = config
    self.maxjobs = 16 * multiprocessing.cpu_count()

  def run_command(self, binary):
    if '_ssl_' in binary:
      return None
    return [binary]

# different configurations we can run under
_CONFIGS = {
  'dbg': SimpleConfig('dbg'),
  'opt': SimpleConfig('opt'),
  'tsan': SanConfig('tsan'),
  'msan': SanConfig('msan'),
  'asan': SanConfig('asan'),
  'gcov': SimpleConfig('gcov'),
  'memcheck': ValgrindConfig('dbg', 'memcheck'),
  'helgrind': ValgrindConfig('dbg', 'helgrind')
  }


_DEFAULT = ['dbg', 'opt']
_MAKE_TEST_TARGETS = ['buildtests_c', 'buildtests_cxx']

# parse command line
argp = argparse.ArgumentParser(description='Run grpc tests.')
argp.add_argument('-c', '--config',
                  choices=['all'] + sorted(_CONFIGS.keys()),
                  nargs='+',
                  default=_DEFAULT)
argp.add_argument('-t', '--test-filter', nargs='*', default=['*'])
argp.add_argument('-n', '--runs_per_test', default=1, type=int)
argp.add_argument('-f', '--forever',
                  default=False,
                  action='store_const',
                  const=True)
args = argp.parse_args()

# grab config
run_configs = set(_CONFIGS[cfg]
                  for cfg in itertools.chain.from_iterable(
                      _CONFIGS.iterkeys() if x == 'all' else [x]
                      for x in args.config))
build_configs = set(cfg.build_config for cfg in run_configs)
filters = args.test_filter
runs_per_test = args.runs_per_test
forever = args.forever


def _build_and_run(check_cancelled):
  """Do one pass of building & running tests."""
  # build latest, sharing cpu between the various makes
  if not jobset.run(
      (['make',
        '-j', '%d' % (multiprocessing.cpu_count() + 1),
        'CONFIG=%s' % cfg] + _MAKE_TEST_TARGETS
       for cfg in build_configs),
      check_cancelled, maxjobs=1):
    return 1

  # run all the tests
  if not jobset.run(
      itertools.ifilter(
          lambda x: x is not None, (
              config.run_command(x)
              for config in run_configs
              for filt in filters
              for x in itertools.chain.from_iterable(itertools.repeat(
                  glob.glob('bins/%s/%s_test' % (
                      config.build_config, filt)),
                  runs_per_test)))),
              check_cancelled,
              maxjobs=min(c.maxjobs for c in run_configs)):
    return 2

  return 0


if forever:
  while True:
    dw = watch_dirs.DirWatcher(['src', 'include', 'test'])
    initial_time = dw.most_recent_change()
{'fraction_non_alphanumeric': 0.1257376305038584, 'fraction_numerical': 0.0025722499621727945, 'mean_word_length': 3.02557856272838, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '4485337', 'n_tokens_mistral': 2155, 'n_tokens_neox': 2084, 'n_words': 544}