ó
E®Uc           @   s\   d  Z  d d l Z d d l m Z d d l m Z d d l m Z d e f d „  ƒ  YZ d S(   s   Tests for samba.messaging.iÿÿÿÿN(   t	   Messaging(   t   TestCase(   t	   server_idt   MessagingTestsc           B   s>   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         O   s    t  j j ƒ  | d <t | | Ž  S(   Nt   lp_ctx(   t   sambat   testst   env_loadparmR    (   t   selft   argst   kwargs(    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   get_context   s    c         C   s8   |  j  ƒ  } d „  } | j | ƒ } | j | | ƒ d  S(   Nc           S   s   d  S(   N(    (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   callback"   s    (   R   t   registert
   deregister(   R   t   xR   t   msg_type(    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   test_register    s    	c         C   s,   |  j  ƒ  } |  j t | j ƒ  t ƒ ƒ d  S(   N(   R   t
   assertTruet
   isinstancet   irpc_all_serverst   list(   R   R   (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   test_all_servers'   s    c         C   sI   |  j  ƒ  } x6 | j ƒ  D]( } |  j t | j | j ƒ t ƒ ƒ q Wd  S(   N(   R   R   R   R   t   irpc_servers_bynamet   nameR   (   R   R   R   (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   test_by_name+   s    c         C   s)   |  j  ƒ  } |  j t | j t ƒ ƒ d  S(   N(   R   R   R   R   (   R   R   (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   test_assign_server_id0   s    c            s–   |  j  d	 ƒ ‰  ‡  f d †  } d „  } ˆ  j | ƒ } ˆ  j | ƒ } d „  } |  j  d
 ƒ } | j | ƒ } | j d | d ƒ | j d | d ƒ d  S(   Ni    i   c            s   ˆ  j  |  | ƒ d  S(   N(   t   send(   t   srct   data(   t
   server_ctx(    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   ping_callback6   s    c           S   s	   d GHd  S(   Ns   received exit(    (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   exit_callback8   s    c           S   s	   d GHd  S(   Ns   received pong(    (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   pong_callback=   s    i   t   testingt    (   i    i   (   i    i   (   i    i   (   i    i   (   R   R   R   (   R   R   R    t   msg_pingt   msg_exitR!   t
   client_ctxt   msg_pong(    (   R   s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   test_ping_speed4   s    		(   t   __name__t
   __module__R   R   R   R   R   R(   (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyR      s   					(	   t   __doc__R   t   samba.messagingR    t   samba.testsR   t   samba.dcerpc.server_idR   R   (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/messaging.pyt   <module>   s
   