ó
uæîUc           @   sO   d  Z  d d l m Z m Z d d l m Z d „  Z d e f d „  ƒ  YZ d S(   s   Tests for samba.dcerpc.sam.iÿÿÿÿ(   t   samrt   security(   t   RpcInterfaceTestCasec         C   sW   |  \ } } } g  } x; t  | ƒ D]- } | j | j | j | j | j f ƒ q" W| S(   N(   t   ranget   appendt   entriest   idxt   name(   t   .0t   handlet   arrayt   num_entriest   rett   x(    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyt   toArray   s
    +t	   SamrTestsc           B   s,   e  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   get_loadparmt   conn(   t   self(    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyR   #   s    c         C   s.   |  j  j d  d d t j ƒ  ƒ \ } } } d  S(   Ni    i   (   R   t   Connect5t   NoneR    t   ConnectInfo1(   R   t   levelt   infoR	   (    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyt   test_connect5'   s    c         C   s/   |  j  j d  t j ƒ } |  j | d  k	 ƒ d  S(   N(   R   t   Connect2R   R   t   SEC_FLAG_MAXIMUM_ALLOWEDt
   assertTrue(   R   R	   (    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyt   test_connect2*   s    c         C   sJ   |  j  j d  t j ƒ } t |  j  j | d d ƒ ƒ } |  j  j | ƒ d  S(   Ni    l   ÿÿ (   R   R   R   R   R   R   t   EnumDomainst   Close(   R   R	   t   domains(    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyt   test_EnumDomains.   s    (   t   __name__t
   __module__R   R   R   R"   (    (    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyR   !   s   			N(   t   __doc__t   samba.dcerpcR    R   t   samba.testsR   R   R   (    (    (    s:   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/sam.pyt   <module>   s   	