def main(argv): if len(argv) > 1: raise app.UsageError('Unrecognized arguments') if FLAGS.generator_batch_size <= 0: raise app.UsageError('--generator_batch_size must be positive') datastore_config = services.ServiceConfigFromFlag( 'datastore_config', datastore_pb2.DataStore()) generator_config = services.ServiceConfigFromFlag( 'generator_config', generator_pb2.ClgenGenerator()) datastore_stub = services.GetServiceStub( datastore_config, datastore_pb2_grpc.DataStoreServiceStub) generator_stub = services.GetServiceStub( generator_config, generator_pb2_grpc.GeneratorServiceStub) target_total_testcases = FLAGS.target_total_testcases generator_batch_size = FLAGS.generator_batch_size capabilities = GetGeneratorCapabilities(generator_stub) while True: num_testcases = GetNumberOfTestcasesInDataStore( datastore_stub, capabilities) logging.info(f'Number of testcases in datastore: %d', num_testcases) if 0 <= target_total_testcases <= num_testcases: logging.info('Stopping generation with %d testcases in the DataStore.', num_testcases) break num_to_generate = generator_batch_size if target_total_testcases >= 0: num_to_generate = min( generator_batch_size, target_total_testcases - num_testcases) testcases = GenerateTestcases(generator_stub, num_to_generate) SubmitTestcases(datastore_stub, testcases)
def main(argv): if len(argv) > 1: raise app.UsageError("Unrecognized arguments") if FLAGS.harness_batch_size <= 0: raise app.UsageError("--harness_batch_size must be positive") datastore_config = services.ServiceConfigFromFlag( "datastore_config", datastore_pb2.DataStore()) harness_config = services.ServiceConfigFromFlag( "harness_config", harness_pb2.CldriveHarness()) datastore_stub = services.GetServiceStub( datastore_config, datastore_pb2_grpc.DataStoreServiceStub) harness_stub = services.GetServiceStub(harness_config, harness_pb2_grpc.HarnessServiceStub) target_total_results = FLAGS.target_total_results harness_batch_size = FLAGS.harness_batch_size capabilities = GetHarnessCapabilities(harness_stub) testbeds = collections.deque(capabilities.testbeds) if testbeds: app.Log( 1, "%d testbeds: %s", len(capabilities.testbeds), ", ".join(x.name for x in capabilities.testbeds), ) while testbeds: testbed = testbeds.popleft() testcases = GetTestcasesToRun( datastore_stub, capabilities.harness, testbed, target_total_results, harness_batch_size, ) app.Log( 1, "Received %d testcases to execute on %s", len(testcases), testbed.name, ) if testcases: results = RunTestcases(harness_stub, testbed, testcases) SubmitResults(datastore_stub, results) # If there are testcases to run, then we add it back to the testbeds # queue, as there may be more. testbeds.append(testbed) app.Log(1, "done") else: app.Warning("No testbeds, nothing to do!")
def main(argv): if len(argv) > 1: raise app.UsageError('Unrecognized arguments') if FLAGS.harness_batch_size <= 0: raise app.UsageError('--harness_batch_size must be positive') datastore_config = services.ServiceConfigFromFlag( 'datastore_config', datastore_pb2.DataStore()) harness_config = services.ServiceConfigFromFlag( 'harness_config', harness_pb2.CldriveHarness()) datastore_stub = services.GetServiceStub( datastore_config, datastore_pb2_grpc.DataStoreServiceStub) harness_stub = services.GetServiceStub(harness_config, harness_pb2_grpc.HarnessServiceStub) target_total_results = FLAGS.target_total_results harness_batch_size = FLAGS.harness_batch_size capabilities = GetHarnessCapabilities(harness_stub) testbeds = collections.deque(capabilities.testbeds) if testbeds: logging.info('%d testbeds: %s', len(capabilities.testbeds), ', '.join(x.name for x in capabilities.testbeds)) while testbeds: testbed = testbeds.popleft() testcases = GetTestcasesToRun(datastore_stub, capabilities.harness, testbed, target_total_results, harness_batch_size) logging.info('Received %d testcases to execute on %s', len(testcases), testbed.name) if testcases: results = RunTestcases(harness_stub, testbed, testcases) SubmitResults(datastore_stub, results) # If there are testcases to run, then we add it back to the testbeds # queue, as there may be more. testbeds.append(testbed) logging.info('done') else: logging.warning('No testbeds, nothing to do!')
def main(argv): if len(argv) > 1: raise app.UsageError('Unrecognized arguments') harness_config = services.ServiceConfigFromFlag( 'harness_config', harness_pb2.CldriveHarness()) server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) services.AssertLocalServiceHostname(harness_config.service) service = CldriveHarness(harness_config) harness_pb2_grpc.add_HarnessServiceServicer_to_server(service, server) server.add_insecure_port(f'[::]:{harness_config.service.port}') logging.info('%s listening on %s:%s', type(service).__name__, harness_config.service.hostname, harness_config.service.port) server.start() try: while True: time.sleep(3600 * 24) except KeyboardInterrupt: server.stop(0)
def RunMain(argv: typing.List[str]) -> None: if len(argv) > 1: raise app.UsageError('Unrecognized arguments') generator_config = services.ServiceConfigFromFlag( 'generator_config', config_proto_class()) server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) services.AssertLocalServiceHostname(generator_config.service) service = cls(generator_config) generator_pb2_grpc.add_GeneratorServiceServicer_to_server(service, server) server.add_insecure_port(f'[::]:{generator_config.service.port}') logging.info('%s listening on %s:%s', type(service).__name__, generator_config.service.hostname, generator_config.service.port) server.start() try: while True: time.sleep(3600 * 24) except KeyboardInterrupt: server.stop(0)
def main(argv): if len(argv) > 1: raise app.UsageError("Unrecognized arguments") datastore_config = services.ServiceConfigFromFlag( "datastore_config", datastore_pb2.DataStore()) server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) services.AssertLocalServiceHostname(datastore_config.service) service = DataStore(datastore_config) datastore_pb2_grpc.add_DataStoreServiceServicer_to_server(service, server) server.add_insecure_port(f"[::]:{datastore_config.service.port}") app.Log( 1, "%s listening on %s:%s", type(service).__name__, datastore_config.service.hostname, datastore_config.service.port, ) server.start() try: while True: time.sleep(3600 * 24) except KeyboardInterrupt: server.stop(0)
def main(argv): """Main entry point.""" if len(argv) > 1: raise app.UsageError("Unrecognized arguments") config = services.ServiceConfigFromFlag("harness_config", harness_pb2.ClLauncherHarness()) server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) services.AssertLocalServiceHostname(config.service) service = ClLauncherHarness(config) harness_pb2_grpc.add_HarnessServiceServicer_to_server(service, server) server.add_insecure_port(f"[::]:{config.service.port}") app.Log( 1, "%s listening on %s:%s", type(service).__name__, config.service.hostname, config.service.port, ) server.start() try: while True: time.sleep(3600 * 24) except KeyboardInterrupt: server.stop(0)