testing/mozbase/mozdevice/sut_tests/genfiles.py@6474c204b198
testing/mozbase/mozdevice/sut_tests/genfiles.py
Wed, 31 Dec 2014 06:09:35 +0100
- author
- Michael Schloh von Bennewitz <michael@schloh.com>
- date
- Wed, 31 Dec 2014 06:09:35 +0100
- changeset 0
- 6474c204b198
- permissions
- -rw-r--r--
Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.
1 # This Source Code Form is subject to the terms of the Mozilla Public
2 # License, v. 2.0. If a copy of the MPL was not distributed with this
3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
6 from random import randint
7 from zipfile import ZipFile
8 import os
9 import shutil
12 def gen_binary_file(path, size):
13 with open(path, 'wb') as f:
14 for i in xrange(size):
15 byte = '%c' % randint(0, 255)
16 f.write(byte)
19 def gen_zip(path, files, stripped_prefix=''):
20 with ZipFile(path, 'w') as z:
21 for f in files:
22 new_name = f.replace(stripped_prefix, '')
23 z.write(f, new_name)
26 def mkdir(path, *args):
27 try:
28 os.mkdir(path, *args)
29 except OSError:
30 pass
33 def gen_folder_structure():
34 root = 'test-files'
35 prefix = os.path.join(root, 'push2')
36 mkdir(prefix)
38 gen_binary_file(os.path.join(prefix, 'file4.bin'), 59036)
39 mkdir(os.path.join(prefix, 'sub1'))
40 shutil.copyfile(os.path.join(root, 'mytext.txt'),
41 os.path.join(prefix, 'sub1', 'file1.txt'))
42 mkdir(os.path.join(prefix, 'sub1', 'sub1.1'))
43 shutil.copyfile(os.path.join(root, 'mytext.txt'),
44 os.path.join(prefix, 'sub1', 'sub1.1', 'file2.txt'))
45 mkdir(os.path.join(prefix, 'sub2'))
46 shutil.copyfile(os.path.join(root, 'mytext.txt'),
47 os.path.join(prefix, 'sub2', 'file3.txt'))
50 def gen_test_files():
51 gen_folder_structure()
52 flist = [
53 os.path.join('test-files', 'push2'),
54 os.path.join('test-files', 'push2', 'file4.bin'),
55 os.path.join('test-files', 'push2', 'sub1'),
56 os.path.join('test-files', 'push2', 'sub1', 'file1.txt'),
57 os.path.join('test-files', 'push2', 'sub1', 'sub1.1'),
58 os.path.join('test-files', 'push2', 'sub1', 'sub1.1', 'file2.txt'),
59 os.path.join('test-files', 'push2', 'sub2'),
60 os.path.join('test-files', 'push2', 'sub2', 'file3.txt')
61 ]
62 gen_zip(os.path.join('test-files', 'mybinary.zip'),
63 flist, stripped_prefix=('test-files' + os.path.sep))
64 gen_zip(os.path.join('test-files', 'mytext.zip'),
65 [os.path.join('test-files', 'mytext.txt')])
68 def clean_test_files():
69 ds = [os.path.join('test-files', d) for d in ('push1', 'push2')]
70 for d in ds:
71 try:
72 shutil.rmtree(d)
73 except OSError:
74 pass
76 fs = [os.path.join('test-files', f) for f in ('mybinary.zip', 'mytext.zip')]
77 for f in fs:
78 try:
79 os.remove(f)
80 except OSError:
81 pass
84 if __name__ == '__main__':
85 gen_test_files()
