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)