Esempio n. 1
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)
Esempio n. 2
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)
Esempio n. 3
0
 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)
Esempio n. 4
0
 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)
Esempio n. 5
0
    def test_decrement(self):
        client = BatchClient("localhost")
        client._socket = self.mock_socket
        client.decrement("event", 3, 0.7)
        client.flush()
        self.mock_sendto.assert_called_with(
            bytearray("event:-3|c|@0.7\n".encode()),
            ("127.0.0.2", 8125)
        )
        self.mock_sendto.reset_mock()
        client.prefix = "pre."
        client.decrement("session")
        client.decrement("session.fail", 2, 0.2)
        client.decrement("session.ok", rate=0.6)
        client.flush()

        self.mock_sendto.assert_called_once_with(
            bytearray("pre.session:-1|c\npre.session.ok:-1|c|@0.6\n".encode()),
            ("127.0.0.2", 8125)
        )
Esempio n. 6
0
    def test_decrement(self):
        client = BatchClient("localhost")
        client._socket = self.mock_socket
        client.decrement("event", 3, 0.7)
        client.flush()
        self.mock_sendto.assert_called_with(
            bytearray("event:-3|c|@0.7\n".encode()),
            ("127.0.0.2", 8125)
        )
        self.mock_sendto.reset_mock()
        client.prefix = "pre."
        client.decrement("session")
        client.decrement("session.fail", 2, 0.2)
        client.decrement("session.ok", rate=0.6)
        client.flush()

        self.mock_sendto.assert_called_once_with(
            bytearray("pre.session:-1|c\npre.session.ok:-1|c|@0.6\n".encode()),
            ("127.0.0.2", 8125)
        )