def setUp(self): self.service_name = 'dummy_service' self.initial_trace_id = 300 # greater than 256 to not use cached ints self.tx = Tx(self.tx_tree, Mock(), 1, CocaineHeaders(), self.service_name, trace_id=self.initial_trace_id) self.initial_log = self.tx.log
def test_extra_static_values(): h = CocaineHeaders() assert len(CocaineHeaders.STATIC_TABLE) == 82, len( CocaineHeaders.STATIC_TABLE) assert h.get_by_index(80) == ( b'trace_id', b'\x00\x00\x00\x00\x00\x00\x00\x00'), h.get_by_index(80) assert h.get_by_index(81) == (b'span_id', b'\x00\x00\x00\x00\x00\x00\x00\x00') assert h.get_by_index(82) == (b'parent_id', b'\x00\x00\x00\x00\x00\x00\x00\x00')
def setUp(self): self.service_name = 'dummy_service' self.tx = Tx(self.tx_tree, Mock(), 1, CocaineHeaders(), self.service_name) self.initial_log = self.tx.log assert not isinstance(self.initial_log, TraceAdapter)
def test_extra_static_values(): h = CocaineHeaders() assert len(CocaineHeaders.STATIC_TABLE) == 82, len(CocaineHeaders.STATIC_TABLE) assert h.get_by_index(80) == (b'trace_id', b'\x00\x00\x00\x00\x00\x00\x00\x00'), h.get_by_index(80) assert h.get_by_index(81) == (b'span_id', b'\x00\x00\x00\x00\x00\x00\x00\x00') assert h.get_by_index(82) == (b'parent_id', b'\x00\x00\x00\x00\x00\x00\x00\x00')
def test_stream_timeout(): io = IOLoop.current() stream = Stream(None, CocaineHeaders()) io.run_sync(lambda: stream.get(timeout=0.5))
def test_stream(): io = IOLoop.current() stream = Stream(None, CocaineHeaders()) stream.error((0, 100), "TESTERROR", None) io.run_sync(stream.get)
def test_tx_on_done(self): io = IOLoop.current() tx = Tx(self.tx_tree, self.PipeMock(), 1, CocaineHeaders(), self.service_name) tx.done() io.run_sync(tx.get, timeout=1)
def test_tx(self): tx = Tx(self.tx_tree, self.PipeMock(), 1, CocaineHeaders(), self.service_name) tx.dummy().wait(4) tx.failed().wait(4)
def test_print(self): log.info( Tx(self.tx_tree, self.PipeMock(), 1, CocaineHeaders(), self.service_name))