def test_timing(self):
client = Client("localhost")
client._socket = self.mock_socket
client.timing("event", 10)
self.mock_sendto.assert_called_with(
"event:10|ms".encode(),
("127.0.0.2", 8125)
)
client.timing("db.event name", 34.5, 0.5)
self.mock_sendto.assert_called_with(
"db.event_name:34|ms|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.timing("db/query", rate=0.7, milliseconds=22.22)
self.mock_sendto.assert_called_with(
"region.c_db-query:22|ms|@0.7".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.timing("low.rate", 12, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
self.assertRaises(AssertionError, client.timing, "negative", -1)
def test_client_creates_stopwatch(self):
test_start_timestamp = time()
one_minute_before_test = test_start_timestamp - 60
client = Client("localhost")
client._socket = self.mock_socket
stopwatch = client.stopwatch("event")
self.assertIsInstance(stopwatch, Stopwatch)
self.assertEqual(stopwatch.client, client)
self.assertEqual(stopwatch.rate, 1)
self.assertGreaterEqual(stopwatch.reference, test_start_timestamp)
stopwatch_low_rate = client.stopwatch("low_rate", rate=0.001)
self.assertEqual(stopwatch_low_rate.rate, 0.001)
self.assertGreaterEqual(stopwatch.reference, test_start_timestamp)
stopwatch_1min_ref = client.stopwatch("low_rate", reference=one_minute_before_test)
self.assertGreaterEqual(test_start_timestamp, stopwatch_1min_ref.reference)
with client.stopwatch("something"):
sleep(0.01)
self.assertEqual(self.mock_sendto.call_count, 1)
request_args = self.mock_sendto.call_args[0]
self.assertEqual(len(request_args), 2)
request = request_args[0]
self.assertRegex(request.decode(), "something:[1-9]\d{0,3}\|ms")
def test_client_creates_stopwatch(self):
test_start_timestamp = time()
one_minute_before_test = test_start_timestamp - 60
client = Client("localhost")
client._socket = self.mock_socket
stopwatch = client.stopwatch("event")
self.assertIsInstance(stopwatch, Stopwatch)
self.assertEqual(stopwatch.client, client)
self.assertEqual(stopwatch.rate, 1)
self.assertGreaterEqual(stopwatch.reference, test_start_timestamp)
stopwatch_low_rate = client.stopwatch("low_rate", rate=0.001)
self.assertEqual(stopwatch_low_rate.rate, 0.001)
self.assertGreaterEqual(stopwatch.reference, test_start_timestamp)
stopwatch_1min_ref = client.stopwatch("low_rate", reference=one_minute_before_test)
self.assertGreaterEqual(test_start_timestamp, stopwatch_1min_ref.reference)
with client.stopwatch("something"):
sleep(0.01)
self.assertEqual(self.mock_sendto.call_count, 1)
request_args = self.mock_sendto.call_args[0]
self.assertEqual(len(request_args), 2)
request = request_args[0]
self.assertRegex(request.decode(), "something:[1-9]\d{0,3}\|ms")
def test_decrement(self):
client = Client("localhost")
client._socket = self.mock_socket
client.decrement("event")
self.mock_sendto.assert_called_with(
"event:-1|c".encode(),
("127.0.0.2", 8125)
)
client.decrement("event2", 5)
self.mock_sendto.assert_called_with(
"event2:-5|c".encode(),
("127.0.0.2", 8125)
)
client.decrement("region.event name", 2, 0.5)
self.mock_sendto.assert_called_with(
"region.event_name:-2|c|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.decrement("active!users", rate=0.7)
self.mock_sendto.assert_called_with(
"region.c_activeusers:-1|c|@0.7".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.decrement("low.rate", rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_increment(self):
client = Client("localhost")
client._socket = self.mock_socket
client.increment("event")
self.mock_sendto.assert_called_with(
"event:1|c".encode(),
("127.0.0.2", 8125)
)
client.increment("event2", 5)
self.mock_sendto.assert_called_with(
"event2:5|c".encode(),
("127.0.0.2", 8125)
)
client.increment("region.event name", 2, 0.5)
self.mock_sendto.assert_called_with(
"region.event_name:2|c|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.increment("@login#", rate=0.6)
self.mock_sendto.assert_called_with(
"region.c_login:1|c|@0.6".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.increment("low.rate", rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_increment(self):
client = Client("localhost")
client._socket = self.mock_socket
client.increment("event")
self.mock_sendto.assert_called_with(
"event:1|c".encode(),
("127.0.0.2", 8125)
)
client.increment("event2", 5)
self.mock_sendto.assert_called_with(
"event2:5|c".encode(),
("127.0.0.2", 8125)
)
client.increment("region.event name", 2, 0.5)
self.mock_sendto.assert_called_with(
"region.event_name:2|c|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.increment("@login#", rate=0.6)
self.mock_sendto.assert_called_with(
"region.c_login:1|c|@0.6".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.increment("low.rate", rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_timing(self):
client = Client("localhost")
client._socket = self.mock_socket
client.timing("event", 10)
self.mock_sendto.assert_called_with(
"event:10|ms".encode(),
("127.0.0.2", 8125)
)
client.timing("db.event name", 34.5, 0.5)
self.mock_sendto.assert_called_with(
"db.event_name:34|ms|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.timing("db/query", rate=0.7, milliseconds=22.22)
self.mock_sendto.assert_called_with(
"region.c_db-query:22|ms|@0.7".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.timing("low.rate", 12, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
self.assertRaises(AssertionError, client.timing, "negative", -1)
def test_decrement(self):
client = Client("localhost")
client._socket = self.mock_socket
client.decrement("event")
self.mock_sendto.assert_called_with(
"event:-1|c".encode(),
("127.0.0.2", 8125)
)
client.decrement("event2", 5)
self.mock_sendto.assert_called_with(
"event2:-5|c".encode(),
("127.0.0.2", 8125)
)
client.decrement("region.event name", 2, 0.5)
self.mock_sendto.assert_called_with(
"region.event_name:-2|c|@0.5".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region.c_"
client.decrement("active!users", rate=0.7)
self.mock_sendto.assert_called_with(
"region.c_activeusers:-1|c|@0.7".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.decrement("low.rate", rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_context_manager_flushes_metrics_when_context_ends(self):
client = Client("localhost", prefix="_.")
client._socket = self.mock_socket
with client.batch_client() as batch_client:
batch_client.increment("event", rate=0.5)
batch_client.timing("query", 1200)
batch_client.decrement("event", rate=0.2)
self.assertEqual(self.mock_sendto.call_count, 0)
expected_calls = [
mock.call(bytearray("_.event:1|c|@0.5\n_.query:1200|ms\n".encode()), ("127.0.0.2", 8125)),
]
self.assertEqual(self.mock_sendto.mock_calls, expected_calls)
def test_context_manager_flushes_metrics(self):
client = Client("localhost", prefix="_.")
client._socket = self.mock_socket
with client.batch_client() as batch_client:
batch_client.increment("event", rate=0.5)
batch_client.timing("query", 1200)
batch_client.decrement("event", rate=0.2)
self.assertEqual(self.mock_sendto.call_count, 0)
expected_calls = [
mock.call(bytearray("_.event:1|c|@0.5\n_.query:1200|ms\n".encode()), ("127.0.0.2", 8125)),
]
self.assertEqual(self.mock_sendto.mock_calls, expected_calls)
def test_context_manager_flushes_metrics_when_context_raises_errors(self):
client = Client("localhost", prefix="_.")
client._socket = self.mock_socket
with self.assertRaises(RuntimeError):
with client.batch_client() as batch_client:
batch_client.increment("event", rate=0.5)
batch_client.timing("query", 300)
self.assertEqual(self.mock_sendto.call_count, 0)
raise RuntimeError('mock error')
batch_client.timing("not.there", 300)
expected_calls = [
mock.call(bytearray("_.event:1|c|@0.5\n_.query:300|ms\n".encode()), ("127.0.0.2", 8125)),
]
self.assertEqual(self.mock_sendto.mock_calls, expected_calls)
def test_timing_since_with_datetime_timestamp(self):
start_time = datetime.now()
client = Client("localhost")
client._socket = self.mock_socket
sleep(0.01)
client.timing_since("event", start_time)
self.assertEqual(self.mock_sendto.call_count, 1)
socket_sendto_args = self.mock_sendto.call_args
self.assertEqual(len(socket_sendto_args), 2)
request, remote_address = socket_sendto_args[0]
self.assertRegex(request.decode(), "event:[1-9]\d{0,3}\|ms")
self.assertEqual(remote_address, ("127.0.0.2", 8125))
self.mock_sendto.reset_mock()
client.timing_since("low.rate", start_time, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_timing_since_with_datetime_timestamp(self):
start_time = datetime.now()
client = Client("localhost")
client._socket = self.mock_socket
sleep(0.01)
client.timing_since("event", start_time)
self.assertEqual(self.mock_sendto.call_count, 1)
socket_sendto_args = self.mock_sendto.call_args
self.assertEqual(len(socket_sendto_args), 2)
request, remote_address = socket_sendto_args[0]
self.assertRegex(request.decode(), "event:[1-9]\d{0,3}\|ms")
self.assertEqual(remote_address, ("127.0.0.2", 8125))
self.mock_sendto.reset_mock()
client.timing_since("low.rate", start_time, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_timing_since_with_timestamp_as_number(self):
start_time = time()
client = Client("localhost")
client._socket = self.mock_socket
self.assertRaises(AssertionError, client.timing_since, "negative", -1)
sleep(0.01)
client.timing_since("event", start_time)
self.assertEqual(self.mock_sendto.call_count, 1)
socket_sendto_args = self.mock_sendto.call_args
self.assertEqual(len(socket_sendto_args), 2)
request, remote_address = socket_sendto_args[0]
self.assertRegex(request.decode(), "event:[1-9]+\d{0,3}\|ms")
self.assertEqual(remote_address, ("127.0.0.2", 8125))
self.mock_sendto.reset_mock()
client.timing_since("low.rate", start_time, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_gauge_delta(self):
client = Client("localhost")
client._socket = self.mock_socket
client.gauge_delta("memory!", 128)
self.mock_sendto.assert_called_with(
"memory:+128|g".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.gauge_delta("cpu percentage%", rate=0.9, delta=-12)
self.mock_sendto.assert_called_with(
"region.cpu_percentage:-12|g|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.gauge_delta("low.rate", 10, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_set(self):
client = Client("localhost")
client._socket = self.mock_socket
client.set("ip address", "10.10.10.1")
self.mock_sendto.assert_called_with(
"ip_address:10.10.10.1|s".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.set("~username*", rate=0.9, value='first')
self.mock_sendto.assert_called_with(
"region.username:first|s|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.set("low.rate", 256, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_timing_since_with_timestamp_as_number(self):
start_time = time()
client = Client("localhost")
client._socket = self.mock_socket
self.assertRaises(AssertionError, client.timing_since, "negative", -1)
sleep(0.01)
client.timing_since("event", start_time)
self.assertEqual(self.mock_sendto.call_count, 1)
socket_sendto_args = self.mock_sendto.call_args
self.assertEqual(len(socket_sendto_args), 2)
request, remote_address = socket_sendto_args[0]
self.assertRegex(request.decode(), "event:[1-9]+\d{0,3}\|ms")
self.assertEqual(remote_address, ("127.0.0.2", 8125))
self.mock_sendto.reset_mock()
client.timing_since("low.rate", start_time, rate=0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_set(self):
client = Client("localhost")
client._socket = self.mock_socket
client.set("ip address", "10.10.10.1")
self.mock_sendto.assert_called_with(
"ip_address:10.10.10.1|s".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.set("~username*", rate=0.9, value='first')
self.mock_sendto.assert_called_with(
"region.username:first|s|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.set("low.rate", 256, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_gauge_delta(self):
client = Client("localhost")
client._socket = self.mock_socket
client.gauge_delta("memory!", 128)
self.mock_sendto.assert_called_with(
"memory:+128|g".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.gauge_delta("cpu percentage%", rate=0.9, delta=-12)
self.mock_sendto.assert_called_with(
"region.cpu_percentage:-12|g|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.gauge_delta("low.rate", 10, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
def test_context_manager_creates_batch_client(self):
client = Client("localhost")
client._socket = self.mock_socket
with client.batch_client() as batch_client:
self.assertIsInstance(batch_client, BatchClient)
self.assertGreater(batch_client.batch_size, 0)
self.assertEqual(client.host, batch_client.host)
self.assertEqual(client.port, batch_client.port)
self.assertEqual(
client._remote_address,
batch_client._remote_address
)
self.assertEqual(
client._socket,
batch_client._socket
)
with client.batch_client(2048) as batch_client:
self.assertEqual(batch_client.batch_size, 2048)
def test_context_manager_creates_batch_client(self):
client = Client("localhost")
client._socket = self.mock_socket
with client.batch_client() as batch_client:
self.assertIsInstance(batch_client, BatchClient)
self.assertGreater(batch_client.batch_size, 0)
self.assertEqual(client.host, batch_client.host)
self.assertEqual(client.port, batch_client.port)
self.assertEqual(
client._remote_address,
batch_client._remote_address
)
self.assertEqual(
client._socket,
batch_client._socket
)
with client.batch_client(2048) as batch_client:
self.assertEqual(batch_client.batch_size, 2048)
def test_gauge(self):
client = Client("localhost")
client._socket = self.mock_socket
client.gauge("memory", 10240)
self.mock_sendto.assert_called_with(
"memory:10240|g".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.gauge("cpu percentage%", rate=0.9, value=98.3)
self.mock_sendto.assert_called_with(
"region.cpu_percentage:98.3|g|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.gauge("low.rate", 128, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
self.assertRaises(AssertionError, client.gauge, "negative", -5)
def test_gauge(self):
client = Client("localhost")
client._socket = self.mock_socket
client.gauge("memory", 10240)
self.mock_sendto.assert_called_with(
"memory:10240|g".encode(),
("127.0.0.2", 8125)
)
client.prefix = "region."
client.gauge("cpu percentage%", rate=0.9, value=98.3)
self.mock_sendto.assert_called_with(
"region.cpu_percentage:98.3|g|@0.9".encode(),
("127.0.0.2", 8125)
)
self.mock_sendto.reset_mock()
client.gauge("low.rate", 128, 0.1)
self.assertEqual(self.mock_sendto.call_count, 0)
self.assertRaises(AssertionError, client.gauge, "negative", -5)
