1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""pytests for validating the addition of extra fields within GELF logs"""
import logging
import pytest
from graypy import GELFTLSHandler, GELFTCPHandler, GELFUDPHandler, GELFHTTPHandler
from tests.helper import (
TEST_CERT,
TEST_TCP_PORT,
TEST_HTTP_PORT,
TEST_TLS_PORT,
TEST_UDP_PORT,
)
from tests.integration import LOCAL_GRAYLOG_UP
from tests.integration.helper import get_unique_message, get_graylog_response
class DummyFilter(logging.Filter):
def filter(self, record):
record.ozzy = "diary of a madman"
record.van_halen = 1984
record.id = 42
return True
@pytest.fixture(
params=[
GELFTCPHandler("127.0.0.1", TEST_TCP_PORT, extra_fields=True),
GELFUDPHandler("127.0.0.1", TEST_UDP_PORT, extra_fields=True),
GELFUDPHandler("127.0.0.1", TEST_UDP_PORT, compress=False, extra_fields=True),
GELFHTTPHandler("127.0.0.1", TEST_HTTP_PORT, extra_fields=True),
GELFHTTPHandler("127.0.0.1", TEST_HTTP_PORT, compress=False, extra_fields=True),
GELFTLSHandler("127.0.0.1", TEST_TLS_PORT, extra_fields=True),
GELFTLSHandler(
"127.0.0.1",
TEST_TLS_PORT,
validate=True,
ca_certs=TEST_CERT,
extra_fields=True,
),
]
)
def handler(request):
return request.param
@pytest.yield_fixture
def logger(handler):
logger = logging.getLogger("test")
dummy_filter = DummyFilter()
logger.addFilter(dummy_filter)
logger.addHandler(handler)
yield logger
logger.removeHandler(handler)
logger.removeFilter(dummy_filter)
@pytest.mark.skipif(not LOCAL_GRAYLOG_UP, reason="local Graylog instance not up")
def test_dynamic_fields(logger):
message = get_unique_message()
logger.error(message)
graylog_response = get_graylog_response(message, fields=["ozzy", "van_halen"])
assert message == graylog_response["message"]
assert "long_message" not in graylog_response
assert "timestamp" in graylog_response
assert "diary of a madman" == graylog_response["ozzy"]
assert 1984 == graylog_response["van_halen"]
assert 42 != graylog_response["_id"]
assert "id" not in graylog_response
|
