allow to define header filenames via env vars

Author: mr-mixas

nested_diff/diff_tool.py

@@ -15,6 +15,7 @@
 """Nested diff command line tool."""
 
 import argparse
+import os
 import sys
 
 import nested_diff
@@ -97,8 +98,17 @@ def generate_diffs(self):
 
                 equal, diff = self.diff(a['data'], b['data'])
 
+                try:
+                    name_a = os.environ['HEADER_NAME_A']
+                    name_b = os.environ['HEADER_NAME_B']
+                except KeyError:
+                    name_a = a['name']
+                    name_b = b['name']
+                else:
+                    headers_enabled = True
+
                 if headers_enabled:
-                    header = self.dumper.get_diff_header(a['name'], b['name'])
+                    header = self.dumper.get_diff_header(name_a, name_b)
 
                 a = b
 

tests/cli/test_diff_tool.py

@@ -629,6 +629,40 @@ def test_show_several_args_tty(capsys, rpath, stringio_tty):
     assert stringio_tty.getvalue().startswith('\033[33m--- /dev/null')
 
 
+@mock.patch('os.environ', {'HEADER_NAME_A': 'foo', 'HEADER_NAME_B': 'bar'})
+def test_header_names_from_env(capsys, rpath):
+    app = nested_diff.diff_tool.App(
+        args=(
+            rpath('shared.lists.a.json'),
+            rpath('shared.lists.b.json'),
+        ),
+    )
+    exit_code = app.run()
+
+    captured = capsys.readouterr()
+    assert captured.err == ''
+    assert exit_code == 1
+
+    assert captured.out.startswith('--- foo\n+++ bar\n')
+
+
+@mock.patch('os.environ', {'HEADER_NAME_A': 'foo'})
+def test_header_names_from_env_partial(capsys, rpath):
+    app = nested_diff.diff_tool.App(
+        args=(
+            rpath('shared.lists.a.json'),
+            rpath('shared.lists.b.json'),
+        ),
+    )
+    exit_code = app.run()
+
+    captured = capsys.readouterr()
+    assert captured.err == ''
+    assert exit_code == 1
+
+    assert not captured.out.startswith('--- ')
+
+
 def test_values_none(capsys, expected, rpath):
     exit_code = nested_diff.diff_tool.App(
         args=(