test: add tests for logfmt parser

Author: spaceone

tests/test_formatter.py

@@ -7,7 +7,7 @@
 
 import pytest
 
-from logfmter.formatter import Logfmter
+from logfmter.formatter import Logfmter, parse_logfmt
 
 STRING_ESCAPE_RULES = [
     # If the string contains a space, then it must be quoted.
@@ -91,6 +91,108 @@ def test_format_params(value, expected):
     assert Logfmter.format_params(value) == expected
 
 
+@pytest.mark.parametrize(
+    "value",
+    [
+        {"at": "INFO"},
+        {"levelname": "INFO", "a": "1"},
+        {"at": "INFO", "msg": "test", "a": "1"},
+        {"at": "INFO", "msg": "="},
+        {"at": "INFO", "msg": "alpha", "exc_info": 'exc\n"info"\ntb'},
+        {"a": " "},
+        {"_": " "},
+        {"a": '"'},
+        {"a": "\\"},
+        {"a": "\\", "b": "c"},
+        {"a": "\n"},
+        {"a": "", "b": "c"},
+        {"a": "1", "b": "2"},
+        {"a": "1.2", "b": "2"},
+        {"a": "1.2.3", "b": "0.2", "c": "1.0", "d": ".", "e": "1..2"},
+        {"foo.bar": "baz", "foo.blah": "blub"},
+    ],
+)
+def test_parse_logfmt_round_trip(value):
+    assert parse_logfmt(Logfmter.format_params(value)) == value
+
+
+@pytest.mark.parametrize(
+    "value",
+    [
+        {"a": 1},
+        {"a": 1, "b": 2},
+        {"a": 1.2, "b": 2},
+        {"a": "1.2.3", "b": 0.2, "c": 1.0, "d": ".", "e": "1..2"},
+    ],
+)
+def test_parse_logfmt_numeric_round_trip(value):
+    assert parse_logfmt(Logfmter.format_params(value), convert_numeric=True) == value
+
+
+@pytest.mark.parametrize(
+    "value,expected,kwargs",
+    [
+        ("at=INFO", {"at": "INFO"}, {}),
+        ('at="INFO"', {"at": "INFO"}, {"reverse": False}),
+        (
+            "at=INFO a=1",
+            {"levelname": "INFO", "a": "1"},
+            {"aliases": {"at": "levelname"}, "reverse": False},
+        ),
+        ("at=INFO msg=test a=1", {"at": "INFO", "msg": "test", "a": "1"}, {}),
+        ('at=INFO msg="="', {"at": "INFO", "msg": "="}, {}),
+        (
+            "at=INFO first_name=josh",
+            {"at": "INFO", "first name": "josh"},
+            {"aliases": {"first_name": "first name"}, "reverse": False},
+        ),
+        (
+            r'at=INFO msg=alpha exc_info="exc\n\"info\"\ntb"',
+            {"at": "INFO", "msg": "alpha", "exc_info": 'exc\n"info"\ntb'},
+            {},
+        ),
+        ('a=" "', {"a": " "}, {}),
+        ('_=" "', {"_": " "}, {}),
+        ('a="\\""', {"a": '"'}, {}),
+        (r'a="\\"', {"a": "\\"}, {"reverse": False}),
+        (r'a="\n"', {"a": "\n"}, {}),
+        (r"a=\n", {"a": "n"}, {"reverse": False}),
+        ("a= b=c", {"a": "", "b": "c"}, {}),
+        ("a b=c", {"a": "", "b": "c"}, {"reverse": False}),
+        ("a=1", {"a": 1}, {"convert_numeric": True}),
+        ("a=1 b=2", {"a": "1", "b": "2"}, {}),
+        ("a=1 b=2", {"a": 1, "b": 2}, {"convert_numeric": True}),
+        ("a=1.2 b=2", {"a": "1.2", "b": "2"}, {}),
+        ("a=1.2 b=2", {"a": 1.2, "b": 2}, {"convert_numeric": True}),
+        (
+            "a=1.2.3 b=.2 c=1. d=. e=1..2",
+            {"a": "1.2.3", "b": 0.2, "c": 1.0, "d": ".", "e": "1..2"},
+            {"convert_numeric": True, "reverse": False},
+        ),
+        ("foo.bar=baz foo.blah=blub", {"foo.bar": "baz", "foo.blah": "blub"}, {}),
+        ("\\n=foo", {"n": "foo"}, {"reverse": False}),
+        ("a=' '", {"'": "", "a": "'"}, {"reverse": False}),
+    ],
+)
+def test_parse_logfmt(value, kwargs, expected):
+    reverse = kwargs.pop("reverse", True)
+    assert parse_logfmt(value, **kwargs) == expected
+    if reverse:
+        assert Logfmter.format_params(expected) == value
+
+
+@pytest.mark.parametrize(
+    "value",
+    [
+        "a=\n",
+        pytest.param("a = b", marks=pytest.mark.xfail()),
+    ],
+)
+def test_parse_invalid_logfmt(value):
+    with pytest.raises(ValueError):
+        print(parse_logfmt(value))
+
+
 @pytest.mark.parametrize(
     "value,expected",
     [