mirrors/dotdrop
1
0
Fork 0
mirror of https://github.com/deadc0de6/dotdrop.git synced 2026-02-04 17:24:46 +00:00
Code Activity

improve tests

Browse Source
This commit is contained in:
deadc0de6
2018-11-15 18:45:57 +01:00
parent ba31a9ad93
commit 0d407337ea
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/helpers.py
View File

@@ -38,7 +38,8 @@ def get_tempdir():
return tempfile.mkdtemp(suffix=TMPSUFFIX)
def create_random_file(directory, content=None, binary=False):
def create_random_file(directory, content=None,
binary=False, template=False):
'''Create a new file in directory with random content.'''
fname = get_string(8)
mode = 'w'
@@ -46,9 +47,15 @@ def create_random_file(directory, content=None, binary=False):
mode = 'wb'
if content is None:
if binary:
content = bytes(get_string(100), 'ascii')
pre = bytes()
if template:
pre = bytes('{{@@ header() @@}}\n', 'ascii')
content = bytes('{}{}\n'.format(pre, get_string(100)), 'ascii')
else:
content = get_string(100)
pre = ''
if template:
pre = '{{@@ header() @@}}\n'
content = '{}{}\n'.format(pre, get_string(100))
path = os.path.join(directory, fname)
with open(path, mode) as f:
f.write(content)