def test_gauge_with_timestamp(self):
"""
Test the 'gauge' wrapper to see if it doesn't strip anything for gauge functions that accept timestamps
"""
def gauge_v5(self,
metric,
value,
tags=None,
hostname=None,
device_name=None,
timestamp=None):
"""
A gauge function that should accept the 'timestamp' argument
"""
# Make sure we get the original arguments back
assert metric == METRIC_NAME
assert value == METRIC_VALUE
assert tags == METRIC_TAGS
assert hostname is None
assert device_name is None
assert timestamp == METRIC_TIMESTAMP
# This should call 'gauge_v5' with all the same arguments that we passed in
with patch('datadog_checks.checks.AgentCheck.gauge', new=gauge_v5):
nagios = Nagios(CHECK_NAME, {}, {})
nagios.gauge(METRIC_NAME,
METRIC_VALUE,
tags=METRIC_TAGS,
timestamp=METRIC_TIMESTAMP)
def test_gauge_without_timestamp(self):
"""
Test the 'gauge' wrapper to see if it strips the 'timestamp' arg for gauge functions that do not accept it
"""
def gauge_v6(self,
name,
value,
tags=None,
hostname=None,
device_name=None):
"""
A gauge function that does not accept the 'timestamp' argument
"""
# Make sure we get the original arguments back and timestamp is not being received
assert name == METRIC_NAME
assert value == METRIC_VALUE
assert tags == METRIC_TAGS
assert hostname is None
assert device_name is None
# This should call 'gauge_v6' with all the same arguments that we passed in, except for the 'timestamp' argument
with patch('datadog_checks.checks.AgentCheck.gauge', new=gauge_v6):
nagios = Nagios(CHECK_NAME, {}, {})
nagios.gauge(METRIC_NAME,
METRIC_VALUE,
tags=METRIC_TAGS,
timestamp=METRIC_TIMESTAMP)
