michaeljoseph / cookiecutter

        f.write("f = open('python_pre.txt', 'w')\n")

        f.write("f.close()\n")

    if sys.platform.startswith('win'):

        post = 'post_gen_project.bat'

        with open(os.path.join(hook_dir, post), 'w') as f:

            f.write("@echo off\n")

            f.write("\n")

            f.write("echo post generation hook\n")

            f.write("echo. >shell_post.txt\n")

    else:

        post = 'post_gen_project.sh'

        filename = os.path.join(hook_dir, post)

        with open(filename, 'w') as f:

            f.write("#!/bin/bash\n")

            f.write("\n")

            f.write("echo 'post generation hook';\n")

            f.write("touch 'shell_post.txt'\n")

        # Set the execute bit

        os.chmod(filename, os.stat(filename).st_mode | stat.S_IXUSR)

    return post

    def test_run_hook(self):

        """Execute hook from specified template in specified output

        directory.

        """

        tests_dir = os.path.join(self.repo_path, 'input{{hooks}}')

        with utils.work_in(self.repo_path):

            hooks.run_hook('pre_gen_project', tests_dir, {})

            assert os.path.isfile(os.path.join(tests_dir, 'python_pre.txt'))

            hooks.run_hook('post_gen_project', tests_dir, {})

            assert os.path.isfile(os.path.join(tests_dir, 'shell_post.txt'))

    def test_run_failing_hook(self):

        hook_path = os.path.join(self.hooks_path, 'pre_gen_project.py')

        tests_dir = os.path.join(self.repo_path, 'input{{hooks}}')

        with open(hook_path, 'w') as f:

            f.write("#!/usr/bin/env python\n")

            f.write("import sys; sys.exit(1)\n")