cee-formatter/test_format.py at master · jcarrano/cee-formatter · GitHub

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
import datetime
import json
import logging

from cee_formatter import CEEFormatter

try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO


def test_datetime_format():
    stream = StringIO()

    logger = logging.getLogger('test')

    handler = logging.StreamHandler(stream)
    formatter = CEEFormatter()
    handler.setFormatter(formatter)

    logger.addHandler(handler)

    date = datetime.datetime.now()

    logger.error('test', extra={
        'd': date,
    })

    value = stream.getvalue()

    assert value.startswith('@cee: {')

    json_value = value.replace('@cee: ', '')

    json_dict = json.loads(json_value)

    assert json_dict['d'] == date.isoformat()