def __init__(self, *args, **kwargs): # original call if len(args) == 0 and "agent_url" not in kwargs: kwargs["agent_url"] = "http://localhost:8126" super(DummyWriter, self).__init__(*args, **kwargs) self.spans = [] self.traces = [] self.json_encoder = JSONEncoder() self.msgpack_encoder = Encoder(4 << 20, 4 << 20)
def test_init(self): """ When calling `Payload.init` With an encoder We use that encoder With no encoder We use the default encoder """ default_encoder_type = type(Encoder()) payload = Payload() self.assertIsInstance(payload.encoder, default_encoder_type) json_encoder = JSONEncoder() payload = Payload(encoder=json_encoder) self.assertEqual(payload.encoder, json_encoder)
def _init_encoder(): return Encoder()
import sys from ddtrace.internal.encoding import Encoder from ddtrace.internal.encoding import MsgpackEncoderV03 from ddtrace.internal.encoding import MsgpackEncoderV05 from tests.tracer.test_encoders import gen_trace ENCODERS = { "v0.3": MsgpackEncoderV03, "v0.5": MsgpackEncoderV05, } try: encoder = ENCODERS.get(sys.argv[1], Encoder)(8 << 20, 8 << 20) except IndexError: encoder = Encoder(8 << 20, 8 << 20) trace = gen_trace(nspans=1000) for i in range(200): encoder.put(trace) encoder.encode()
def _init_encoder(max_size=8 << 20, max_item_size=8 << 20): return Encoder(max_size, max_item_size)