Current File : /opt/hc_python/lib/python3.12/site-packages/nose/plugins/__pycache__//plugintest.cpython-312.pyc
�
H��g�4 � �6 � d Z ddlZddlZddlmZ ddlmZ ddgZddl m
Z
G d� d e � Z dd
l
mZ e Z G d � de � Z G d � d
e � Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zedk( rddlZ ej6 � yy# e$ r ddlmZ Y �rw xY w# e$ r eZY �aw xY w)a_
Testing Plugins
===============
The plugin interface is well-tested enough to safely unit test your
use of its hooks with some level of confidence. However, there is also
a mixin for unittest.TestCase called PluginTester that's designed to
test plugins in their native runtime environment.
Here's a simple example with a do-nothing plugin and a composed suite.
>>> import unittest
>>> from nose.plugins import Plugin, PluginTester
>>> class FooPlugin(Plugin):
... pass
>>> class TestPluginFoo(PluginTester, unittest.TestCase):
... activate = '--with-foo'
... plugins = [FooPlugin()]
... def test_foo(self):
... for line in self.output:
... # i.e. check for patterns
... pass
...
... # or check for a line containing ...
... assert "ValueError" in self.output
... def makeSuite(self):
... class TC(unittest.TestCase):
... def runTest(self):
... raise ValueError("I hate foo")
... return [TC('runTest')]
...
>>> res = unittest.TestResult()
>>> case = TestPluginFoo('test_foo')
>>> _ = case(res)
>>> res.errors
[]
>>> res.failures
[]
>>> res.wasSuccessful()
True
>>> res.testsRun
1
And here is a more complex example of testing a plugin that has extra
arguments and reads environment variables.
>>> import unittest, os
>>> from nose.plugins import Plugin, PluginTester
>>> class FancyOutputter(Plugin):
... name = "fancy"
... def configure(self, options, conf):
... Plugin.configure(self, options, conf)
... if not self.enabled:
... return
... self.fanciness = 1
... if options.more_fancy:
... self.fanciness = 2
... if 'EVEN_FANCIER' in self.env:
... self.fanciness = 3
...
... def options(self, parser, env=os.environ):
... self.env = env
... parser.add_option('--more-fancy', action='store_true')
... Plugin.options(self, parser, env=env)
...
... def report(self, stream):
... stream.write("FANCY " * self.fanciness)
...
>>> class TestFancyOutputter(PluginTester, unittest.TestCase):
... activate = '--with-fancy' # enables the plugin
... plugins = [FancyOutputter()]
... args = ['--more-fancy']
... env = {'EVEN_FANCIER': '1'}
...
... def test_fancy_output(self):
... assert "FANCY FANCY FANCY" in self.output, (
... "got: %s" % self.output)
... def makeSuite(self):
... class TC(unittest.TestCase):
... def runTest(self):
... raise ValueError("I hate fancy stuff")
... return [TC('runTest')]
...
>>> res = unittest.TestResult()
>>> case = TestFancyOutputter('test_fancy_output')
>>> _ = case(res)
>>> res.errors
[]
>>> res.failures
[]
>>> res.wasSuccessful()
True
>>> res.testsRun
1
� N)�warn)�StringIO� PluginTester�run)�getpidc �< � e Zd ZdZd� Zd� Zd� Zd� Zd
d�Zd� Z d� Z
y ) �MultiProcessFilea\
helper for testing multiprocessing
multiprocessing poses a problem for doctests, since the strategy
of replacing sys.stdout/stderr with file-like objects then
inspecting the results won't work: the child processes will
write to the objects, but the data will not be reflected
in the parent doctest-ing process.
The solution is to create file-like objects which will interact with
multiprocessing in a more desirable way.
All processes can write to this object, but only the creator can read.
This allows the testing system to see a unified picture of I/O.
c � � t � | _ t � j � | _ t � | _ d| _ y �Nr )r �_MultiProcessFile__master�Manager�Queue�_MultiProcessFile__queuer �_MultiProcessFile__buffer� softspace��selfs �F/opt/hc_python/lib/python3.12/site-packages/nose/plugins/plugintest.py�__init__zMultiProcessFile.__init__~ s- � � ���
� �y���(�� � �
��
���� c �> � t � | j k7 ry ddlm} ddlm} |t � } | j j � \ }}|dk( rd}||xx |z
cc< �3# |$ r Y nw xY wt |� D ] }| j j || � �" y )Nr )�Empty)� defaultdict� )g}Ô%�I�T) r r �queuer � collectionsr �strr �
get_nowait�sortedr �write)r r r �cache�pid�datas r �bufferzMultiProcessFile.buffer� s� � � �8�t�}�}� $� � �+��C� ���
� �L�L� 3� 3� 5� ��T� �b�y� !�� �#�J�$� �J� �� �
��
�� �%�=�C� �M�M� � ��s�� -� !s � A% �%A-�,A-c �l � ddl m} |� j }| j j ||f� y )Nr )�current_process)�multiprocessingr& � _identityr �put)r r# r& r" s r r zMultiProcessFile.write� s+ � � 4� � �)�)�� � � ���#�t��%r c �: � | j � | j S )z getattr doesn't work for iter())r$ r r s r �__iter__zMultiProcessFile.__iter__� s � � � � �
��}�}� r c �Z � | j � | j j ||� S �N)r$ r �seek)r �offset�whences r r. zMultiProcessFile.seek� s" � � � � �
��}�}�!�!�&�&�1�1r c �V � | j � | j j � S r- )r$ r �getvaluer s r r2 zMultiProcessFile.getvalue� s � � � � �
��}�}�%�%�'�'r c �. � t | j |� S r- )�getattrr )r �attrs r � __getattr__z MultiProcessFile.__getattr__� s � ��t�}�}�d�+�+r N)r ) �__name__�
__module__� __qualname__�__doc__r r$ r r+ r. r2 r6 r r r r r n s* � �� �.�(&� �2�(�,r r )r
c �> � e Zd ZdZdZdZdZi ZdZg Z dZ
d� Z d� Z d� Z
y)r a% A mixin for testing nose plugins in their runtime environment.
Subclass this and mix in unittest.TestCase to run integration/functional
tests on your plugin. When setUp() is called, the stub test suite is
executed with your plugin so that during an actual test you can inspect the
artifacts of how your plugin interacted with the stub test suite.
- activate
- the argument to send nosetests to activate the plugin
- suitepath
- if set, this is the path of the suite to test. Otherwise, you
will need to use the hook, makeSuite()
- plugins
- the list of plugins to make available during the run. Note
that this does not mean these plugins will be *enabled* during
the run -- only the plugins enabled by the activate argument
or other settings in argv or env will be enabled.
- args
- a list of arguments to add to the nosetests command, in addition to
the activate argument
- env
- optional dict of environment variables to send nosetests
Nc � � t �)a� returns a suite object of tests to run (unittest.TestSuite())
If self.suitepath is None, this must be implemented. The returned suite
object will be executed with all plugins activated. It may return
None.
Here is an example of a basic suite object you can return ::
>>> import unittest
>>> class SomeTest(unittest.TestCase):
... def runTest(self):
... raise ValueError("Now do something, plugin!")
...
>>> unittest.TestSuite([SomeTest()]) # doctest: +ELLIPSIS
<unittest...TestSuite tests=[<...SomeTest testMethod=runTest>]>
)�NotImplementedErrorr s r � makeSuitezPluginTester.makeSuite� s
� �$ "�!r c �T � ddl m} ddlm} ddlm} d}t
� } || j | || j �� �� }| j �| j |_ | j s| j � } || j ||d� � | _
t |� | _ y)
z7execute the plugin on the internal test suite.
r ��Config)� TestProgram��
PluginManagerN��plugins)�env�streamrF F)�argv�config�suite�exit)� nose.configrA � nose.corerB �nose.plugins.managerrD �BufferrG rF � ignoreFiles� suitepathr> rI �nose�AccessDecorator�output)r rA rB rD rK rH �confs r � _execPluginzPluginTester._execPlugin� s� � � '�)�6�������$�(�(� #� +�D�L�L� A�C�� � � � '� #� /� /�D� ��~�~��N�N�$�E� �T�Y�Y�t�5�%*�,�� �%�f�-�� r c � � d| j g| _ | j r%| j j | j � | j r%| j j | j � | j
� y)zUruns nosetests with the specified test suite, all plugins
activated.
� nosetestsN)�activaterI �args�extendrR �appendrW r s r �setUpzPluginTester.setUp sZ � � !�$�-�-�0�� � �9�9� �I�I� � �T�Y�Y� '� �>�>� �I�I� � �T�^�^� ,� ���r )r7 r8 r9 r: rZ rR r[ rG rI rF rQ r> rW r^ r r r r r � s< � � �B �H��I� �D�
�C� �D��G��K�"�(.�*
r c �, � e Zd ZdZdZd� Zd� Zd� Zd� Zy)rT Nc � � || _ |j d� |j � | _ |j d� y r )rH r. �read�_buf)r rH s r r zAccessDecorator.__init__ s, � � �� �� � �A���K�K�M�� �� � �A�r c � � || j v S r- �rb )r �vals r � __contains__z AccessDecorator.__contains__ s � ��d�i�i� � r c �, � t | j � S r- )�iterrH r s r r+ zAccessDecorator.__iter__ s � ��D�K�K� � r c � � | j S r- rd r s r �__str__zAccessDecorator.__str__ s � ��y�y�r ) r7 r8 r9 rH rb r rf r+ rj r r r rT rT s � �
�F� �D��
�!�r rT c # � K � g }| j d� D ]^ }|j |� |j � }|r$|j d� s�8|j d� r�Jdj |� �� g }�` |rdj |� �� yy�w)z9a bunch of === characters is also considered a blank lineTz===�=� N)�
splitlinesr] �strip�
startswith�join)�text�block�lines r �blankline_separated_blocksru # sx � �� � �E�����%��
� � �T���z�z�|���t���u�-�d�j�j��o��'�'�%�.� ��E� &�
��g�g�e�n� �
�s �A B�B� .Bc � � t j dt j t j z t j z � }g }t | � D ]# }|j
|j d|� � �% dj |� S )NaD
# Grab the traceback header. Different versions of Python have
# said different things on the first traceback line.
^(?P<hdr> Traceback\ \(
(?: most\ recent\ call\ last
| innermost\ last
) \) :
)
\s* $ # toss trailing whitespace on the header.
(?P<stack> .*?) # don't blink: absorb stuff until...
^(?=\w) # a line *starts* with alphanum.
.*?(?P<exception> \w+ ) # exception name
(?P<msg> [:\n] .*) # the rest
z"\g<hdr>\n...\n\g<exception>\g<msg>rm ) �re�compile�VERBOSE� MULTILINE�DOTALLru r] �subrq )�out� traceback_re�blocksrs s r �remove_stack_tracesr� 0 sn � ��:�:�
� �Z�Z�"�,�,�
&����
2�
4�L� �F�+�C�0���
�
�l�&�&�'L�e�T�U� 1�
�7�7�6�?�r c � � t j dt j t j z � }|j d| � S )Nz�
# Cut the file and line no, up to the warning name
^.*:\d+:\s
(?P<category>\w+): \s+ # warning category
(?P<detail>.+) $ \n? # warning message
^ .* $ # stack frame
z\g<category>: \g<detail>)rw rx ry rz r| )r} �warn_res r �simplify_warningsr� F s; � ��j�j� � �Z�Z�"�,�,�
&�
(�G� �;�;�2�C� 8�8r c �0 � t j dd| � S )Nz Ran (\d+ tests?) in [0-9.]+szRan \1 in ...s)rw r| �r} s r �remove_timingsr� Q s � �
�6�6�'�):�C� A� Ar c �d � t | � } t | � } t | � } | j � S )z6Modify nose output to make it easy to use in doctests.)r� r� r� ro r� s r � munge_nose_output_for_doctestr� V s, � �
�c�
"�C�
�C�
�C�
��
�C� �9�9�;�r c � � ddl m} ddlm} ddlm} t
� }d|vrJ|j dg � }t |t � r ||�� }|j di � } |||� � |d< d
|vrd d g|d
< ||d _
t j }t j } |j d
d� r|xt _
t _ d}
nd}
t dt d�� || i |�� |
r|t _ | t _
|j! � } t# t% | � � y# |
r|t _ | t _
w w xY w)a�
Specialized version of nose.run for use inside of doctests that
test test runs.
This version of run() prints the result output to stdout. Before
printing, the output is processed by replacing the timing
information with an ellipsis (...), removing traceback stacks, and
removing trailing whitespace.
Use this version of run wherever you are writing a doctest that
tests nose (or unittest) test result output.
Note: do not use doctest: +ELLIPSIS when testing nose output,
since ellipses ("test_foo ... ok") in your expected test runner
output may match multiple lines of output, causing spurious test
passes!
r �r r@ rC rJ rF rE rG )rG rF rI rY z-v�
buffer_allFTa The behavior of nose.plugins.plugintest.run() will change in the next release of nose. The current behavior does not correctly account for output to stdout and stderr. To enable correct behavior, use run_buffered() instead, or pass the keyword argument buffer_all=True to run().� )�
stacklevelN)rS r rM rA rO rD rP �pop�
isinstance�listrH �sys�stderr�stdoutr �DeprecationWarningr2 �printr� ) �arg�kwr rA rD r$ rF rG r� r� �restorer} s r r r ^ s% � �$ �"�2�
�X�F��r���&�&��B�'�� �g�t� $�#�G�4�G��f�f�U�B� �� �#�w�7��8� �
�R��!�4�(��6�
� �B�x�L�� �Z�Z�F�
�Z�Z�F� �v�v�l�E�"�"(�(��
�S�Z����� � >�
�A� /� � �S��B�� � �C�J� �C�J�
�/�/�
�C� �
'��
,�-�� � �C�J� �C�J� �s � D& �&Ec �$ � d|d< t | i |�� y )NTr� r� )r� r� s r � run_bufferedr� � s � ��B�|����O��Or �__main__) r: rw r� �warningsr �ior � ImportError�__all__�osr �objectr r' r
rP r rT ru r� r� r� r� r r� r7 �doctest�testmodr r r �<module>r� s� ��_�B
�
� � �� �5�
!�� �=,�v� =,�~�'�
�F�]�6� ]�@
�f�
�
��,9�A�
�9.�x� �z����G�O�O�� ��m � �� ��P � �
�F��s" �A= �B �= B �
B �B�B