Merge pull request #184 from ocefpaf/test_mode

Test if mode is unique

Author: lsetiawan

test/test_cli.py

@@ -6,7 +6,7 @@
 
 import pytest
 
-from wof.wofpy_config import makedirs
+from wof.wofpy_config import cli, makedirs
 
 
 @pytest.fixture(scope='function')
@@ -32,10 +32,16 @@ def test_makedirs_overwrite_soft():
     assert _stat.st_ino == stat.st_ino
 
 
-@pytest.mark.skipif(sys.platform == 'win32',
-                    reason='st_ino does not work on Windows')
-def test_makedirs_overwrite_hard():
-    _directory, _stat = mkdtemp().next()
-    makedirs(_directory, overwrite='hard')
-    stat = os.stat(_directory)
-    assert _stat.st_ino != stat.st_ino
+
+def test_cli_overwrite_hard():
+    _directory, _ = mkdtemp().next()
+    args = {
+        'INDIR': _directory,
+        '--mode': 'production',
+        '--overwrite': 'hard'
+    }
+    cli(args)
+    assert os.path.isdir(os.path.join(_directory, 'production_configs'))
+    args.update({'--mode': 'development'})
+    cli(args)
+    assert not os.path.isdir(os.path.join(_directory, 'production_configs'))

wof/wofpy_config.py

@@ -64,15 +64,21 @@ def copytree(src, dst, symlinks=False, ignore=None, overwrite='soft'):
             shutil.copy2(s, d)
 
 
-def main():
+def parse_args():
     args = docopt(__doc__, version='1.0.0')
+    return args
+
+
+def cli(args):
     directory = args.get('INDIR')
     mode = args.get('--mode').lower()
+    overwrite = args.get('--overwrite').lower()
+
     if mode not in ['development', 'production']:
         raise ValueError(
             'Got mode: {!r}, expected development, or production.'.format(mode)
             )
-    overwrite = args.get('--overwrite').lower()
+
     if overwrite not in ['hard', 'soft']:
         raise ValueError(
             'Got overwrite: {!r}, expected hard or soft.'.format(overwrite)
@@ -89,5 +95,9 @@ def main():
         copytree(_CONFIG, _PRODUCTION, overwrite=overwrite)
 
 
+def main():
+    args = parse_args()
+    cli(args)
+
 if __name__ == '__main__':
     main()