Issue 7037 - [XTM Integration] Make REST API url customizable

tests/test_xtm_translate.py

 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-present eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
@@ skipping 46 matching lines @@
 @pytest.fixture
 def env_valid_token():
     old_env = os.environ
     os.environ = _ENV_TOKEN_VALID
     yield True
     os.environ = old_env
 
 
 @pytest.mark.script_launch_mode('subprocess')
 @pytest.mark.parametrize('args,exp_msg', [
-    (['-h'], 'usage: xtm_translations.py [-h] [-v] '
+    (['-h'], 'usage: xtm_translations.py [-h] [-v] [--api-urlAPI_URL]'
              '{login,create,upload,download} ...'),
     (['create', '-h'], 'usage: xtm_translations.py create [-h] --name NAME '
                        '--desc DESC --customer-id CUSTOMER_ID --ref-id REF_ID '
                        '[--workflow-id WORKFLOW_ID] [--source-lang '
                        'SOURCE_LANG] [--save-id] [--workflow-name '
                        'WORKFLOW_NAME] [source_dir]'),
     (['upload', '-h'], 'usage: xtm_translations.py upload [-h] '
                        '[--no-overwrite] [source_dir]'),
     (['download', '-h'], 'usage: xtm_translations.py download [-h] '
                          '[source_dir]'),
@@ skipping 31 matching lines @@
 @pytest.mark.script_launch_mode('subprocess')
 @pytest.mark.parametrize('source_dir,args,env,exp_msg', [
     ('str(temp_site)', _CREATION_ARGS_INVALID_TYPE_1, _ENV_NO_TOKEN,
      "--customer-id: invalid int value: 'foo'"),
     ('str(temp_site)', _CREATION_ARGS_INVALID_TYPE_2, _ENV_NO_TOKEN,
      "--workflow-id: invalid int value: 'foo'"),
     ('str(temp_site)', _CREATION_ARGS_DEFAULT, _ENV_NO_TOKEN,
      const.ErrorMessages.NO_TOKEN_PROVIDED.split('\n')[0]),
     ('str(temp_site_valid_project)', _CREATION_ARGS_DEFAULT, _ENV_TOKEN_VALID,
      const.ErrorMessages.PROJECT_EXISTS.format(1234)),
-    ('str(temp_site)', _CREATION_ARGS_DEFAULT, _ENV_TOKEN_INVALID,
+    ('str(temp_site_no_project)', _CREATION_ARGS_DEFAULT, _ENV_TOKEN_INVALID,
      'Authentication failed'),
 ])
 def test_creation_error_messages(temp_site, intercept, script_runner, args,
                                  source_dir, temp_site_valid_project, env,
-                                 exp_msg):
+                                 exp_msg, temp_site_no_project):
     """Test if error cases are treated correctly when creating a project."""
     cmd = list(_CMD_START)
     cmd.extend(['create', eval(source_dir)])
     cmd.extend(args)
 
     ret = script_runner.run(*cmd, env=env)
 
     assert not ret.success
     assert exp_msg in ret.stderr
 
@@ skipping 30 matching lines @@
 def test_login(intercept, monkeypatch, capsys):
     """Test if the login functionality works as expected."""
     exp_output = const.Token.SAVE_COMMAND.format(const.Token.ENV_VAR,
                                                  'TheXTM-APIToken-VALID')
     monkeypatch.setattr(
         'cms.translations.xtm.cli.input_fn',
         lambda inp: 'admin' if 'client name' in inp.lower() else '20',
     )
     monkeypatch.setattr('getpass.getpass', lambda prompt: 'pass')
 
-    generate_token(None)
+    generate_token(XtmMockArgs.LoginArgsNamespace)
     out, err = capsys.readouterr()
 
     assert err == ''
     assert exp_output in out
 
 
 def test_login_wrong_credentials(intercept, monkeypatch, capsys):
     """Test exception handling when generating the tokens."""
     monkeypatch.setattr(
         'cms.translations.xtm.cli.input_fn',
         lambda inp: 'foo' if 'client name' in inp.lower() else '50',
     )
     monkeypatch.setattr('getpass.getpass', lambda prompt: 'pass')
     monkeypatch.setattr('sys.exit', lambda x: sys.stderr.write(str(x)))
 
-    generate_token(None)
+    generate_token(XtmMockArgs.LoginArgsNamespace)
     out, err = capsys.readouterr()
 
     assert 'Invalid credentials' in err
     assert out == ''
 
 
 @pytest.mark.script_launch_mode('subprocess')
 @pytest.mark.parametrize('args,env,exp_msg', [
     (['str(temp_site)'], _ENV_NO_TOKEN,
      const.ErrorMessages.NO_TOKEN_PROVIDED.split('\n')[0]),
@@ skipping 10 matching lines @@
 
     assert not ret.success
     assert exp_msg in ret.stderr
 
 
 def test_upload_too_many_languages(intercept_too_many_targets,
                                    env_valid_token, temp_site_valid_project):
     namespace = _UploadArgsNamespace()
     namespace.source_dir = str(temp_site_valid_project)
 
-    with pytest.raises(Exception) as err:
+    with pytest.raises(SystemExit) as err:
         main_project_handler(namespace)
 
-    assert 'languages are enabled in the API, but not listed in locales' in \
-           str(err.value)
+    assert ('languages are enabled in the API, but not listed in locales' in
+            str(err.value))
 
 
 def test_upload_successful(intercept, env_valid_token,
                            temp_site_valid_project, monkeypatch, capsys):
     namespace = _UploadArgsNamespace()
     namespace.source_dir = str(temp_site_valid_project)
     monkeypatch.setattr('logging.info', lambda x: sys.stderr.write(x))
 
     main_project_handler(namespace)
     _, err = capsys.readouterr()
@@ skipping 37 matching lines @@
 
     main_project_handler(namespace)
 
     with open(os.path.join(temp_site_valid_project, 'locales', 'de',
                            'file.json')) as f:
         assert json.dumps({'foo': 'bar', 'faz': 'baz'}) == f.read()
 
     with open(os.path.join(temp_site_valid_project, 'locales', 'de', 'foo',
                            'file-utf8.json'), 'rb') as f:
         assert json.loads(f.read()) == {'foo': '\u1234'}