OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	05/14/2024 03:19:10 PM	rwxr-xr-x
📄 __init__.py	479 bytes	05/14/2024 03:19:10 PM	rw-r--r--
📁 __pycache__	-	05/14/2024 03:19:10 PM	rwxr-xr-x
📄 helper.py	1.74 KB	05/14/2024 03:19:10 PM	rw-r--r--
📄 test_chunked_logging.py	1.26 KB	05/14/2024 03:19:10 PM	rw-r--r--
📄 test_common_logging.py	827 bytes	05/14/2024 03:19:10 PM	rw-r--r--
📄 test_debugging_fields.py	1.96 KB	05/14/2024 03:19:10 PM	rw-r--r--
📄 test_extra_fields.py	2.2 KB	05/14/2024 03:19:10 PM	rw-r--r--
📄 test_status_issue.py	1.93 KB	05/14/2024 03:19:10 PM	rw-r--r--

Editing: test_chunked_logging.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""pytests sending logs to a local Graylog instance that need to be chunked"""

import logging

import pytest

from graypy.handler import (
    SYSLOG_LEVELS,
    GELFUDPHandler,
    GELFWarningChunker,
    BaseGELFChunker,
    GELFTruncatingChunker,
)

from tests.helper import TEST_UDP_PORT
from tests.integration import LOCAL_GRAYLOG_UP
from tests.integration.helper import get_unique_message, get_graylog_response

@pytest.mark.parametrize(
    "gelf_chunker", [BaseGELFChunker, GELFWarningChunker, GELFTruncatingChunker]
)
@pytest.mark.skipif(not LOCAL_GRAYLOG_UP, reason="local Graylog instance not up")
def test_chunked_logging(gelf_chunker):
    """Test sending a log that requires chunking to be fully sent"""
    logger = logging.getLogger("test_chunked_logger")
    handler = GELFUDPHandler(
        "127.0.0.1", TEST_UDP_PORT, gelf_chunker=gelf_chunker(chunk_size=10)
    )
    logger.addHandler(handler)
    message = get_unique_message()
    logger.error(message)
    graylog_response = get_graylog_response(message)
    assert message == graylog_response["message"]
    assert "long_message" not in graylog_response
    assert "timestamp" in graylog_response
    assert SYSLOG_LEVELS[logging.ERROR] == graylog_response["level"]