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(), )