def test_clear(self):
client = BatchClient("localhost", batch_size=20)
client._socket = self.mock_socket
client.increment("first")
client.decrement("second")
client.timing("db.query", 1)
client.clear()
client.flush()
self.assertEqual(self.mock_sendto.call_count, 0)
def test_clear(self):
client = BatchClient("localhost", batch_size=20)
client._socket = self.mock_socket
client.increment("first")
client.decrement("second")
client.timing("db.query", 1)
client.clear()
client.flush()
self.assertEqual(self.mock_sendto.call_count, 0)
def test_sending_batch_metrics(self):
start = datetime.now()
start_timestamp = time()
client = BatchClient("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)
self.__class__.server.requests.clear()
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 = BatchClient("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)
self.__class__.server.requests.clear()
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_increment(self):
client = BatchClient("localhost")
client._socket = self.mock_socket
client.increment("event", 2, 0.5)
client.flush()
self.mock_sendto.assert_called_with(
bytearray("event:2|c|@0.5\n".encode()),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.prefix = "pre."
client.increment("login")
client.increment("login.fail", 5, 0.2)
client.increment("login.ok", rate=0.8)
client.flush()
self.mock_sendto.assert_called_once_with(
bytearray("pre.login:1|c\npre.login.ok:1|c|@0.8\n".encode()),
("127.0.0.2", 8125)
)
def test_increment(self):
client = BatchClient("localhost")
client._socket = self.mock_socket
client.increment("event", 2, 0.5)
client.flush()
self.mock_sendto.assert_called_with(
bytearray("event:2|c|@0.5\n".encode()),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.prefix = "pre."
client.increment("login")
client.increment("login.fail", 5, 0.2)
client.increment("login.ok", rate=0.8)
client.flush()
self.mock_sendto.assert_called_once_with(
bytearray("pre.login:1|c\npre.login.ok:1|c|@0.8\n".encode()),
("127.0.0.2", 8125)
)
def test_metrics_partitioned_into_batches(self):
client = BatchClient("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()), ("127.0.0.2", 8125)),
mock.call(bytearray("_:1|ms\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("larger.than.batch.becomes.a.batch:5|c|@0.9\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("12:-1|c\nab:z|s\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("small:9|ms\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("overflow.previous:10|g\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("next:-10|g\n_:1|c\n".encode()), ("127.0.0.2", 8125)),
]
self.assertEqual(self.mock_sendto.mock_calls, expected_calls)
def test_metrics_partitioned_into_batches(self):
client = BatchClient("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()), ("127.0.0.2", 8125)),
mock.call(bytearray("_:1|ms\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("larger.than.batch.becomes.a.batch:5|c|@0.9\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("12:-1|c\nab:z|s\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("small:9|ms\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("overflow.previous:10|g\n".encode()), ("127.0.0.2", 8125)),
mock.call(bytearray("next:-10|g\n_:1|c\n".encode()), ("127.0.0.2", 8125)),
]
self.assertEqual(self.mock_sendto.mock_calls, expected_calls)