| 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 from __future__ import unicode_literals | 16 from __future__ import unicode_literals |
| 17 | 17 |
| 18 import os | 18 import os |
| 19 from ConfigParser import SafeConfigParser | 19 from ConfigParser import SafeConfigParser |
| 20 import io | 20 import io |
| 21 import sys | 21 import sys |
| 22 import json | 22 import json |
| 23 | 23 |
| 24 import pytest | 24 import pytest |
| 25 | 25 |
| 26 from cms.translations.xtm import constants as const | 26 from cms.translations.xtm import constants as const |
| 27 from cms.translations.xtm.projects_handler import ( | |
| 28 create_project, upload_files, download_files, | |
| 29 ) | |
| 30 from cms.translations.xtm.cli import ( | 27 from cms.translations.xtm.cli import ( |
| 31 generate_token, handle_projects as main_project_handler, | 28 generate_token, handle_projects as main_project_handler, |
| 32 ) | 29 ) |
| 30 from tests.utils import XtmMockArgs |
| 33 | 31 |
| 34 _CMD_START = ['python', '-m', 'cms.bin.xtm_translations'] | 32 _CMD_START = ['python', '-m', 'cms.bin.xtm_translations'] |
| 35 | 33 |
| 36 _ENV_NO_TOKEN = dict(os.environ) | 34 _ENV_NO_TOKEN = dict(os.environ) |
| 37 _ENV_NO_TOKEN.pop(const.Token.ENV_VAR, None) | 35 _ENV_NO_TOKEN.pop(const.Token.ENV_VAR, None) |
| 38 | 36 |
| 39 _ENV_TOKEN_VALID = dict(os.environ) | 37 _ENV_TOKEN_VALID = dict(os.environ) |
| 40 _ENV_TOKEN_VALID[const.Token.ENV_VAR] = 'TheXTM-APIToken-VALID' | 38 _ENV_TOKEN_VALID[const.Token.ENV_VAR] = 'TheXTM-APIToken-VALID' |
| 41 | 39 |
| 42 _ENV_TOKEN_INVALID = dict(os.environ) | 40 _ENV_TOKEN_INVALID = dict(os.environ) |
| 43 _ENV_TOKEN_INVALID[const.Token.ENV_VAR] = 'TheXTM-APIToken-INVALID' | 41 _ENV_TOKEN_INVALID[const.Token.ENV_VAR] = 'TheXTM-APIToken-INVALID' |
| 44 | 42 |
| 45 | |
| 46 class _CreationArgsNamespace: | |
| 47 def __init__(self): | |
| 48 pass | |
| 49 | |
| 50 name = 'bar' | |
| 51 desc = 'foo' | |
| 52 client_id = 10 | |
| 53 ref_id = 'faz' | |
| 54 workflow_id = 20 | |
| 55 save_id = False | |
| 56 source_dir = None | |
| 57 projects_func = staticmethod(create_project) | |
| 58 source_lang = 'en_US' | |
| 59 | |
| 60 | |
| 61 class _UploadArgsNamespace: | |
| 62 def __init__(self): | |
| 63 pass | |
| 64 | |
| 65 source_dir = None | |
| 66 projects_func = staticmethod(upload_files) | |
| 67 no_overwrite = False | |
| 68 | |
| 69 | |
| 70 class _DownloadArgsNamespace: | |
| 71 def __init__(self): | |
| 72 pass | |
| 73 | |
| 74 source_dir = None | |
| 75 projects_func = staticmethod(download_files) | |
| 76 | |
| 77 | |
| 78 _CREATION_ARGS_DEFAULT = ['--name', 'bar', '--desc', 'foo', '--client-id', | 43 _CREATION_ARGS_DEFAULT = ['--name', 'bar', '--desc', 'foo', '--client-id', |
| 79 '10', '--ref-id', 'faz', '--workflow-id', '20'] | 44 '10', '--ref-id', 'faz', '--workflow-id', '20'] |
| 80 _CREATION_ARGS_INVALID_TYPE_1 = ['--name', 'bar', '--desc', 'foo', | 45 _CREATION_ARGS_INVALID_TYPE_1 = ['--name', 'bar', '--desc', 'foo', |
| 81 '--client-id', 'foo', '--ref-id', 'faz', | 46 '--client-id', 'foo', '--ref-id', 'faz', |
| 82 '--workflow-id', '3'] | 47 '--workflow-id', '3'] |
| 83 _CREATION_ARGS_INVALID_TYPE_2 = ['--name', 'bar', '--desc', 'foo', | 48 _CREATION_ARGS_INVALID_TYPE_2 = ['--name', 'bar', '--desc', 'foo', |
| 84 '--client-id', '23', '--ref-id', 'faz', | 49 '--client-id', '23', '--ref-id', 'faz', |
| 85 '--workflow-id', 'foo'] | 50 '--workflow-id', 'foo'] |
| 86 | 51 |
| 52 _UploadArgsNamespace = XtmMockArgs.UploadArgsNamespace |
| 53 _CreationArgsNamespace = XtmMockArgs.CreationArgsNamespace |
| 54 _DownloadArgsNamespace = XtmMockArgs.DownloadArgsNamespace |
| 55 |
| 87 | 56 |
| 88 @pytest.fixture | 57 @pytest.fixture |
| 89 def env_valid_token(): | 58 def env_valid_token(): |
| 90 old_env = os.environ | 59 old_env = os.environ |
| 91 os.environ = _ENV_TOKEN_VALID | 60 os.environ = _ENV_TOKEN_VALID |
| 92 yield True | 61 yield True |
| 93 os.environ = old_env | 62 os.environ = old_env |
| 94 | 63 |
| 95 | 64 |
| 96 @pytest.mark.script_launch_mode('subprocess') | 65 @pytest.mark.script_launch_mode('subprocess') |
| 97 @pytest.mark.parametrize('args,exp_msg', [ | 66 @pytest.mark.parametrize('args,exp_msg', [ |
| 98 (['-h'], 'usage: xtm_translations.py [-h] [-v] ' | 67 (['-h'], 'usage: xtm_translations.py [-h] [-v] ' |
| 99 '{login,create,upload,download} ...'), | 68 '{login,create,upload,download} ...'), |
| 100 (['create', '-h'], 'usage: xtm_translations.py create [-h] --name NAME ' | 69 (['create', '-h'], 'usage: xtm_translations.py create [-h] --name NAME ' |
| 101 '--desc DESC --client-id CLIENT_ID --ref-id REF_ID ' | 70 '--desc DESC --client-id CLIENT_ID --ref-id REF_ID ' |
| 102 '--workflow-id WORKFLOW_ID [--source-lang SOURCE_LANG] ' | 71 '--workflow-id WORKFLOW_ID [--source-lang ' |
| 103 '[--save-id] [source_dir]'), | 72 'SOURCE_LANG] [--save-id] [--workflow-name ' |
| 73 'WORKFLOW_NAME] [source_dir]'), |
| 104 (['upload', '-h'], 'usage: xtm_translations.py upload [-h] ' | 74 (['upload', '-h'], 'usage: xtm_translations.py upload [-h] ' |
| 105 '[--no-overwrite] [source_dir]'), | 75 '[--no-overwrite] [source_dir]'), |
| 106 (['download', '-h'], 'usage: xtm_translations.py download [-h] ' | 76 (['download', '-h'], 'usage: xtm_translations.py download [-h] ' |
| 107 '[source_dir]'), | 77 '[source_dir]'), |
| 108 ]) | 78 ]) |
| 109 def test_usage_messages(args, exp_msg, script_runner): | 79 def test_usage_messages(args, exp_msg, script_runner): |
| 110 """Test if appropriate usage messages are displayed.""" | 80 """Test if appropriate usage messages are displayed.""" |
| 111 cmd = list(_CMD_START) | 81 cmd = list(_CMD_START) |
| 112 cmd.extend(args) | 82 cmd.extend(args) |
| 113 ret = script_runner.run(*cmd) | 83 ret = script_runner.run(*cmd) |
| (...skipping 190 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 304 | 274 |
| 305 main_project_handler(namespace) | 275 main_project_handler(namespace) |
| 306 | 276 |
| 307 with open(os.path.join(temp_site_valid_project, 'locales', 'de', | 277 with open(os.path.join(temp_site_valid_project, 'locales', 'de', |
| 308 'file.json')) as f: | 278 'file.json')) as f: |
| 309 assert json.dumps({'foo': 'bar', 'faz': 'baz'}) == f.read() | 279 assert json.dumps({'foo': 'bar', 'faz': 'baz'}) == f.read() |
| 310 | 280 |
| 311 with open(os.path.join(temp_site_valid_project, 'locales', 'de', 'foo', | 281 with open(os.path.join(temp_site_valid_project, 'locales', 'de', 'foo', |
| 312 'file-utf8.json'), 'rb') as f: | 282 'file-utf8.json'), 'rb') as f: |
| 313 assert json.loads(f.read()) == {'foo': '\u1234'} | 283 assert json.loads(f.read()) == {'foo': '\u1234'} |
| OLD | NEW |