def test_sending_batch_metrics(self):
start = datetime.now()
start_timestamp = time()
client = TCPBatchClient("localhost", self.__class__.port)
client.increment("1.test", 8)
client.increment("2.login")
client.timing("3.query", 9600)
client.gauge("4.memory", 102600)
client.gauge_delta("5.memory", 2560)
client.gauge_delta("6.memory", -1280)
client.set("7.ip", "127.0.0.2")
client.flush()
expected = [
"1.test:8|c",
"2.login:1|c",
"3.query:9600|ms",
"4.memory:102600|g",
"5.memory:+2560|g",
"6.memory:-1280|g",
"7.ip:127.0.0.2|s",
]
self.assert_server_received_expected_requests(expected)
client.timing_since("1.query", start_timestamp)
client.timing_since("2.other_query", start)
client.flush()
chronometer = client.chronometer()
chronometer.time_callable("3.sleepy", sleep, 1, (0.02, ))
@chronometer.wrap("4.wait_a_sec")
def wait_a_sec():
sleep(0.01)
wait_a_sec()
with client.stopwatch("5.my_with_block"):
sleep(0.02)
client.flush()
expected_patterns = [
"1.query:[1-9]\d{0,4}\|ms",
"2.other_query:[1-9]\d{0,4}\|ms",
"3.sleepy:[1-9]\d{0,4}\|ms",
"4.wait_a_sec:[1-9]\d{0,4}\|ms",
"5.my_with_block:[1-9]\d{0,4}\|ms",
]
self.assert_server_received_expected_request_regex(expected_patterns)
def test_sending_batch_metrics(self):
start = datetime.now()
start_timestamp = time()
client = TCPBatchClient("localhost", self.__class__.port)
client.increment("1.test", 8)
client.increment("2.login")
client.timing("3.query", 9600)
client.gauge("4.memory", 102600)
client.gauge_delta("5.memory", 2560)
client.gauge_delta("6.memory", -1280)
client.set("7.ip", "127.0.0.2")
client.flush()
expected = [
"1.test:8|c",
"2.login:1|c",
"3.query:9600|ms",
"4.memory:102600|g",
"5.memory:+2560|g",
"6.memory:-1280|g",
"7.ip:127.0.0.2|s",
]
self.assert_server_received_expected_requests(expected)
client.timing_since("1.query", start_timestamp)
client.timing_since("2.other_query", start)
client.flush()
chronometer = client.chronometer()
chronometer.time_callable("3.sleepy", sleep, 1, (0.02,))
@chronometer.wrap("4.wait_a_sec")
def wait_a_sec():
sleep(0.01)
wait_a_sec()
with client.stopwatch("5.my_with_block"):
sleep(0.02)
client.flush()
expected_patterns = [
"1.query:[1-9]\d{0,4}\|ms",
"2.other_query:[1-9]\d{0,4}\|ms",
"3.sleepy:[1-9]\d{0,4}\|ms",
"4.wait_a_sec:[1-9]\d{0,4}\|ms",
"5.my_with_block:[1-9]\d{0,4}\|ms",
]
self.assert_server_received_expected_request_regex(expected_patterns)
def test_metrics_partitioned_into_batches(self):
client = TCPBatchClient("localhost", batch_size=20)
client._socket = self.mock_socket
client.increment("fit.a.batch.123")
client.timing("_", 1)
client.increment("larger.than.batch.becomes.a.batch", 5, 0.9)
client.decrement("12")
client.set("ab", 'z')
client.timing("small", 9)
client.gauge("overflow.previous", 10)
client.gauge_delta("next", -10)
client.increment("_")
client.flush()
expected_calls = [
mock.call(bytearray("fit.a.batch.123:1|c\n".encode())),
mock.call(bytearray("_:1|ms\n".encode())),
mock.call(bytearray("larger.than.batch.becomes.a.batch:5|c|@0.9\n".encode())),
mock.call(bytearray("12:-1|c\nab:z|s\n".encode())),
mock.call(bytearray("small:9|ms\n".encode())),
mock.call(bytearray("overflow.previous:10|g\n".encode())),
mock.call(bytearray("next:-10|g\n_:1|c\n".encode()))
]
self.assertEqual(self.mock_sendall.mock_calls, expected_calls)
def test_gauge_delta(self):
client = TCPBatchClient("localhost")
client._socket = self.mock_socket
client.gauge_delta("memory", -512)
client.flush()
self.mock_sendall.assert_called_with(
bytearray("memory:-512|g\n".encode())
)
self.mock_sendall.reset_mock()
client.prefix = "pre."
client.gauge_delta("memory", 2048)
client.gauge_delta("memory", 8096, 0.2)
client.gauge_delta("storage", -128, rate=0.7)
client.flush()
self.mock_sendall.assert_called_once_with(
bytearray("pre.memory:+2048|g\npre.storage:-128|g|@0.7\n".encode())
)
