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
|
import os
import pytest
import logbook
from .utils import require_module
@require_module("win32con")
@require_module("win32evtlog")
@require_module("win32evtlogutil")
@pytest.mark.skipif(
os.environ.get("ENABLE_LOGBOOK_NTEVENTLOG_TESTS") is None,
reason="Don't clutter NT Event Log unless enabled.",
)
def test_nteventlog_handler():
from win32con import (
EVENTLOG_ERROR_TYPE,
EVENTLOG_INFORMATION_TYPE,
EVENTLOG_WARNING_TYPE,
)
from win32evtlog import (
EVENTLOG_BACKWARDS_READ,
EVENTLOG_SEQUENTIAL_READ,
OpenEventLog,
ReadEventLog,
)
from win32evtlogutil import SafeFormatMessage
logger = logbook.Logger("Test Logger")
with logbook.NTEventLogHandler("Logbook Test Suite"):
logger.info("The info log message.")
logger.warning("The warning log message.")
logger.error("The error log message.")
def iter_event_log(handle, flags, offset):
while True:
events = ReadEventLog(handle, flags, offset)
yield from events
if not events:
break
handle = OpenEventLog(None, "Application")
flags = EVENTLOG_BACKWARDS_READ | EVENTLOG_SEQUENTIAL_READ
for event in iter_event_log(handle, flags, 0):
source = str(event.SourceName)
if source == "Logbook Test Suite":
message = SafeFormatMessage(event, "Application")
if "Message Level: INFO" in message:
assert "The info log message" in message
assert event.EventType == EVENTLOG_INFORMATION_TYPE
if "Message Level: WARNING" in message:
assert "The warning log message" in message
assert event.EventType == EVENTLOG_WARNING_TYPE
if "Message Level: ERROR" in message:
assert "The error log message" in message
assert event.EventType == EVENTLOG_ERROR_TYPE
|
