import os import mock import pytest from datadog_checks.dev import docker_run, get_docker_hostname, get_here HERE = get_here() HOST = get_docker_hostname() TIDB_PORT = 10080 TIKV_PORT = 20180 PD_PORT = 2379 TIFLASH_PROXY_PORT = 20292 TIFLASH_PORT = 8234 TICDC_PORT = 8301 DM_MASTER_PORT = 8261 DM_WORKER_PORT = 8262 PUMP_PORT = 8250 # Mock metrics @pytest.fixture() def mock_tidb_metrics(): with mock.patch( 'requests.get', return_value=mock.MagicMock( status_code=200, iter_lines=lambda **kwargs: _get_mock_metrics( "mock_tidb_metrics.txt").split("\n"), headers={'Content-Type': "text/plain"},
def dd_environment(): with docker_run(os.path.join(get_here(), 'docker', 'docker-compose.yml')): instance = load_jmx_config() instance['instances'][0]['host'] = HOST instance['instances'][0]['port'] = 18983 yield instance, {'use_jmx': True}
from . import common PWD_ENV = { "ROOT_PASSWORD": common.INSTANCE_INTEGRATION.get("password", "1234"), "SSH_SERVER_IMAGE": common.SSH_SERVER_IMAGE, "ROOT_KEYPAIR_LOGIN_ENABLED": "false", } KEYPAIR_ENV = { "ROOT_PASSWORD": common.INSTANCE_INTEGRATION.get("password", "1234"), "SSH_SERVER_IMAGE": common.SSH_SERVER_IMAGE, "ROOT_KEYPAIR_LOGIN_ENABLED": "true", } LOG_PATTERN = "Server listening on 0.0.0.0 port 22." PRIVATE_KEY_FILE = os.path.join(get_here(), 'private_key') @pytest.fixture(scope="session") def dd_environment(): with docker_run( compose_file=os.path.join(get_here(), "compose", "docker-compose.yml"), env_vars=PWD_ENV, log_patterns=LOG_PATTERN, ): yield common.INSTANCE_INTEGRATION @pytest.fixture(scope="session") def dd_environment_keypair():
def dd_environment(): compose_file = os.path.join(get_here(), 'docker-compose.yml') with docker_run(compose_file, endpoints=[URL]): yield INSTANCE