Add tests for exec_hook, exec_prebuild, exec_postbuild.

tests/test_compose.py

@@ -0,0 +1,95 @@
+from focker.compose import exec_hook, \
+    exec_prebuild, \
+    exec_postbuild
+from tempfile import TemporaryDirectory
+import os
+import pytest
+import fcntl
+from focker.misc import focker_lock, \
+    focker_unlock
+import inspect
+import ast
+
+
+def test_exec_hook_01():
+    spec = [
+        'touch test-exec-hook-01',
+        'touch test-exec-hook-02'
+    ]
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+        assert os.path.exists(os.path.join(d, 'test-exec-hook-01'))
+        assert os.path.exists(os.path.join(d, 'test-exec-hook-02'))
+    assert not os.path.exists(d)
+
+
+def test_exec_hook_02():
+    spec = 'touch test-exec-hook-01 && touch test-exec-hook-02'
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+        assert os.path.exists(os.path.join(d, 'test-exec-hook-01'))
+        assert os.path.exists(os.path.join(d, 'test-exec-hook-02'))
+    assert not os.path.exists(d)
+
+
+@pytest.mark.xfail(raises=ValueError, strict=True)
+def test_exec_hook_03a():
+    spec = 1
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+
+
+@pytest.mark.xfail(raises=TypeError, strict=True)
+def test_exec_hook_03b():
+    spec = [1]
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+
+
+@pytest.mark.xfail(raises=FileNotFoundError, strict=True)
+def test_exec_hook_04():
+    spec = 'ls'
+    exec_hook(spec, '/non-existent-directory/wcj20fy103', 'test-exec-hook')
+
+
+def test_exec_hook_05():
+    spec = 'ls'
+    oldwd = os.getcwd()
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+    assert os.getcwd() == oldwd
+
+
+@pytest.mark.xfail(raises=RuntimeError, strict=True)
+def test_exec_hook_06():
+    spec = '/non-existent-command/hf249h'
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+
+
+def test_exec_hook_07():
+    os.chdir('/')
+    spec = 'flock --nonblock /var/lock/focker.lock -c ls'
+    focker_lock()
+    assert fcntl.flock(focker_lock.fd, fcntl.LOCK_EX | fcntl.LOCK_NB) != 0
+    with TemporaryDirectory() as d:
+        exec_hook(spec, d, 'test-exec-hook')
+    assert fcntl.flock(focker_lock.fd, fcntl.LOCK_EX | fcntl.LOCK_NB) != 0
+    focker_unlock()
+
+
+def _test_simple_forward(fun, fwd_fun_name='exec_hook'):
+    src = inspect.getsource(fun)
+    mod = ast.parse(src)
+    assert isinstance(mod.body[0], ast.FunctionDef)
+    assert isinstance(mod.body[0].body[0], ast.Return)
+    assert isinstance(mod.body[0].body[0].value, ast.Call)
+    assert mod.body[0].body[0].value.func.id == fwd_fun_name
+
+
+def test_exec_prebuild():
+    _test_simple_forward(exec_prebuild)
+
+
+def test_exec_postbuild():
+    _test_simple_forward(exec_postbuild)