Python AmbariCheck.get_service_status_and_metricsの例

コード例 #1

def test_should_not_collect_if_disabled(instance):
    ambari = AmbariCheck('Ambari', {'collect_service_metrics': False, 'collect_service_status': False}, [instance])
    _mock_clusters(ambari)
    ambari.get_host_metrics = MagicMock()
    ambari.get_service_status_and_metrics = MagicMock()
    ambari.check(instance)

    assert not ambari.get_service_status_and_metrics.called

コード例 #2

def test_should_collect_host_metrics(instance):
    ambari = AmbariCheck(
        init_config={'collect_service_metrics': False, 'collect_service_status': False}, instances=[instance]
    )
    _mock_clusters(ambari)
    ambari.get_host_metrics = MagicMock()
    ambari.get_service_status_and_metrics = MagicMock()
    ambari.check(instance)

    assert ambari.get_host_metrics.called
    assert not ambari.get_service_status_and_metrics.called