[generator] Test cases for init_basic and save_id

ref: #230

couchapp/generator.py

@@ -13,6 +13,7 @@
 from couchapp.errors import AppError
 from couchapp import localdoc
 from couchapp.util import is_py2exe, is_windows, relpath, setup_dir, user_path
+from couchapp.util import setup_dirs
 
 __all__ = ["init_basic", "init_template", "generate_function", "generate"]
 
@@ -40,16 +41,9 @@ def init_basic(path):
  views/
  '''
  setup_dir(path, require_empty=True)
+ setup_dirs(os.path.join(path, n) for n in DEFAULT_APP_TREE)
 
- for n in DEFAULT_APP_TREE:
- tp = os.path.join(path, n)
- os.makedirs(tp)
-
- fid = os.path.join(path, '_id')
- if not os.path.isfile(fid):
- with open(fid, 'wb') as f:
- f.write('_design/{0}'.format(os.path.split(path)[1]))
-
+ save_id(path, '_design/{0}'.format(os.path.split(path)[-1]))
  localdoc.document(path, create=True)
 
 

tests/test_generator.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+
+from couchapp.generator import init_basic, save_id
+
+from mock import patch
+
+
+@patch('couchapp.generator.save_id')
+@patch('couchapp.generator.setup_dirs')
+@patch('couchapp.generator.setup_dir')
+@patch('couchapp.generator.localdoc')
+def test_init_basic(localdoc, setup_dir, setup_dirs, save_id):
+ init_basic('/mock/app')
+
+ assert setup_dir.called
+ assert setup_dirs.called
+ assert save_id.called
+ assert localdoc.document.called
+
+
+@patch('couchapp.generator.open')
+def test_save_id(open_):
+ save_id('/mock/app', 'someid')
+
+ open_.assert_called_with('/mock/app/_id', 'wb')