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)