| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| 1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
| 2 # Copyright (C) 2006-present eyeo GmbH | 2 # Copyright (C) 2006-present eyeo GmbH |
| 3 # | 3 # |
| 4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
| 5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
| 6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
| 7 # | 7 # |
| 8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
| 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
| 12 # | 12 # |
| 13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
| 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| 15 | 15 |
| 16 import contextlib | 16 import contextlib |
| 17 import os | 17 import os |
| 18 import signal | 18 import signal |
| 19 import subprocess | 19 import subprocess |
| 20 import time | 20 import time |
| 21 import zipfile | 21 import zipfile |
| 22 from io import BytesIO | 22 from io import BytesIO |
| 23 | 23 |
| 24 import pytest | 24 import pytest |
| 25 | 25 |
| 26 from cms.translations.xtm.projects_handler import ( | |
| 27 create_project, upload_files, download_files, | |
| 28 ) | |
| 29 | |
| 26 | 30 |
| 27 def get_dir_contents(path): | 31 def get_dir_contents(path): |
| 28 # TODO: This function is duplicated in test_page_outputs.py. | 32 # TODO: This function is duplicated in test_page_outputs.py. |
| 29 dirdata = {} | 33 dirdata = {} |
| 30 for dirpath, dirnames, filenames in os.walk(path): | 34 for dirpath, dirnames, filenames in os.walk(path): |
| 31 for output_file in filenames: | 35 for output_file in filenames: |
| 32 filepath = os.path.join(dirpath, output_file) | 36 filepath = os.path.join(dirpath, output_file) |
| 33 with open(filepath) as f: | 37 with open(filepath) as f: |
| 34 locale = os.path.split(os.path.split(filepath)[0])[1] | 38 locale = os.path.split(os.path.split(filepath)[0])[1] |
| 35 dirdata[os.path.join(locale, output_file)] = f.read().strip() | 39 dirdata[os.path.join(locale, output_file)] = f.read().strip() |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 73 | 77 |
| 74 with zipfile.ZipFile(memory_zip, 'w') as zf: | 78 with zipfile.ZipFile(memory_zip, 'w') as zf: |
| 75 for idx in range(len(file_names)): | 79 for idx in range(len(file_names)): |
| 76 zf.writestr(file_names[idx], file_data[idx], zipfile.ZIP_DEFLATED) | 80 zf.writestr(file_names[idx], file_data[idx], zipfile.ZIP_DEFLATED) |
| 77 | 81 |
| 78 memory_zip.seek(0) | 82 memory_zip.seek(0) |
| 79 return memory_zip | 83 return memory_zip |
| 80 | 84 |
| 81 | 85 |
| 82 def exception_test(func, exception, exp_msg, *args, **kw): | 86 def exception_test(func, exception, exp_msg, *args, **kw): |
| 87 """Test if a function raises the correct exception. | |
| 88 | |
| 89 Parameters | |
| 90 ---------- | |
| 91 func: function | |
| 92 The function we're testing. | |
| 93 exception: classobj | |
| 94 The exception we expect to be raised. | |
| 95 exp_msg: str | |
| 96 The message we expect the exception to contain. | |
| 97 | |
| 98 """ | |
| 83 with pytest.raises(exception) as err: | 99 with pytest.raises(exception) as err: |
| 84 func(*args, **kw) | 100 func(*args, **kw) |
| 85 | 101 |
| 86 assert exp_msg in str(err.value) | 102 assert exp_msg in str(err.value) |
| 103 | |
| 104 | |
| 105 class XtmMockArgs: | |
| 106 | |
| 107 class CreationArgsNamespace: | |
| 108 def __init__(self): | |
|
Vasily Kuznetsov
2018/10/16 14:52:54
You don't really need these empty __init__'s, not
Tudor Avram
2018/10/18 16:29:23
Done.
| |
| 109 pass | |
| 110 | |
| 111 name = 'bar' | |
| 112 desc = 'foo' | |
| 113 client_id = 10 | |
| 114 ref_id = 'faz' | |
| 115 workflow_id = 20 | |
| 116 save_id = False | |
| 117 source_dir = None | |
| 118 projects_func = staticmethod(create_project) | |
| 119 source_lang = 'en_US' | |
| 120 workflow_name = None | |
| 121 | |
| 122 class UploadArgsNamespace: | |
| 123 def __init__(self): | |
| 124 pass | |
| 125 | |
| 126 source_dir = None | |
| 127 projects_func = staticmethod(upload_files) | |
| 128 no_overwrite = False | |
| 129 | |
| 130 class DownloadArgsNamespace: | |
| 131 def __init__(self): | |
| 132 pass | |
| 133 | |
| 134 source_dir = None | |
| 135 projects_func = staticmethod(download_files) | |
| OLD | NEW |