Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use compose_config method in yandex-tank

Best Python code snippet using yandex-tank

fungiculture-builder.py

Source:fungiculture-builder.py

...51                pass52        else:53            pass54        return self.cluster_compute_nodes55    def generate_compose_config(self):56        """57        method: construct dict data structure to layout Docker Compose configuration58        """59        self.compose_config = {}60        self.compose_config['version'] = self.version61        self.compose_config['networks'] = {}62        self.compose_config['networks'][self.network_name] = {'driver': 'bridge'}63        # set up nodes64        self.compose_config['services'] = {}65        # head node66        self.compose_config['services'][self.cluster_head_node] = {}67        self.compose_config['services'][self.cluster_head_node]['image'] = self.image68        self.compose_config['services'][self.cluster_head_node]['networks'] = [self.network_name]69        # compute nodes70        for node in self.cluster_compute_nodes:71            self.compose_config['services'][node] = {}72            self.compose_config['services'][node]['image'] = self.image73            self.compose_config['services'][node]['networks'] = [self.network_name]74        return self.compose_config75if __name__ == '__main__':76    # set up command arguments77    parser = argparse.ArgumentParser(description='Mushroom HPC Cluster Docker Compose Configuration Generator')78    parser.add_argument('--version', type=str, required=False, default='3.8', help='Docker Compose template version (default: 3.8)')79    parser.add_argument('--image', type=str, required=True, help='Docker image name')80    parser.add_argument('--network-name', type=str, required=True, help='Docker network name(bridge mode)')81    parser.add_argument('--head-node', type=str, required=True, help='Docker Mushroom HPC head node name')82    parser.add_argument('--compute-nodes', type=str, required=True, help='Docker Mushroom HPC compute node names (example: spore1,spore2,spore3)')83    args = parser.parse_args()84    # initialize config object85    mushroom_dc_config = ComposeBuilder()86    mushroom_dc_config.set_version(args.version)87    mushroom_dc_config.set_image(args.image)88    mushroom_dc_config.set_network_name(args.network_name)89    mushroom_dc_config.set_cluster_head_node(args.head_node)90    mushroom_dc_config.set_cluster_compute_nodes(set(args.compute_nodes.split(',')))91    # generate Docker Compose configuration92    mushroom_dc_config.generate_compose_config()93    # print Docker Compose configuration...

test_helper.py

Source:test_helper.py

1import requests2import logging3import sys4import os5import pytest6import shlex7import subprocess8from retry import retry9from singledispatch import singledispatch10from subprocess import call11logging.basicConfig(stream=sys.stderr )12logging.getLogger("log").setLevel( logging.DEBUG )13log=logging.getLogger("log")14# ASSERTION HELPER METHODS15@singledispatch16def assert_status(req, status_code=200):17  log.error('input type unmatched')18  assert False  # since input type unmatched19@assert_status.register(str)20def _(url, status_code=200):21  req = requests.get(url)22  assert req.status_code == status_code, req.text23  return req24@assert_status.register(requests.Response)25def _(req, status_code=200):26  assert req.status_code == status_code, req.text27  return req28# FIXTURE HELPERS FOR USING COMPOSE29def docker_compose(compose_config, command):30  compose = 'docker-compose -f ' + os.environ['CONFIGURATIONS'] + '/' + compose_config + '.yml'31  log.debug("RUN: " + compose + ' ' + command)32  subprocess.check_output(shlex.split(compose + ' ' + command))33@retry(tries=30, delay=3)34def wait_for_web_service(url):35  log.debug("Trying to reach " + url + " ...")36  req = requests.get(url)37  if req.status_code != 200:38    raise Exception('status=' + str(req.status_code))39  return req40@pytest.fixture(scope="class")41def stack(request):42  compose_config = getattr(request.module, "compose_config", "unknown")43  wait_urls = getattr(request.module, "wait_urls", "http://unknown:123")44  log.debug('compose_config: ' + compose_config)45  docker_compose(compose_config, 'kill')46  docker_compose(compose_config, 'rm -f -v')47  docker_compose(compose_config, 'up -d')48  for url in wait_urls:49    wait_for_web_service(url)50  yield51  docker_compose(compose_config, 'kill')...

transform_circle.py

Source:transform_circle.py

1import argparse2import yaml3def convert(compose_config, container_name):4    for service in compose_config.get('services', {}).values():5        if 'volumes' in service:6            service['volumes_from'] = [f'container:{container_name}']7            del service['volumes']8parser = argparse.ArgumentParser(9    description="Transform docker-compose yaml to CircleCI-compatible files")10parser.add_argument('compose_yaml', type=argparse.FileType('rb'))11parser.add_argument('container_name')12if __name__ == '__main__':13    args = parser.parse_args()14    compose_config = yaml.safe_load(args.compose_yaml)15    convert(compose_config, args.container_name)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.