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_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_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_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_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)