ó
E®Uc           @   s@   d  Z  d d l m Z d d l m Z d e f d „  ƒ  YZ d S(   s    Tests for samba.dcerpc.registry.iÿÿÿÿ(   t   winreg(   t   RpcInterfaceTestCaset   WinregTestsc           B   s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C   s;   t  t |  ƒ j ƒ  t j d |  j ƒ  |  j ƒ  ƒ |  _ d  S(   Ns   ncalrpc:(   t   superR   t   setUpR    t   get_loadparmt   get_credentialst   conn(   t   self(    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyR      s    c         C   s   |  j  j d  t j t j Bƒ S(   N(   R   t   OpenHKLMt   NoneR    t   KEY_QUERY_VALUEt   KEY_ENUMERATE_SUB_KEYS(   R   (    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyt   get_hklm   s    c         C   s3   |  j  j d  t j t j Bƒ } |  j  j | ƒ d  S(   N(   R   R	   R
   R    R   R   t   CloseKey(   R   t   handle(    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyt	   test_hklm#   s    c         C   sE   |  j  ƒ  } |  j j | ƒ } |  j t | j ƒ |  j j | ƒ d  S(   N(   R   R   t
   GetVersiont   assertEqualst   intt	   __class__R   (   R   R   t   version(    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyt   test_getversion(   s    c         C   sd   |  j  j d  t j t j Bƒ } |  j  j | t j ƒ  ƒ } |  j d t	 | ƒ ƒ |  j  j
 | ƒ d  S(   Ni	   (   R   R	   R
   R    R   R   t   QueryInfoKeyt   StringR   t   lenR   (   R   R   t   x(    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyt   test_getkeyinfo.   s
    (   t   __name__t
   __module__R   R   R   R   R   (    (    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyR      s
   				N(   t   __doc__t   samba.dcerpcR    t   samba.testsR   R   (    (    (    s?   /usr/lib/python2.7/dist-packages/samba/tests/dcerpc/registry.pyt   <module>   s   