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
