def test_all_fields(self):
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message("my_message")
x.set_sampling_rate(0.5)
x.set_sequence_id(1)
x.set_sequence_number(2)
x.set_sequence_number_delta(3)
x.set_timestamp(42L)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message,sample_rate:0.5},ts:"42",si:"1",sn:2,sd:3}',
output_buffer.getvalue(),
)
self.assertEquals(
json_lib.JsonObject(
thread="foo",
ts="42",
si="1",
sn=2,
sd=3,
attrs=json_lib.JsonObject(parser="bar", message="my_message", sample_rate=0.5),
),
json_lib.parse(output_buffer.getvalue()),
)
def test_individual_fields(self):
# snd
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message('my_message')
x.set_sequence_number_delta(3)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},sd:3}',
output_buffer.getvalue())
# timestamp
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message('my_message')
x.set_timestamp(42)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},ts:"42"}',
output_buffer.getvalue())
# sampling_rate
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message('my_message')
x.set_sampling_rate(0.5)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message,sample_rate:0.5}}',
output_buffer.getvalue())
# sid
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message('my_message')
x.set_sequence_id('hi')
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},si:"hi"}',
output_buffer.getvalue())
# seq num
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message('my_message')
x.set_sequence_number(5)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},sn:5}',
output_buffer.getvalue())
def test_only_message(self):
x = Event()
x.set_message("my_message")
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals("{attrs:{message:`s\x00\x00\x00\nmy_message}}", output_buffer.getvalue())
def test_fast_path_fields(self):
x = Event(thread_id='foo', attrs={"parser": "bar"})
x.set_message("my_message")
x.set_sequence_number_delta(3)
x.set_timestamp(42L)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},sd:3,ts:"42"}',
output_buffer.getvalue())
def test_no_attrs(self):
x = Event(thread_id="biz")
x.set_message("my_message")
x.set_sampling_rate(0.5)
x.set_sequence_id(1)
x.set_sequence_number(2)
x.set_sequence_number_delta(3)
x.set_timestamp(42L)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"biz", attrs:{message:`s\x00\x00\x00\nmy_message,sample_rate:0.5},ts:"42",si:"1",sn:2,sd:3}',
output_buffer.getvalue())
def test_create_from_template(self):
x = Event(thread_id='foo', attrs={"parser": "bar"})
x = Event(base=x)
x.set_message("my_message")
x.set_sampling_rate(0.5)
x.set_sequence_id(1)
x.set_sequence_number(2)
x.set_sequence_number_delta(3)
x.set_timestamp(42L)
output_buffer = StringIO()
x.serialize(output_buffer)
self.assertEquals(
'{thread:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message,sample_rate:0.5},ts:"42",si:"1",sn:2,sd:3}',
output_buffer.getvalue())
def test_set_log_line_attributes_with_base_attributes(self):
request = AddEventsRequest(self.__body)
request.set_client_time(1)
request.add_log_and_thread("log2", "Log two", {})
event_base = Event()
event_base.add_attributes(
{"source": "stdout", "base": "base"}, overwrite_existing=False
)
event_one = Event(base=event_base)
event_one.set_message("eventOne")
event_one.add_attributes(
{"source": "stdin", "event": "event"}, overwrite_existing=True
)
self.assertTrue(request.add_event(event_one, timestamp=1))
self.assertEquals(
request.get_payload(),
"""{"token":"fakeToken", events: [{attrs:{"event":"event","source":"stdin",message:`s\x00\x00\x00\x08eventOne},ts:"1"}], """
"""logs: [{"attrs":{},"id":"log2"}], threads: [{"id":"log2","name":"Log two"}], client_time: 1 }""",
)
request.close()
def test_individual_fields(self):
# snd
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message(b"my_message")
x.set_sequence_number_delta(3)
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},sd:3}',
output_buffer.getvalue(),
)
# timestamp
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message("my_message")
x.set_timestamp(42)
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},ts:"42"}',
output_buffer.getvalue(),
)
# sampling_rate
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message("my_message")
x.set_sampling_rate(0.5)
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message,sample_rate:0.5}}',
output_buffer.getvalue(),
)
# sid
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message("my_message")
x.set_sequence_id("hi")
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},si:"hi"}',
output_buffer.getvalue(),
)
# seq num
x = Event(thread_id="foo", attrs={"parser": "bar"})
x.set_message("my_message")
x.set_sequence_number(5)
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"parser":"bar",message:`s\x00\x00\x00\nmy_message},sn:5}',
output_buffer.getvalue(),
)
