adding unittests

tests/helpers.py

@@ -0,0 +1,65 @@
+"""
+author: deadc0de6 (https://github.com/deadc0de6)
+Copyright (c) 2017, deadc0de6
+helpers for the unittests
+"""
+
+import os
+import shutil
+import string
+import random
+import tempfile
+
+TMPSUFFIX = '.dotdrop'
+
+
+def clean(path):
+    '''Delete file or folder.'''
+    if not os.path.exists(path):
+        return
+    if os.path.isdir(path):
+        shutil.rmtree(path)
+    else:
+        os.remove(path)
+
+
+def get_string(length):
+    '''Get a random string of length "length".'''
+    alpha = string.ascii_uppercase + string.digits
+    return ''.join(random.choice(alpha) for _ in range(length))
+
+
+def get_tempfolder():
+    '''Get a temporary folder'''
+    return tempfile.mkdtemp(suffix=TMPSUFFIX)
+
+
+def create_random_file(folder, content=None):
+    '''Create a new file in folder with random content.'''
+    fname = get_string(8)
+    if content is None:
+        content = get_string(100)
+    path = os.path.join(folder, fname)
+    with open(path, 'w') as f:
+        f.write(content)
+    return path, content
+
+
+def create_dir(path):
+    '''Create a folder'''
+    os.mkdir(path)
+    return path
+
+
+def create_fake_config(folder, configname='config.yaml',
+                       dotpath='dotfiles', backup=True, create=True):
+    '''Create a fake config file'''
+    path = os.path.join(folder, configname)
+    with open(path, 'w') as f:
+        f.write('config:\n')
+        f.write('  backup: %s\n' % (str(backup)))
+        f.write('  create: %s\n' % (str(create)))
+        f.write('  dotpath: %s\n' % (dotpath))
+        f.write('dotfiles:\n')
+        f.write('profiles:\n')
+    return path