|
11 | 11 | importgitlab.base |
12 | 12 | fromgitlabimportcli |
13 | 13 | fromgitlab.exceptionsimportGitlabError |
| 14 | +fromgitlab.mixinsimportCreateMixin,UpdateMixin |
| 15 | +fromgitlab.typesimportRequiredOptional |
14 | 16 | fromgitlab.v4importcliasv4_cli |
15 | 17 |
|
16 | 18 |
|
@@ -157,6 +159,65 @@ def test_v4_parser(): |
157 | 159 | assertactions["--name"].required |
158 | 160 |
|
159 | 161 |
|
| 162 | +deftest_extend_parser(): |
| 163 | +classExceptionArgParser(argparse.ArgumentParser): |
| 164 | +deferror(self,message): |
| 165 | +"Raise error instead of exiting on invalid arguments, to make testing easier" |
| 166 | +raiseValueError(message) |
| 167 | + |
| 168 | +classFake: |
| 169 | +_id_attr=None |
| 170 | + |
| 171 | +classFakeManager(gitlab.base.RESTManager,CreateMixin,UpdateMixin): |
| 172 | +_obj_cls=Fake |
| 173 | +_create_attrs=RequiredOptional( |
| 174 | +required=("create",), |
| 175 | +optional=("opt_create",), |
| 176 | +exclusive=("create_a","create_b"), |
| 177 | + ) |
| 178 | +_update_attrs=RequiredOptional( |
| 179 | +required=("update",), |
| 180 | +optional=("opt_update",), |
| 181 | +exclusive=("update_a","update_b"), |
| 182 | + ) |
| 183 | + |
| 184 | +parser=ExceptionArgParser() |
| 185 | +withmock.patch.dict( |
| 186 | +"gitlab.v4.objects.__dict__", {"FakeManager":FakeManager},clear=True |
| 187 | + ): |
| 188 | +v4_cli.extend_parser(parser) |
| 189 | + |
| 190 | +assertparser.parse_args(["fake","create","--create","1"]) |
| 191 | +assertparser.parse_args(["fake","create","--create","1","--opt-create","1"]) |
| 192 | +assertparser.parse_args(["fake","create","--create","1","--create-a","1"]) |
| 193 | +assertparser.parse_args(["fake","create","--create","1","--create-b","1"]) |
| 194 | + |
| 195 | +withpytest.raises(ValueError): |
| 196 | +# missing required "create" |
| 197 | +parser.parse_args(["fake","create","--opt_create","1"]) |
| 198 | + |
| 199 | +withpytest.raises(ValueError): |
| 200 | +# both exclusive options |
| 201 | +parser.parse_args( |
| 202 | + ["fake","create","--create","1","--create-a","1","--create-b","1"] |
| 203 | + ) |
| 204 | + |
| 205 | +assertparser.parse_args(["fake","update","--update","1"]) |
| 206 | +assertparser.parse_args(["fake","update","--update","1","--opt-update","1"]) |
| 207 | +assertparser.parse_args(["fake","update","--update","1","--update-a","1"]) |
| 208 | +assertparser.parse_args(["fake","update","--update","1","--update-b","1"]) |
| 209 | + |
| 210 | +withpytest.raises(ValueError): |
| 211 | +# missing required "update" |
| 212 | +parser.parse_args(["fake","update","--opt_update","1"]) |
| 213 | + |
| 214 | +withpytest.raises(ValueError): |
| 215 | +# both exclusive options |
| 216 | +parser.parse_args( |
| 217 | + ["fake","update","--update","1","--update-a","1","--update-b","1"] |
| 218 | + ) |
| 219 | + |
| 220 | + |
160 | 221 | @pytest.mark.skipif(sys.version_info< (3,8),reason="added in 3.8") |
161 | 222 | deftest_legacy_display_without_fields_warns(fake_object_no_id): |
162 | 223 | printer=v4_cli.LegacyPrinter() |
|