Mercurial > libervia-backend
annotate tests/e2e/libervia-cli/conftest.py @ 3913:944f51f9c2b4
core (xmpp): make `send` a blocking method, fix `sendMessageData` calls:
original `send` method is blocking, and it is used as such by Wokkel and thus can't be
changed to an async method easily. However, an Async method is necessary to have an async
trigger at the very end of the send workflow for end-to-end encryption. To workaround
that, `send` is an async method which call `a_send`, an async method which actually does
the sending. This way legacy method can still call `send` while `a_send` can be await
otherwise.
Fix calls to `sendMessageData`: the method now being an `async` one, `ensureDeferred` had
to be used in some calls.
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sat, 24 Sep 2022 16:31:39 +0200 |
| parents | d4b8f1e8f0b4 |
| children | 4b842c1fb686 |
| rev | line source |
|---|---|
| 3415 | 1 #!/usr/bin/env python3 |
| 2 | |
|
3480
7550ae9cfbac
Renamed the project from "Salut à Toi" to "Libervia":
Goffi <goffi@goffi.org>
parents:
3479
diff
changeset
|
3 # Libervia: an XMPP client |
| 3479 | 4 # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org) |
| 3415 | 5 |
| 6 # This program is free software: you can redistribute it and/or modify | |
| 7 # it under the terms of the GNU Affero General Public License as published by | |
| 8 # the Free Software Foundation, either version 3 of the License, or | |
| 9 # (at your option) any later version. | |
| 10 | |
| 11 # This program is distributed in the hope that it will be useful, | |
| 12 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 14 # GNU Affero General Public License for more details. | |
| 15 | |
| 16 # You should have received a copy of the GNU Affero General Public License | |
| 17 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 18 | |
| 19 import sys | |
| 20 import os | |
| 21 import tempfile | |
| 22 from pathlib import Path | |
| 23 from textwrap import dedent | |
| 24 import json | |
| 25 import pytest | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
26 from sh import li |
| 3415 | 27 |
| 28 | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
29 class LiberviaCliJson: |
|
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
30 """li like commands parsing result as JSON""" |
| 3415 | 31 |
| 32 def __init__(self): | |
| 33 self.subcommands = [] | |
| 34 | |
| 35 def __call__(self, *args, **kwargs): | |
| 36 args = self.subcommands + list(args) | |
| 37 self.subcommands.clear() | |
|
3659
d4b8f1e8f0b4
tests (e2e/cli): update JSON and XML outputs following changes in Libervia CLI
Goffi <goffi@goffi.org>
parents:
3498
diff
changeset
|
38 kwargs['output'] = 'json-raw' |
| 3415 | 39 kwargs['_tty_out'] = False |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
40 cmd = li(*args, **kwargs) |
| 3415 | 41 return json.loads(cmd.stdout) |
| 42 | |
| 43 def __getattr__(self, name): | |
| 44 if name.startswith('_'): | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
45 # no li subcommand starts with a "_", |
| 3415 | 46 # and pytest uses some attributes with this name scheme |
| 47 return super().__getattr__(name) | |
| 48 self.subcommands.append(name) | |
| 49 return self | |
| 50 | |
| 51 | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
52 class LiberviaCliElt(LiberviaCliJson): |
|
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
53 """li like commands parsing result as domishElement""" |
| 3415 | 54 |
| 55 def __init__(self): | |
| 56 super().__init__() | |
| 57 from sat.tools.xml_tools import ElementParser | |
| 58 self.parser = ElementParser() | |
| 59 | |
| 60 def __call__(self, *args, **kwargs): | |
| 61 args = self.subcommands + list(args) | |
| 62 self.subcommands.clear() | |
|
3659
d4b8f1e8f0b4
tests (e2e/cli): update JSON and XML outputs following changes in Libervia CLI
Goffi <goffi@goffi.org>
parents:
3498
diff
changeset
|
63 kwargs['output'] = 'xml-raw' |
| 3415 | 64 kwargs['_tty_out'] = False |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
65 cmd = li(*args, **kwargs) |
| 3415 | 66 return self.parser(cmd.stdout.decode().strip()) |
| 67 | |
| 68 | |
| 69 class Editor: | |
| 70 | |
| 71 def __init__(self): | |
| 72 # temporary directory will be deleted Automatically when this object will be | |
| 73 # destroyed | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
74 self.tmp_dir_obj = tempfile.TemporaryDirectory(prefix="libervia_e2e_test_editor_") |
| 3415 | 75 self.tmp_dir_path = Path(self.tmp_dir_obj.name) |
| 76 if not sys.executable: | |
| 77 raise Exception("Can't find python executable") | |
| 78 self.editor_set = False | |
| 79 self.editor_path = self.tmp_dir_path / "editor.py" | |
| 80 self.ori_content_path = self.tmp_dir_path / "original_content" | |
| 81 self.new_content_path = self.tmp_dir_path / "new_content" | |
| 82 self.base_script = dedent(f"""\ | |
| 83 #!{sys.executable} | |
| 84 import sys | |
| 85 | |
| 86 def content_filter(content): | |
| 87 return {{content_filter}} | |
| 88 | |
| 89 with open(sys.argv[1], 'r+') as f: | |
| 90 original_content = f.read() | |
| 91 f.seek(0) | |
| 92 new_content = content_filter(original_content) | |
| 93 f.write(new_content) | |
| 94 f.truncate() | |
| 95 | |
| 96 with open("{self.ori_content_path}", "w") as f: | |
| 97 f.write(original_content) | |
| 98 | |
| 99 with open("{self.new_content_path}", "w") as f: | |
| 100 f.write(new_content) | |
| 101 """ | |
| 102 ) | |
| 103 self._env = os.environ.copy() | |
| 104 self._env["EDITOR"] = str(self.editor_path) | |
| 105 | |
| 106 def set_filter(self, content_filter: str = "content"): | |
| 107 """Python code to modify original content | |
| 108 | |
| 109 The code will be applied to content received by editor. | |
| 110 The original content received by editor is in the "content" variable. | |
| 111 If filter_ is not specified, original content is written unmodified. | |
| 112 Code must be on a single line. | |
| 113 """ | |
| 114 if '\n' in content_filter: | |
| 115 raise ValueError("new lines can't be used in filter_") | |
| 116 with self.editor_path.open('w') as f: | |
| 117 f.write(self.base_script.format(content_filter=content_filter)) | |
| 118 self.editor_path.chmod(0o700) | |
| 119 self.editor_set = True | |
| 120 | |
| 121 @property | |
| 122 def env(self): | |
| 123 """Get environment variable with the editor set""" | |
| 124 if not self.editor_set: | |
| 125 self.set_filter() | |
| 126 return self._env | |
| 127 | |
| 128 @property | |
| 129 def original_content(self): | |
| 130 """Last content received by editor, before any modification | |
| 131 | |
| 132 returns None if editor has not yet been called | |
| 133 """ | |
| 134 try: | |
| 135 with self.ori_content_path.open() as f: | |
| 136 return f.read() | |
| 137 except FileNotFoundError: | |
| 138 return None | |
| 139 | |
| 140 @property | |
| 141 def new_content(self): | |
| 142 """Last content writen by editor | |
| 143 | |
| 144 This is the final content, after filter has been applied to original content | |
| 145 returns None if editor has not yet been called | |
| 146 """ | |
| 147 try: | |
| 148 with self.new_content_path.open() as f: | |
| 149 return f.read() | |
| 150 except FileNotFoundError: | |
| 151 return None | |
| 152 | |
| 153 | |
| 154 @pytest.fixture(scope="session") | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
155 def li_json(): |
|
3659
d4b8f1e8f0b4
tests (e2e/cli): update JSON and XML outputs following changes in Libervia CLI
Goffi <goffi@goffi.org>
parents:
3498
diff
changeset
|
156 """Run li with "json-raw" output, and returns the parsed value""" |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
157 return LiberviaCliJson() |
| 3415 | 158 |
| 159 | |
| 160 @pytest.fixture(scope="session") | |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
161 def li_elt(): |
|
3659
d4b8f1e8f0b4
tests (e2e/cli): update JSON and XML outputs following changes in Libervia CLI
Goffi <goffi@goffi.org>
parents:
3498
diff
changeset
|
162 """Run li with "xml-raw" output, and returns the parsed value""" |
|
3498
d78b5eae912a
tests: update following names change
Goffi <goffi@goffi.org>
parents:
3480
diff
changeset
|
163 return LiberviaCliElt() |
| 3415 | 164 |
| 165 | |
| 166 @pytest.fixture() | |
| 167 def editor(): | |
| 168 """Create a fake editor to automatise edition from CLI""" | |
| 169 return Editor() |