mirror of
https://github.com/cookiecutter/cookiecutter-django.git
synced 2024-11-14 05:36:58 +03:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import os
|
|
import shutil
|
|
import unittest
|
|
from os.path import exists, dirname, join
|
|
|
|
import sh
|
|
|
|
from cookiecutter.main import cookiecutter
|
|
|
|
|
|
class DjangoCookieTestCase(unittest.TestCase):
|
|
|
|
root_dir = dirname(dirname(__file__))
|
|
ctx = {}
|
|
destpath = None
|
|
|
|
def generate_project(self, extra_context=None):
|
|
ctx = {
|
|
"project_name": "My Test Project",
|
|
"repo_name": "my_test_project",
|
|
"author_name": "Test Author",
|
|
"email": "test@example.com",
|
|
"description": "A short description of the project.",
|
|
"domain_name": "example.com",
|
|
"version": "0.1.0",
|
|
"timezone": "UTC",
|
|
"now": "2015/01/13",
|
|
"year": "2015"
|
|
}
|
|
if extra_context:
|
|
assert isinstance(extra_context, dict)
|
|
ctx.update(extra_context)
|
|
|
|
self.ctx = ctx
|
|
self.destpath = join(self.root_dir, self.ctx['repo_name'])
|
|
|
|
cookiecutter(template='./', checkout=None, no_input=True, extra_context=ctx)
|
|
|
|
# Build a list containing absolute paths to the generated files
|
|
paths = [os.path.join(dirpath, file_path)
|
|
for dirpath, subdirs, files in os.walk(self.destpath)
|
|
for file_path in files]
|
|
return paths
|
|
|
|
def clean(self):
|
|
if exists(self.destpath):
|
|
shutil.rmtree(self.destpath)
|
|
sh.cd(self.root_dir)
|
|
|
|
def tearDown(self):
|
|
self.clean()
|