Exemplo n.º 1
0
def test_append_events():
    request_data = Queue()

    def handler(request, data):
        request_data.put((request.__dict__, data))

    with testutil.http_server(handler) as url:
        client = AnalyticsClient(url)

        e1 = new_event({"val": 1})
        e2 = new_event({"val": 2})
        e3 = new_event({"val": 3})

        client.append_events([e1, e2])  # batch 1
        client.append_events([e3])  # batch 2

        request1, data1 = request_data.get(timeout=2)
        request2, data2 = request_data.get(timeout=2)

    assert request_data.qsize() == 0

    # assert that http request/payload is correct
    assert request1["path"] == request2["path"] == "/events"

    doc1 = json.loads(data1)
    doc2 = json.loads(data2)
    assert isinstance(doc1["events"], list)
    assert len(doc1["events"]) == 2
    assert isinstance(doc2["events"], list)
    assert len(doc2["events"]) == 1

    # assert headers are set
    assert request1["headers"]["Localstack-Session-Id"] == get_session_id()
    assert request1["headers"][
        "User-Agent"] == f"localstack/{constants.VERSION}"

    # assert content is correct
    e1 = doc1["events"][0]
    e2 = doc1["events"][1]
    e3 = doc2["events"][0]

    assert e1["name"] == "test"
    assert e2["name"] == "test"
    assert e3["name"] == "test"

    assert e1["metadata"]["session_id"] == get_session_id()
    assert e2["metadata"]["session_id"] == get_session_id()
    assert e3["metadata"]["session_id"] == get_session_id()

    assert e1["payload"]["val"] == 1
    assert e2["payload"]["val"] == 2
    assert e3["payload"]["val"] == 3
Exemplo n.º 2
0
def new_event(payload=None) -> Event:
    return Event(
        "test",
        EventMetadata(get_session_id(), str(datetime.datetime.now())),
        payload=payload,
    )
Exemplo n.º 3
0
 def _do_get_session_id():
     calls.put(get_session_id())
Exemplo n.º 4
0
def test_logger_uses_default_session_id(collector):
    log = EventLogger(collector)
    log.event("foo")

    assert log.session_id == get_session_id()
    assert collector.next().metadata.session_id == get_session_id()