def test_push_job_ping(self): job_name = "my-job" p = Pusher(job_name, TEST_URL) registry = Registry() c = Counter("total_requests", "Total requests.", {}) registry.register(c) c.inc({ 'url': "/p/user", }) # Push to the pushgateway p.replace(registry) # Check the objects that setted the server thread self.assertEqual(Pusher.PATH.format(job_name), self.request['path'])
def test_push_replace(self): job_name = "my-job" p = Pusher(job_name, TEST_URL) registry = Registry() counter = Counter("counter_test", "A counter.", {'type': "counter"}) registry.register(counter) counter_data = (({'c_sample': '1', 'c_subsample': 'b'}, 400), ) [counter.set(c[0], c[1]) for c in counter_data] valid_result = b'[\n\x0ccounter_test\x12\nA counter.\x18\x00"=\n\r\n\x08c_sample\x12\x011\n\x10\n\x0bc_subsample\x12\x01b\n\x0f\n\x04type\x12\x07counter\x1a\t\t\x00\x00\x00\x00\x00\x00y@' # Push to the pushgateway p.replace(registry) # Check the object that setted the server thread self.assertEqual("PUT", self.request['method']) self.assertEqual(valid_result, self.request['body'])