Ejemplo n.º 1
0
def test_get_service_instance_stats():
    # The patch stuff is confusing.
    # Basically we patch the validate_service_instance in the paasta_metastatus module and not the utils module
    instance_config_mock = Mock()
    with patch(
        'paasta_tools.paasta_metastatus.get_instance_config', autospec=True, return_value=instance_config_mock,
    ):
        stats = paasta_metastatus.get_service_instance_stats('fakeservice', 'fakeinstance', 'fakecluster')
        assert set(stats.keys()) == {'mem', 'cpus', 'disk', 'gpus'}
Ejemplo n.º 2
0
def test_get_service_instance_stats():
    # The patch stuff is confusing.
    # Basically we patch the validate_service_instance in the paasta_metastatus module and not the utils module
    instance_config_mock = Mock()
    instance_config_mock.get_gpus.return_value = None
    with patch(
        "paasta_tools.paasta_metastatus.get_instance_config",
        autospec=True,
        return_value=instance_config_mock,
    ):
        stats = paasta_metastatus.get_service_instance_stats(
            "fakeservice", "fakeinstance", "fakecluster"
        )
        assert set(stats.keys()) == {"mem", "cpus", "disk", "gpus"}