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