ó
E®Uc           @   sr   d  Z  d d l m Z d d l m Z m Z d d l m Z m Z d e f d „  ƒ  YZ	 d e f d „  ƒ  YZ
 d	 S(
   s   Tests for samba.dceprc.rpcecho.iÿÿÿÿ(   t   echo(   t   ndr_packt
   ndr_unpack(   t   RpcInterfaceTestCaset   TestCaset   RpcEchoTestsc           B   sY   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d „  Z
 RS(	   c         C   s2   t  t |  ƒ j ƒ  t j d |  j ƒ  ƒ |  _ d  S(   Ns   ncalrpc:(   t   superR   t   setUpR    t   rpcechot   get_loadparmt   conn(   t   self(    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyR      s    c         C   sD   t  j d |  j ƒ  d |  j ƒ|  _ |  j d |  j j d ƒ ƒ d  S(   Ns   ncalrpc:t   basis_connectioni   i   (   R    R   R	   R
   t   conn2t   assertEqualst   AddOne(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_two_contexts   s    $c         C   s   |  j  d |  j j ƒ d  S(   Ns$   60a15ec5-4de8-11d7-a637-005056a20182i   (   s$   60a15ec5-4de8-11d7-a637-005056a20182i   (   R   R
   t   abstract_syntax(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_abstract_syntax#   s    	c         C   s    |  j  d |  j j d ƒ ƒ d  S(   Ni   i   (   R   R
   R   (   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_addone'   s    c         C   s2   |  j  d d d g |  j j d d d g ƒ ƒ d  S(   Ni   i   i   (   R   R
   t   EchoData(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_echodata*   s    c         C   s    |  j  d |  j j d ƒ ƒ d  S(   Nu   foobar(   R   R
   t   TestCall(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt	   test_call-   s    c         C   sZ   t  j ƒ  } d | _ d d d d g | _ |  j j | ƒ } |  j d d g | j ƒ d  S(   Ni   i   i   i   i   i    (   R    t   Surroundingt   xt   surroundingR
   t   TestSurroundingR   (   R   t   surrounding_structt   y(    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_surrounding0   s
    	c         C   s-   |  j  d |  j j d t d ƒ d ƒ ƒ d  S(   Ns      i    i   (   R   R
   t   requestt   chr(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_manual_request7   s    c         C   s   |  j  d  |  j j ƒ d  S(   N(   R   t   NoneR
   t   server_name(   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_server_name:   s    (   t   __name__t
   __module__R   R   R   R   R   R   R   R!   R$   (    (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyR      s   								t   NdrEchoTestsc           B   s   e  Z d  „  Z d „  Z RS(   c         C   s/   t  j ƒ  } d | _ |  j d t | ƒ ƒ d  S(   Ni*   s   *(   R    t   info1t   vR   R   (   R   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_info1_push@   s    	c         C   s)   t  t j d ƒ } |  j | j d ƒ d  S(   Nt   BiB   (   R   R    R(   R   R)   (   R   R   (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   test_info1_pullE   s    (   R%   R&   R*   R,   (    (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyR'   >   s   	N(   t   __doc__t   samba.dcerpcR    t	   samba.ndrR   R   t   samba.testsR   R   R   R'   (    (    (    s>   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/rpcecho.pyt   <module>   s
   %