From 6e444736abd502260c35db2adf69b86abd209990 Mon Sep 17 00:00:00 2001
From: Tim Byrne <sultan@locehilios.com>
Date: Mon, 13 Jan 2020 07:52:02 -0600
Subject: [PATCH] Tests for encrypt_with_checksums hooks

---
 test/test_hooks.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/test/test_hooks.py b/test/test_hooks.py
index f1df91e..d8b3ec1 100644
--- a/test/test_hooks.py
+++ b/test/test_hooks.py
@@ -62,7 +62,7 @@ def test_hook_env(runner, yadm_y, paths):
 
     # write the hook
     hook = paths.hooks.join(f'post_{cmd}')
-    hook.write('#!/bin/sh\nenv\n')
+    hook.write('#!/bin/bash\nenv\ndeclare\n')
     hook.chmod(0o755)
 
     run = runner(yadm_y(cmd, 'extra_args'))
@@ -74,9 +74,33 @@ def test_hook_env(runner, yadm_y, paths):
     # verify hook environment
     assert 'YADM_HOOK_EXIT=1\n' in run.out
     assert f'YADM_HOOK_COMMAND={cmd}\n' in run.out
+    assert f'YADM_HOOK_DIR={paths.yadm}\n' in run.out
     assert f'YADM_HOOK_FULL_COMMAND={cmd} extra_args\n' in run.out
     assert f'YADM_HOOK_REPO={paths.repo}\n' in run.out
     assert f'YADM_HOOK_WORK={paths.work}\n' in run.out
+    assert f'YADM_ENCRYPT_INCLUDE_FILES=\n' in run.out
+
+    # verify the hook environment contains certain exported functions
+    for func in [
+            'builtin_dirname',
+            'relative_path',
+            'unix_path',
+            'mixed_path',
+    ]:
+        assert f'BASH_FUNC_{func}' in run.out
+
+    # verify the hook environment contains the list of encrypted files
+    script = f"""
+        YADM_TEST=1 source {paths.pgm}
+        YADM_HOOKS="{paths.hooks}"
+        HOOK_COMMAND="{cmd}"
+        ENCRYPT_INCLUDE_FILES=(a b c)
+        invoke_hook "post"
+    """
+    run = runner(command=['bash'], inp=script)
+    assert run.success
+    assert run.err == ''
+    assert 'YADM_ENCRYPT_INCLUDE_FILES=a\nb\nc\n' in run.out
 
 
 def create_hook(paths, name, code):