def generate_add_events_request(num_events,
line_length,
attributes_count,
base_body=None):
# type: (int, int, int, Optional[dict]) -> AddEventsRequest
"""
Generate AddEventsRequest object with the num_events number of events with the same payload
payload (line length) size.
"""
if not base_body:
base_body = {
"token": "api key",
"session": "sessionbar",
"threads": []
}
add_events_request = AddEventsRequest(base_body=base_body)
for _ in range(0, num_events):
line = generate_random_line(length=line_length)
attributes = generate_random_dict(keys_count=attributes_count)
event = Event(thread_id=100)
event.set_message(line)
event.add_attributes(attributes)
add_events_request.add_event(event)
return add_events_request
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(),
b"""{"token":"fakeToken", events: [{attrs:{"event":"event","source":"stdin",message:`s\x00\x00\x00\x08eventOne},ts:"1"}], """
b"""logs: [{"attrs":{},"id":"log2"}], threads: [{"id":"log2","name":"Log two"}], client_time: 1 }""",
)
request.close()
def test_create_from_template_with_add_attributes(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(42)
x.add_attributes({"trigger_update": "yes"})
output_buffer = BytesIO()
x.serialize(output_buffer)
self.assertEquals(
b'{thread:"foo", log:"foo", attrs:{"trigger_update":"yes",message:`s\x00\x00\x00\nmy_message,sample_rate:0.5},ts:"42",si:"1",sn:2,sd:3}',
output_buffer.getvalue(),
)
