def test_refresh_tags_cache_should_not_raise_exception(aggregator, dd_run_check, realtime_instance): realtime_instance.update({'collect_tags': True}) check = VSphereCheck('vsphere', {}, [realtime_instance]) check.log = MagicMock() check.api_rest = MagicMock() check.api_rest.get_resource_tags_for_mors.side_effect = APIConnectionError("Some error") check.collect_tags({}) # Error logged, but `refresh_tags_cache` should NOT raise any exception check.log.error.assert_called_once_with("Failed to collect tags: %s", mock.ANY)
def test_renew_rest_api_session_on_failure(aggregator, dd_run_check, realtime_instance): realtime_instance.update({'collect_tags': True}) check = VSphereCheck('vsphere', {}, [realtime_instance]) config = VSphereConfig(realtime_instance, {}, MagicMock()) check.api_rest = VSphereRestAPI(config, MagicMock()) check.api_rest.make_batch = MagicMock(side_effect=[Exception, []]) check.api_rest.smart_connect = MagicMock() tags = check.collect_tags({}) assert tags assert check.api_rest.make_batch.call_count == 2 assert check.api_rest.smart_connect.call_count == 1