Mercurial > libervia-backend
annotate tests/e2e/libervia-cli/conftest.py @ 3863:c04f5e8a3568
plugin pubsub cache: replace empty list by None in `getItemsFromCache`:
if an empty list is used, no item can be retrieved, this patch replace empty list by None
to avoid this case.
author | Goffi <goffi@goffi.org> |
---|---|
date | Wed, 20 Jul 2022 17:46:58 +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() |