Example #1
0
def mock(init_config, instance):
    check = MesosMaster(common.CHECK_NAME, init_config, [instance])
    check._get_master_roles = lambda v, x: json.loads(
        read_fixture('roles.json'))
    check._get_master_stats = lambda v, x: json.loads(
        read_fixture('stats.json'))
    check._get_master_state = lambda v, x: json.loads(
        read_fixture('state.json'))
    return check
def check():
    check = MesosMaster(CHECK_NAME, {}, {})
    check._get_master_roles = lambda v, x, y, z: json.loads(
        read_fixture('roles.json'))
    check._get_master_stats = lambda v, x, y, z: json.loads(
        read_fixture('stats.json'))
    check._get_master_state = lambda v, x, y, z: json.loads(
        read_fixture('state.json'))

    return check
def test_can_connect_service_check(instance, aggregator, test_case_name,
                                   request_mock_side_effects, expected_status,
                                   expected_tags, expect_exception):
    check = MesosMaster('mesos_master', {}, [instance])

    with mock.patch('datadog_checks.base.utils.http.requests') as r:
        r.get.side_effect = request_mock_side_effects

        try:
            check._get_master_state('http://hello.com', ['my:tag'])
            exception_raised = False
        except CheckException:
            exception_raised = True

        assert expect_exception == exception_raised

    aggregator.assert_service_check('mesos_master.can_connect',
                                    count=1,
                                    status=expected_status,
                                    tags=expected_tags)