Update 'logbreak' to support messages

jacebrowning · Jan 11, 2019 · 9b5dad6 · 9b5dad6
1 parent 87c2281
commit 9b5dad6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -35,9 +35,12 @@ def create_tmp():
 
 @pytest.fixture
 def logbreak():
-    def _():
-        width = get_terminal_size().columns - 30
-        line = '-' * width
+    def _(message=""):
+        width = get_terminal_size().columns - 31
+        if message:
+            line = '-' * (width - len(message) - 1) + ' ' + message
+        else:
+            line = '-' * width
         log.info(line)
 
     return _

diff --git a/tests/test_auto.py b/tests/test_auto.py
@@ -61,7 +61,7 @@ def with_getattribute(logbreak, write, expect):
             item: b
             """,
         )
-        logbreak()
+        logbreak("Getting attribute")
 
         expect(sample.item) == 'b'
 
@@ -91,10 +91,10 @@ def with_iter(write, expect):
         expect([x for x in sample]) == [3]
 
     def describe_nesting():
-        def with_getattr(write, expect):
+        def with_getattr(logbreak, write, expect):
             sample = SampleWithNesting(1)
 
-            log.info("Modifying nested file")
+            logbreak("Modifying nested file")
             write(
                 'tmp/sample.yml',
                 """