Current File : //usr/lib/python2.7/site-packages/salt/modules/testinframod.pyc
�
���^c @@ sC d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z e j
e � Z y& d d l
Z
d d l
m Z e Z Wn e k
r� e Z n Xg Z d Z d Z d e f d � � YZ d � Z e d
� Z d � Z d � Z d d
� Z d � Z d � Z d d � Z d � Z! e r?e! � n d S( uu
This module exposes the functionality of the TestInfra library
for use with SaltStack in order to verify the state of your minions.
In order to allow for the addition of new resource types in TestInfra this
module dynamically generates wrappers for the various resources by iterating
over the values in the ``__all__`` variable exposed by the testinfra.modules
namespace.
i ( t absolute_importt unicode_literalst print_functionN( t modulesu testinfrau local://t InvalidArgumentErrorc B@ s e Z RS( ( t __name__t
__module__( ( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyR ! s c C@ s t r
t St d f S( Nu& The Testinfra package is not available( t TESTINFRA_PRESENTt __virtualname__t False( ( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt __virtual__% s c C@ s" t j | � } | j t | � � S( u Retrieve the correct module implementation determined by the backend
being used.
:param module_name: TestInfra module to retrieve
:param backend: string representing backend for TestInfra
:returns: desired TestInfra module object
:rtype: object
( t testinfrat get_backendt
get_modulet _to_pascal_case( t module_namet backendt backend_instance( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _get_module+ s
c C@ sg t j d d | � } g } x< | j � D]. } | j | d j � � | j | d � q( Wd j | � S( u� Convert a snake_case string to its PascalCase equivalent.
:param snake_case: snake_cased string to be converted
:returns: PascalCase string
:rtype: str
u _u i i u ( t ret subt splitt appendt uppert join( t
snake_caset
space_caset wordlistt word( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyR 9 s c C@ s+ t j d d � | � } | j � j d � S( u� Convert a PascalCase string to its snake_case equivalent.
:param pascal_case: PascalCased string to be converted
:returns: snake_case string
:rtype: str
u (^|[a-z])([A-Z])c S@ s. d j | j d � j � | j d � j � � S( Nu {0}_{1}i i ( t formatt groupt lower( t match( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt <lambda>R s u _( R R R t strip( t pascal_caseR ( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _to_snake_caseI s c C@ sH t j d | | � y t | | � } WnP t k
rx y t | | � } Wqy t k
rt t d j | | � � � qy Xn Xt | t � r� | j | � St | t j
t j f � r@| s� t d j | | � � � n y t | | � | d � SWqDt k
rt d j | � � � qDt k
r<t d j | | � � � qDXn | Sd S( u� Given a TestInfra module object, an instance of that module, and a
method name, return the result of executing that method against the
desired module.
:param module: TestInfra module object
:param module_instance: TestInfra module instance
:param method_name: string representing the method to be executed
:param method_arg: boolean or dictionary object to be passed to method
:returns: result of executing desired method with supplied argument
:rtype: variable
u Trying to call %s on %su= The {0} module does not have any property or method named {1}u@ {0} is a method of the {1} module. An argument dict is required.u parameteru< The argument dict supplied has no key named "parameter": {0}N( t logt debugt getattrt AttributeErrorR R t
isinstancet propertyt fgett typest
MethodTypet FunctionTypet KeyErrort None( t module_t module_instancet method_namet
method_argt
method_obj( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _get_method_resultX s6
c C@ s� t j d | | � t | t � r, | | k St | t � r� y t t | d � } Wnp t k
r� | j d � d k r� t j
} q� t d j | j d � � � � n! t
k
r� t j d � � n X| | d | � St d j t | � � � � d S( ui Given the result of a method, verify that it matches the expectation.
This is done by either passing a boolean value as an expecation or a
dictionary with the expected value and a string representing the desired
comparison, as defined in the `operator module <https://docs.python.org/2.7/library/operator.html>`_
(e.g. 'eq', 'ge', etc.). The ``re.search`` function is also available with
a comparison value of ``search``.
:param expected: boolean or dict
:param result: return value of :ref: `_get_method_result`
:returns: success or failure state of assertion
:rtype: bool
u& Expected result: %s. Actual result: %su
comparisonu searchu( Comparison {0} is not a valid selection.uo The comparison dictionary provided is missing expected keys. Either "expected" or "comparison" are not present.u expectedu& Expected bool or dict but received {0}N( R% R& R) t boolt dictR' t operatorR( t getR t searchR R R/ t exceptiont TypeErrort type( t expectedt resultt
comparison( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _apply_assertion� s"
c C@ s | j S( N( t __doc__( R1 ( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt
_build_doc� s c @ sn t j d � � � f d � } | } | d k r@ t | � } n | j } t j | j | j | | j | j
� S( uV
This will generate a function that is registered as either ``module_name``
or ``name``. The contents of the function will be ``_run_tests``. This
will translate the Testinfra module into a salt function and the methods
and properties of that module will be exposed as attributes of the salt
function that is generated. This allows for writing unit tests for a
configured minion using states in the same way as it is configured
Example:
```yaml
minion_is_installed:
testinfra.package:
- name: salt-minion
- is_installed: True
minion_is_running:
testinfra.service:
- name: salt-minion
- is_running: True
- is_enabled: True
file_has_contents:
testinfra.file:
- name: /etc/salt/minion
- exists: True
- contains:
parameter: master
expected: True
comparison: is_
python_is_v2:
testinfra.package:
- name: python
- is_installed: True
- version:
expected: '2.7.9-1'
comparison: eq
```
u$ Generating function for testinfra.%sc
@ s� t } g } g } y3 t j d � � t � � } t j d | j � Wn1 t k
rx t j d � � t } | | | f SXt t d � r� t j
| � } | j } nW t | j
t j � r� t j | j
� } n$ t | d � r� t j | j � } n | j } t j d � | � i } x6 t | � j t | � � D] } | j | � | | <q/Wy1 t | � d k rs| | | � }
n | � }
Wn! t k
r�t j d � � n Xi } t j d | � x. | D]& } | j d
� s�| | | | <q�q�Wt j d | � x� | j � D]� \ }
} t | |
|
| � } t | | � } | sut } | j d j d
� d | d |
d | d | � � q| j d j d
� d | d |
d | d | � � qW| | | f S( Nu Retrieving %s module.u Retrieved module is %su@ The %s module is not supported for this backend and/or platform.u signatureu __call__u$ Parameters accepted by module %s: %si u Module failed to instantiateu Called methods are: %su _u Valid methods are: %su; Assertion failed: {modname} {n} {m} {a}. Actual result: {r}t modnamet nt mt at ru< Assertion passed: {modname} {n} {m} {a}. Actual result: {r}( t TrueR% R& R t __dict__t NotImplementedErrorR< R t hasattrt inspectt signaturet
parametersR) t __init__R, R- t
getargspect __call__t argst sett intersectiont popt lenR= t
startswitht itemsR6 RB R R ( t namet methodst successt pass_msgst fail_msgst modt mod_sigRP t additional_argst argt modinstancet
valid_methodst meth_namet methR@ t assertion_result( R ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt
_run_tests� sb
"
%&N( R% R& R0 t strR R, R. t __code__t __globals__t __defaults__t __closure__( R R[ Ri t func( ( R s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _copy_function� s )9 c C@ s� y& g t j D] } t | � ^ q
} Wn- t k
rU g t j D] } | ^ q@ } n XxF | D]> } t | | � } t | � | _ t j | � | t � | <q] Wd S( u�
Iterate through the exposed Testinfra modules, convert them to salt
functions, and then register them in the module namespace so that they
can be called via salt.
N( R t __all__R$ R( Rp RD RC R t globals( R1 t modules_t mod_namet mod_func( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt _register_functions% s &
(" RC t
__future__R R R RN t loggingR9 R R, t getLoggerR R% R R RJ R t ImportErrorR Rq R t default_backendt ExceptionR R
R R R$ R0 R6 RB RD Rp Rv ( ( ( s= /usr/lib/python2.7/site-packages/salt/modules/testinframod.pyt <module> s8
/ ( r
Mini Shell