def test_tcp_crlf_dp(self):
tcp_receiver = MetricReceiverTcp(self.config)
my_spool = tcp_receiver.spool
tcp_receiver._sock.set_metric(self.test_metric, 20, lf='\r\n')
setup_tcp_receiver(tcp_receiver)
reciever_run_shutdown(tcp_receiver, 1)
self.assertEqual(len(my_spool._spools), 1)
def test_udp_crlf_dp(self):
my_spool = FakeSpool()
udp_receiver = MetricReceiverUdp(self.config)
udp_receiver._sock.set_metric(self.test_metric, 20, lf='\r\n')
self.setup_udp_receiver(udp_receiver, my_spool)
reciever_run_shutdown(udp_receiver, 1)
self.assertEqual(len(my_spool.metrics), 1)
def test_tcp_multi_dp(self):
tcp_receiver = MetricReceiverTcp(self.config)
my_spool = tcp_receiver.spool
[tcp_receiver._sock.set_metric(self.test_metric, 20) for _ in range(20)]
setup_tcp_receiver(tcp_receiver)
reciever_run_shutdown(tcp_receiver, 5)
self.assertEqual(len(my_spool._spools), 1)
def test_key_in_metric(self):
tcp_receiver = MetricReceiverTcp(self.config)
my_spool = tcp_receiver.spool
tcp_receiver._sock.metric = "%s.%s %s\n" % (API_KEY, "foo.bar.baz", 20)
tcp_receiver._sock.metric_count += 1
setup_tcp_receiver(tcp_receiver)
reciever_run_shutdown(tcp_receiver, 1)
self.assertEqual(len(my_spool._spools), 1)
def test_invalid_metric(self):
tcp_receiver = MetricReceiverTcp(self.config)
my_spool = tcp_receiver.spool
tcp_receiver._sock.set_metric("@@1.---=++", 20)
setup_tcp_receiver(tcp_receiver)
# this won't result in any update.. sleep then shutdown.
reciever_run_shutdown(tcp_receiver, 1)
# no valid metric, no spool created.
self.assertEqual(len(my_spool._spools), 0)
def test_udp_single_dp(self):
my_spool = FakeSpool()
udp_receiver = MetricReceiverUdp(self.config)
udp_receiver._sock.set_metric(self.test_metric, 20)
self.setup_udp_receiver(udp_receiver, my_spool)
reciever_run_shutdown(udp_receiver, 1)
self.assertEqual(len(my_spool.metrics), 1)
self.assertEqual(my_spool.metrics[0].metric, "%s.foo.bar.baz" % API_KEY)
self.assertEqual(my_spool.metrics[0].value, 20)
def test_udp_multi_dp(self):
my_spool = FakeSpool()
udp_receiver = MetricReceiverUdp(self.config)
[udp_receiver._sock.set_metric(self.test_metric, 20) for _ in range(10)]
self.setup_udp_receiver(udp_receiver, my_spool)
reciever_run_shutdown(udp_receiver, 10)
self.assertEqual(len(my_spool.metrics), 10)
self.assertEqual(my_spool.metrics[random.choice(range(10))].metric,
"%s.foo.bar.baz" % API_KEY)
self.assertEqual(my_spool.metrics[random.choice(range(10))].value, 20)
def test_fileno_1_string(self):
tcp_receiver = MetricReceiverTcp(self.config)
my_spool = tcp_receiver.spool
tcp_receiver._sock.fileno_res = 1
tcp_receiver._sock.set_metric(self.test_metric, 20)
setup_tcp_receiver(tcp_receiver)
time.sleep(1)
tcp_receiver._sock.fileno_res = 0
tcp_receiver._sock.set_metric(self.test_metric, 20)
tcp_receiver._sock.fileno_res = 0
time.sleep(1)
reciever_run_shutdown(tcp_receiver, 1)
self.assertEqual(len(my_spool._spools), 1)
def test_tcp_multi_dp_low_timeout(self):
conf = self.config
conf['tcp']['timeout'] = 0.0001
tcp_receiver = MetricReceiverTcp(conf)
my_spool = tcp_receiver.spool
tcp_receiver._sock.set_metric(self.test_metric, 20)
setup_tcp_receiver(tcp_receiver)
time.sleep(2)
tcp_receiver._sock.set_metric(self.test_metric, 20)
tcp_receiver._sock.set_metric(self.test_metric, 22)
time.sleep(2)
reciever_run_shutdown(tcp_receiver, 2)
self.assertEqual(len(my_spool._spools), 1)
def test_too_many_spools(self):
tcp_receiver = MetricReceiverTcp(self.config)
# create too many spools, sleep to ensure new ts
for t in range(1000, 1011):
write_spool(t)
my_spool = tcp_receiver.spool
tcp_receiver._sock.set_metric(self.test_metric, 20)
setup_tcp_receiver(tcp_receiver)
time.sleep(1)
reciever_run_shutdown(tcp_receiver, 1)
all_spools = my_spool.lookup_spools()
for ts in all_spools:
self.assertIsInstance(ts, int)
self.assertEqual(len(all_spools), 10)
def test_too_many_spools_rotate_bytes(self):
conf = self.config
conf['spool_rotatesize'] = 10
tcp_receiver = MetricReceiverTcp(conf)
# create too many spools, sleep to ensure new ts
for t in range(1000, 1011):
write_spool(t)
my_spool = tcp_receiver.spool
[tcp_receiver._sock.set_metric(self.test_metric, 20) for _ in range(4)]
setup_tcp_receiver(tcp_receiver)
while len(my_spool.lookup_spools()) > 10:
tcp_receiver._sock.set_metric(self.test_metric, 20)
time.sleep(0.1)
reciever_run_shutdown(tcp_receiver, 1)
self.assertEqual(len(my_spool.lookup_spools()), 10)