Added tests for comment parsing

yandex · May 2, 2017 · 5dfcd5e · 5dfcd5e
1 parent 521d4e3
commit 5dfcd5e
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/tests/parser/test_raw_parser.py b/tests/parser/test_raw_parser.py
@@ -463,6 +463,35 @@ def test_file_delims():
 
     assert_config(config, expected)
 
+
+def test_comments():
+    config = '''
+# Some comment
+add_header X-Some-Comment some;
+
+# 
+# Comment with padding
+# 
+add_header X-Padding-Comment padding;
+
+#
+add_header X-Blank-Comment blank;
+        '''
+
+    expected = [
+        ['Some comment'],
+        ['add_header', 'X-Some-Comment', 'some'],
+        [''],
+        ['Comment with padding'],
+        [''],
+        ['add_header', 'X-Padding-Comment', 'padding'],
+        [''],
+        ['add_header', 'X-Blank-Comment', 'blank'],
+    ]
+
+    assert_config(config, expected)
+
+
 def assert_config(config, expected):
     with mock.patch('%s.open' % builtins.__name__) as mock_open:
         mock_open.return_value = StringIO(config)