/
__init__.py
769 lines (638 loc) · 34.8 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
from __future__ import (
absolute_import,
unicode_literals,
)
import abc
import os
import re
import sys
import traceback
from types import TracebackType
from typing import (
Any,
Dict,
Generator,
List,
Optional,
Tuple,
Type,
Union,
)
import attr
import six
from pysoa.common.types import Body
from pysoa.test.compatibility import mock
from pysoa.test.plan import grammar
from pysoa.test.plan.errors import (
DirectiveError,
FixtureLoadError,
StatusError,
)
from pysoa.test.plan.grammar.directive import (
ActionCase,
ActionResults,
Directive,
TestCase,
TestFixture,
TestFixtureResults,
get_all_directives,
)
from pysoa.test.plan.grammar.tools import substitute_variables
from pysoa.test.plan.parser import ServiceTestPlanFixtureParser
from pysoa.test.server import PyTestServerTestCase
from pysoa.test.stub_service import stub_action
try:
from typing import Literal # type: ignore
except ImportError:
from typing_extensions import Literal # type: ignore
try:
from typing import Protocol
try:
# Py3.8, Py2.7 backport
from typing import runtime_checkable
except ImportError:
# Some Py3.7
from typing import runtime as runtime_checkable # type: ignore
except ImportError:
# Some Py3.7, all Py3<3.7
from typing_extensions import ( # type: ignore
Protocol,
runtime_checkable,
)
__doc__ = """Another option for testing PySOA services is to use its test plan system. Test plans extend
:class:`pysoa.test.plan.ServicePlanTestCase` and define a collection of plain text fixture files (with extension
``.pysoa``) that use a specialized syntax for describing test cases that call actions on your service.
To best understand PySOA test plans and this documentation, you'll need to understand a little bit of nomenclature:
- **Test Plan**: A class that extends :class:`pysoa.test.plan.ServicePlanTestCase` and declares a directory in which
test fixtures can be discovered for that test plan. If you want, you can have a single test plan for all of the test
fixtures in your service. You can also have multiple test plans, each with one or more fixtures. The advantage of
using multiple test plans is that each test plan class can have unique setup activities.
- **Test Fixture**: A ``.pysoa`` file containing one or more test cases defined using the PySOA Test Plan syntax. A
test fixture's name is the test fixture file name absent the ``.pysoa`` extension and any directories.
- **Test Case**: A individual test case within a given test fixture. Each test case must have a name (letters, numbers,
and underscores, only) and a description (a natural language sentence describing its purpose). A test case must have
one or more action cases.
- **Action Case**: An individual call to a service action within a test case. Each action case has an associated set of
inputs used to make the action call and expectations used to assert the results of the action call.
Running Test Plans
******************
PySOA test plans are collected and executed with a PyTest plugin, which is not installed by default. To enable this
plugin, you need to add ``pysoa[pytest]`` to your test requirements. Example:
.. code-block:: python
tests_require = [
'pysoa[pytest]',
...
]
Once you do this and install your testing dependencies, you will be able to run your service's test plans. Without
this, the presence of test plans in your service will result in errors during testing.
By default, all normal tests and test plan tests will run when you invoke ``pytest`` without arguments. If you pass a
directory to ``pytest``, it will run all normal tests and test plan tests in that directory. (NOTE: For the purposes
of directory collection, test plans reside in the test case class that declares them.) You can also easily filter the
tests fixtures and test cases that are run using the ``pytest`` arguments::
# This will match all fixture AND non-fixture test cases with the name: get_user
pytest -k get_user
# This will match only fixture test cases with the name: get_user
pytest --pysoa-test-case get_user
# This will match only fixture test cases with names matching the regular expression ^get\\_user.*
pytest --pysoa-test-case-regex 'get\\_user.*'
# This will match only test cases within test fixtures with the name: user_actions
pytest --pysoa-fixture user_actions
# This will match only test cases named get_user within test fixtures named user_actions
pytest --pysoa-fixture user_actions --pysoa-test-case get_user
Note that ``--pysoa-test-case`` and ``--pysoa-test-case-regex`` are mutually exclusive arguments. Use ``pytest --help``
to get more information about available plugin arguments.
Creating a Test Plan with ``ServicePlanTestCase``
*************************************************
In order to create test plans, the first thing you need to do is create a test case class that extends
:class:`pysoa.test.plan.ServicePlanTestCase`. This class extends :class:`pysoa.test.server.PyTestServerTestCase` (see
`Using PyTestServerTestCase and UnitTestServerTestCase`_), so you need to define the same ``server_class`` and
``server_settings`` attributes. Additionally, you need to define either ``fixture_path`` or ``custom_fixtures``. You
can also optionally specify ``model_constants``, which is used to provide stock values for variable substitution (more
on that later). Here are two
possible examples:
.. code-block:: python
import os
from pysoa.test.plan import ServicePlanTestCase
from user_service.server import Server
class UserServiceFixtures(ServicePlanTestCase):
server_class = Server
server_settings = {}
fixture_path = os.path.dirname(__file__) + '/service_fixtures'
class ExtraServiceFixtures(ServicePlanTestCase):
server_class = Server
server_settings = {}
custom_fixtures = (
os.path.dirname(__file__) + '/extra_fixtures/special_actions_1.pysoa',
os.path.dirname(__file__) + '/extra_fixtures/special_actions_2.pysoa',
)
model_constants = {
'test_users': [
{'id': '1838', 'username': 'john.smith'},
{'id': '1792', 'username': 'jane.sanders'},
],
}
``ServicePlanTestCase`` provides a number of hooks that you can use to set up and tear down plans, fixtures, test
cases, and action cases. To learn more about these hooks, see the docstrings in ``ServicePlanTestCase`` for the
following methods. In each case, if you override the hook, you must call ``super`` as the first line in your hook.
- ``setup_class``
- ``set_up_test_fixture``
- ``setup_method``
- ``set_up_test_case``
- ``set_up_test_case_action``
- ``tear_down_test_case_action``
- ``tear_don_test_case``
- ``teardown_method``
- ``tear_down_test_fixture``
- ``teardown_class``
"""
__doc__ += grammar.__doc__
__doc__ += """Extending Test Plans
********************
You can extend test plan syntax to create your own directives, allowing you to add even more features to your test
plans. The base for all directive behavior is contained in the class
:class:`pysoa.test.plan.grammar.directive.Directive`. Your directives must extend that class directly or indirectly.
Extending the base class directly gives you the ability to manipulate test case-level and global test case-level
behavior. In most cases, you'll want to extend :class:`pysoa.test.plan.grammar.directive.ActionDirective`, which is
the base class for all action case behavior. For more information about how to use and extend these classes, read their
extensive docstrings.
Once you have created one or more new directives, you can register them with the PySOA Test Plan system using one of
the following techniques:
- Call :func:`pysoa.test.plan.grammar.directive.register_directive` to register your directive with the test plan system
manually. However, this requires your code that calls that function to be loaded before the PyTest process starts,
which can be tricky to achieve.
- Use the Python entry point named ``pysoa.test.plan.grammar.directives`` in your ``setup.py`` file. This is a more
reliable approach that works in all scenarios. Example:
.. code-block:: python
from setuptools import setup
...
setup(
name='base_service',
description='A layer on top of PySOA that serves as the base for all of our micro services',
...
entry_points={
'pysoa.test.plan.grammar.directives': [
'auth_token_directive = base_service.test.directives:AuthTokenDirective',
'authentication_directive = base_service.test.directives:AuthProcessingDirective',
],
},
...
)
"""
__test_plan_prune_traceback = True # ensure code in this file is not included in failure stack traces
__all__ = (
'FixtureTestCaseData',
'ServicePlanTestCase',
)
@runtime_checkable
class FixtureTestCaseFunction(Protocol):
# noinspection PyMethodParameters
def __call__(_, self, *args, **kwargs): # type: (ServicePlanTestCase, *Any, **Any) -> None
"""
Type-shed stub
"""
@attr.s
class FixtureTestCaseData(object):
"""
A plain-old Python object that holds fixture test case data.
"""
name = attr.ib() # type: six.text_type
description = attr.ib() # type: six.text_type
fixture_name = attr.ib() # type: six.text_type
fixture_file = attr.ib() # type: six.text_type
line_number = attr.ib() # type: int
skip = attr.ib() # type: Optional[six.text_type]
callable = attr.ib() # type: FixtureTestCaseFunction
test_fixture = attr.ib() # type: TestFixture
is_first_fixture_case = attr.ib() # type: bool
is_last_fixture_case = attr.ib() # type: bool
@six.add_metaclass(abc.ABCMeta)
class ServicePlanTestCase(PyTestServerTestCase):
"""
Serves as the base class for all test plans. Your test plans must extend this class, and may override any of its
methods, although, in most cases, you should not need to do this. Most commonly, you may override one of the setup
or teardown methods in order to bootstrap and clean up dependencies that your tests have.
Your test case class is not limited to running fixture tests. It may also include normal test case methods whose
names start with `test_`, and they will be run normally like any other PyTest test methods.
"""
fixture_path = None # type: Optional[six.text_type]
fixture_regex = re.compile(r'^[^.].*?\.(:?pysoa)$')
custom_fixtures = () # type: Tuple[six.text_type, ...]
model_constants = {} # type: Dict[six.text_type, Union[Dict[six.text_type, Any], List[Dict[six.text_type, Any]]]]
_all_directives = [] # type: List[Type[Directive]]
@classmethod
def setup_class(cls):
# type: () -> None
"""
This method is invoked one time before the test plan (all the fixtures defined in ``fixture_path``) or any of
the normal tests in your test case are run.
"""
cls._all_directives = get_all_directives()
super(ServicePlanTestCase, cls).setup_class()
if not hasattr(cls, '_test_fixture_setup_called'):
setattr(cls, '_test_fixture_setup_called', {})
if not hasattr(cls, '_test_fixture_setup_succeeded'):
setattr(cls, '_test_fixture_setup_succeeded', {})
@classmethod
def teardown_class(cls):
# type: () -> None
"""
This method is invoked one time after the test plan (all the fixtures defined in ``fixture_path``) and any of
the normal tests in your test case are run. If you override this, your call to `super` must be the very first
line of code in the method.
"""
_test_fixture_setup_called = getattr(cls, '_test_fixture_setup_called', None)
if _test_fixture_setup_called:
last_e = None
for fixture_name, tear_down_not_called in six.iteritems(_test_fixture_setup_called):
# If a -k or --pysoa-test-case filter was used to deselect most tests, the last test in a fixture may
# not run, in which case the fixture won't be properly torn down. The only way to solve this is to,
# on class tear-down, check for any not-torn-down fixtures and tear them down. However, this is not
# the ideal time to tear down a fixture, so we only use this for abnormal tear-downs, not for all
# fixture tear-downs.
if tear_down_not_called:
self, test_fixture = tear_down_not_called
try:
self._run_directive_hook('tear_down_test_fixture', test_fixture)
self.tear_down_test_fixture(test_fixture)
except Exception as e:
last_e = e
if last_e:
raise last_e
super(ServicePlanTestCase, cls).teardown_class()
def set_up_test_fixture(self, test_fixture, **kwargs):
# type: (TestFixture, **Any) -> None
"""
This method is invoked once for each fixture file, before any test cases in that fixture file are run.
:param test_fixture: List of test cases in this fixture
"""
def tear_down_test_fixture(self, test_fixture, **kwargs):
# type: (TestFixture, **Any) -> None
"""
This method is invoked once for each fixture file, after all test cases in the fixture file have run.
:param test_fixture: List of test cases in this fixture
"""
def setup_method(self):
# type: () -> None
"""
This method is invoked once for each test case in a fixture file, before the test is run. It is also run once
before each ``test_`` method in your test case, if you have any.
"""
case_data = getattr(self, '_pytest_first_fixture_case', None) # type: Optional[FixtureTestCaseData]
if case_data:
getattr(self.__class__, '_test_fixture_setup_called')[case_data.fixture_name] = self, case_data.test_fixture
self.set_up_test_fixture(case_data.test_fixture)
self._run_directive_hook('set_up_test_fixture', case_data.test_fixture)
getattr(self.__class__, '_test_fixture_setup_succeeded')[case_data.fixture_name] = True
super(ServicePlanTestCase, self).setup_method()
def teardown_method(self):
# type: () -> None
"""
This method is invoked once for each test case in a fixture file, after the test is run. It is also run once
after each ``test_`` method in your test case, if you have any.
"""
outer_exception = None # type: Optional[BaseException]
try:
super(ServicePlanTestCase, self).teardown_method()
except BaseException as e:
outer_exception = e
raise
finally:
case_data = getattr(self, '_pytest_last_fixture_case', None) # type: Optional[FixtureTestCaseData]
if case_data:
try:
getattr(self.__class__, '_test_fixture_setup_succeeded')[case_data.fixture_name] = False
getattr(self.__class__, '_test_fixture_setup_called')[case_data.fixture_name] = False
self._run_directive_hook('tear_down_test_fixture', case_data.test_fixture)
self.tear_down_test_fixture(case_data.test_fixture)
except KeyboardInterrupt:
if outer_exception:
# If an error happened in TRY 1, raise it instead of the interrupt so no mask
raise outer_exception
raise
except BaseException:
if not outer_exception:
raise # If an error did not happen in TRY 1, just raise the tear-down error
self.add_error(*sys.exc_info()) # Otherwise, record the tear-down error so no mask
def set_up_test_case(self, test_case, test_fixture, **kwargs):
# type: (TestCase, TestFixture, **Any) -> None
"""
This method is invoked immediately after `setup_method` and before the test case is run.
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
"""
def tear_down_test_case(self, test_case, test_fixture, **kwargs):
# type: (TestCase, TestFixture, **Any) -> None
"""
This method is invoked immediately before `teardown_method` and after the test case is run.
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
"""
def set_up_test_case_action(self, action_name, action_case, test_case, test_fixture, **kwargs):
# type: (six.text_type, ActionCase, TestCase, TestFixture, **Any) -> None
"""
This method is invoked once fear each service action run within a test case, before the action is called.
:param action_name: The name of the action being run (as described in the test case in the fixture)
:param action_case: The directive instructions for running and asserting this specific action
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
"""
def tear_down_test_case_action(self, action_name, action_case, test_case, test_fixture, **kwargs):
# type: (six.text_type, ActionCase, TestCase, TestFixture, **Any) -> None
"""
This method is invoked once fear each service action run within a test case, before the action is called.
:param action_name: The name of the action being run (as described in the test case in the fixture)
:param action_case: The directive instructions for running and asserting this specific action
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
"""
@classmethod
def get_fixture_test_information(cls):
# type: () -> Generator[FixtureTestCaseData, None, None]
"""
Generates fixture test case information used for executing fixture test cases. Acts as a generator that yields
``FixtureTestCaseData`` instances, which include the test name, description, fixture name, fixture file name,
fixture line number on which the test case begins, whether the fixture is skipped and the reason for its being
skipped, and a callable that can be used to invoke the test.
Tests are returned in the alphabetical order of the fixture file names and then the order the tests appear
in the fixture files.
:return: a generator of fixture test case information.
"""
for fixture_file_name in cls.get_fixture_file_names():
fixture_name, _ = os.path.splitext(os.path.basename(fixture_file_name))
test_fixture_results = [] # type: TestFixtureResults
last_case_data = None # type: Optional[FixtureTestCaseData]
fixture_parser = ServiceTestPlanFixtureParser(fixture_file_name, fixture_name)
test_fixture = fixture_parser.parse_test_fixture()
for test_case in test_fixture:
test_function = cls._create_test_function(
description=test_case['description'],
fixture_name=fixture_name,
test_case=test_case,
test_fixture=test_fixture,
test_fixture_results=test_fixture_results,
)
skip = test_case.get('skip', None) # type: Optional[six.text_type]
case_data = FixtureTestCaseData(
name=test_case['name'],
description=test_case['description'],
fixture_name=fixture_name,
fixture_file=fixture_file_name,
line_number=test_case['line_number'],
skip=skip,
callable=test_function,
test_fixture=test_fixture,
is_first_fixture_case=last_case_data is None,
is_last_fixture_case=False,
)
if not skip:
last_case_data = case_data
yield case_data
if last_case_data:
setattr(last_case_data.callable, '_last_fixture_test', True)
last_case_data.is_last_fixture_case = True
@classmethod
def get_fixture_file_names(cls):
# type: () -> Union[List[six.text_type], Tuple[six.text_type, ...]]
"""
Generate the list of fixture files to run. If ``cls.custom_fixtures`` has a value, its contents will be
returned directly. Otherwise, all fixtures in ``cls.fixture_path`` will be loaded based on
``cls.fixture_regex``.
Results will be returned as a list or tuple, e.g. ``['full_path_to_fixture_1', 'full_path_to_fixture_2']``.
:return: An alphabetically-sorted list of all fixture file names or the custom fixtures, if specified
"""
if cls.custom_fixtures:
return cls.custom_fixtures
if not cls.fixture_path:
return []
if not os.path.isdir(cls.fixture_path):
raise FixtureLoadError(
'Tried loading fixtures from "{}", however this path does not exist. Please specify the correct path '
'by setting `cls.fixture_path`.'.format(cls.fixture_path)
)
fixture_files = []
for directory_path, _, files in os.walk(cls.fixture_path):
for fixture_file in files:
if cls.fixture_regex.search(fixture_file):
fixture_files.append(directory_path + '/' + fixture_file)
if not fixture_files:
raise FixtureLoadError(
'Could not find any fixture files in `cls.fixture_path` "{path}" that matched `cls.fixture_regex` '
'"{regex}". To customize this regex, please override `cls.fixture_regex`.'.format(
path=cls.fixture_path,
regex=cls.fixture_regex.pattern,
)
)
return sorted(fixture_files)
@staticmethod
def _create_test_function(
description, # type: six.text_type
fixture_name, # type: six.text_type
test_case, # type: TestCase
test_fixture, # type: TestFixture
test_fixture_results, # type: TestFixtureResults
): # type: (...) -> FixtureTestCaseFunction
"""
This method creates a test case function, which the PyTest plugin binds to the test case class to make it a
method of the class, and which PyTest later invokes to run the test case.
:param description: The test description, to which the created test function's ``__doc__`` will be set
:param fixture_name: The fixture name
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
:param test_fixture_results: List of all action-call results in the entire fixture
:return: The test function, whose first argument must be an instance of test class inheriting from
``ServicePlanTestCase``, and which accepts other args and kwargs that are currently ignored but
reserved for future use.
"""
def test_function(self, *args, **kwargs):
# type: (ServicePlanTestCase, *Any, **Any) -> None
"""
This guy does the actual work of running a test case, and is invoked by PyTest when the time comes.
:param self: The test case instance
"""
# noinspection PyUnusedLocal
# This instructs the traceback manipulator that this frame belongs to test_function, which is simpler than
# having it analyze the code path details to determine the frame location.
_test_function_frame = True # noqa F841
if not hasattr(self.__class__, '_test_fixture_setup_succeeded'):
setattr(self.__class__, '_test_fixture_setup_succeeded', {})
_test_fixture_setup_succeeded = getattr(self.__class__, '_test_fixture_setup_succeeded')
if not _test_fixture_setup_succeeded.get(fixture_name, False):
# If the fixture was not successfully set up, then fixture setup must have failed on the first test, so
# all remaining tests in this fixture are also invalid.
raise StatusError('Test fixture {} not set up'.format(fixture_name))
outer_exception = None # type: Optional[BaseException]
try: # LABEL: 1
# First, we call the fixture test case setup on the class and on all directives
self.set_up_test_case(test_case, test_fixture)
self._run_directive_hook('set_up_test_case', test_case, test_fixture)
try: # LABEL: 2
# Now we can actually run the test!
self._run_test_case(test_case, test_fixture, test_fixture_results, *args, **kwargs)
except BaseException as e:
# We're only catching (everything) so that we can record which error happened in TRY 1
outer_exception = e
raise
finally:
# noinspection PyBroadException
try: # LABEL: 3
# Now we need to call the fixture test case teardown on the class and on all directives
self._run_directive_hook('tear_down_test_case', test_case, test_fixture)
self.tear_down_test_case(test_case, test_fixture)
except KeyboardInterrupt:
if outer_exception:
# If an error happened in TRY 2, raise it instead of the interrupt so we don't mask it
raise outer_exception
raise
except BaseException:
if not outer_exception:
raise # If an error did not happen in TRY 2, just raise the tear-down error
self.add_error(*sys.exc_info()) # Otherwise, record the tear-down error so we don't mask
except BaseException as e:
outer_exception = e
raise
finally:
if getattr(test_function, '_last_fixture_test'):
# If this is the last fixture test case, we need to assert and clean up the fixture
# noinspection PyBroadException
try: # LABEL: 4
self._run_directive_hook('assert_test_fixture_results', test_fixture_results, test_fixture)
except KeyboardInterrupt:
if outer_exception:
# If an error happened in TRY 1 - 3, raise it instead of the interrupt so we don't mask
raise outer_exception
raise
except AssertionError:
# If the tear-down asserts raised an assertion error
if not outer_exception:
raise # If an error did not happen in TRY 1 - 3, just raise the assertion error
self.add_error(*sys.exc_info()) # Otherwise, record the assertion error so we don't mask
except BaseException:
if not outer_exception:
raise # If an error did not happen in TRY 1 - 3, just raise the on-assert error
self.add_error(*sys.exc_info()) # Otherwise, record the tear-down error so we don't mask
test_function.__doc__ = description
setattr(test_function, '_last_fixture_test', False)
return test_function
@classmethod
def _run_directive_hook(cls, hook, *args, **kwargs):
# type: (six.text_type, *Any, **Any) -> None
"""
Runs the named hook method on all registered directives using the given positional and keyword arguments.
:param hook: The name of the hook
"""
if not cls._all_directives:
raise DirectiveError('FATAL: No directives found!')
for directive_class in cls._all_directives:
directive = directive_class()
if not hasattr(directive, hook):
raise DirectiveError('Directive class {} has no method {}.'.format(directive_class.__name__, hook))
getattr(directive, hook)(*args, **kwargs)
def _run_test_case(self, test_case, test_fixture, test_fixture_results, *_, **__):
# type: (TestCase, TestFixture, TestFixtureResults, *Any, **Any) -> None
"""
This does the actual work of running the test case.
:param test_case: The directive instructions to run and assert this specific test case
:param test_fixture: List of test cases in this fixture
:param test_fixture_results: List of all action-call results in the entire fixture
"""
# noinspection PyUnusedLocal
# This instructs the traceback manipulator that this frame belongs to _run_test_case, which is simpler than
# having it analyze the code path details to determine the frame location.
_run_test_case_frame = True # noqa F841
action_results = {} # type: ActionResults
action_response_bodies = {} # type: Dict[six.text_type, Optional[Body]]
test_fixture_results.append(action_results)
assert self.server_class is not None
assert self.server_class.action_class_map is not None
assert self.server_class.service_name is not None
for action_path in test_case['actions']:
action_name, action_index = action_path.split('.')
action_case = test_case[action_path]
if 'inputs' in action_case:
substitute_variables(action_case['inputs'], action_response_bodies, self.model_constants)
if 'job_control_inputs' in action_case:
substitute_variables(action_case['job_control_inputs'], action_response_bodies, self.model_constants)
if 'job_context_inputs' in action_case:
substitute_variables(action_case['job_context_inputs'], action_response_bodies, self.model_constants)
self.set_up_test_case_action(action_name, action_case, test_case, test_fixture)
self._run_directive_hook('set_up_test_case_action', action_name, action_case, test_case, test_fixture)
stub_context = self._WrapperContextManager() # it's a no-op with no arguments
if (
action_name not in self.server_class.action_class_map and # if the server doesn't have this action
action_name not in ('status', 'introspect') and # if the action isn't one of the built-in actions
hasattr(self, '_process_stub_action_{}'.format(action_name)) # if the test job has a mock action
):
# Hook for custom, test-only actions that are not real commands on the service. Custom actions must
# must work the same as side-effect functions on stub_action (must accept a request body dict and
# return a response body dict, `ActionResponse`, or `JobResponse` or raise an `ActionError` or
# `JobError`).
stub_context = self._WrapperContextManager(
stub_action(self.server_class.service_name, action_name),
getattr(self, '_process_stub_action_{}'.format(action_name)),
)
with stub_context:
job_response = self.client.call_actions(
service_name=self.server_class.service_name,
actions=[{'action': action_name, 'body': action_case.get('inputs', {})}],
raise_job_errors=False,
raise_action_errors=False,
context=action_case.get('job_context_inputs', {}),
control_extra=action_case.get('job_control_inputs', {}),
)
action_results[action_path] = job_response.actions[0] if job_response.actions else None
action_response_bodies[action_path] = job_response.actions[0].body if job_response.actions else None
substitute_variables(action_response_bodies, action_response_bodies, self.model_constants)
substitute_variables(action_case, action_response_bodies, self.model_constants)
try:
self._run_directive_hook(
'assert_test_case_action_results',
action_name,
action_case,
test_case,
test_fixture,
action_results[action_path],
job_response,
action_path,
)
finally:
try:
self._run_directive_hook(
'tear_down_test_case_action',
action_name,
action_case,
test_case,
test_fixture,
)
finally:
self.tear_down_test_case_action(action_name, action_case, test_case, test_fixture)
self._run_directive_hook('assert_test_case_results', action_results, test_case, test_fixture)
@staticmethod
def add_error(exc_type, exc_value, traceback_value):
# type: (Optional[Type[BaseException]], Optional[BaseException], Optional[TracebackType]) -> None
if exc_type or exc_value:
sys.stderr.write('Additionally, {} error occurred: {}'.format(
'a `{}`'.format(exc_type.__name__) if exc_type else 'an unknown',
exc_value if exc_value else 'No details',
))
sys.stderr.write('\n')
sys.stderr.flush()
if traceback_value:
traceback.print_tb(traceback_value, None, sys.stderr)
sys.stderr.flush()
class _WrapperContextManager(object):
def __init__(self, stub_action_context=None, mock_action_side_effect=None):
# type: (stub_action, Any) -> None
self._stub_action_context = stub_action_context
self._mock_action_side_effect = mock_action_side_effect
def __enter__(self): # type: () -> Optional[mock.MagicMock]
if self._stub_action_context:
mock_action = self._stub_action_context.__enter__()
mock_action.side_effect = self._mock_action_side_effect
return mock_action
return None
def __exit__(self, exc_type=None, exc_value=None, traceback=None):
# type: (Any, Any, Any) -> Literal[False]
if self._stub_action_context:
return self._stub_action_context.__exit__(exc_type, exc_value, traceback)
# noinspection PyTypeChecker
return False