
EUc           @   s]   d  Z  d d l Z d d l Z d d l Td d l Z d   Z d e j j f d     YZ d S(   s   Tests for samba.kcc.graphiN(   t   *c         C   s   |  d  k r d  Sd d l m } | j   } d | _ d | _ d | _ | j   } d | _ d | _	 | g | _
 | j   } |  | _ | g | _ | S(   Ni(   t   drsblobsi   i    i   i   (   t   Nonet   samba.dcerpcR   t   schedulet   sizet	   bandwidtht   numberOfSchedulest   scheduleHeadert   typet   offsett   headerArrayt   scheduleSlotst   slotst	   dataArray(   t   timesR   R   t   headert   data(    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/kcc/graph.pyt   ntdsconn_schedule   s    						t   GraphFunctionTestsc           B   s   e  Z d    Z d   Z RS(   c         C   s   d g d } x d g d d f d g d d f d g d d f d d	 g d
 d f t  d  d d f t  d  d d f f D]" \ } } |  j t |  |  q} Wd  S(   Ni   iT   i   i   i   i   i   i   i3   i*   i   i   i   i	   i   i  iP  i   il   iT   (   t   ranget   assertEqualst   total_schedule(   t   selfR   t   total(    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/kcc/graph.pyt   test_total_schedule1   s    #c         C   s   x d g d d g d f d  d g d f d g d d g d f d d g d d	 g d f t d
  d d d d d d d d g d f f D]. \ } } t |  } |  j t |  |  q Wd  S(   Ni   i   i   iT   i   if   i   i
   i:   i   i   i#   iE   i`   i   i4   iV   i   (   R   R   R   R   t   convert_schedule_to_repltimes(   R   t   ntdsconn_timest	   repltimesR   (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/kcc/graph.pyt"   test_convert_schedule_to_repltimes<   s    /(   t   __name__t
   __module__R   R   (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/kcc/graph.pyR   /   s   	(	   t   __doc__t   sambat   samba.testst   samba.kcc.grapht	   itertoolsR   t   testst   TestCaseR   (    (    (    s9   /usr/lib/python2.7/dist-packages/samba/tests/kcc/graph.pyt   <module>   s   
	