
\c           @` s  d  d l  m Z m Z m Z d  d l m Z d  d l m Z d  d l m	 Z	 m
 Z
 m Z m Z m Z d  d l m Z d  d l m Z e
 d  Z e
 d  Z e
 d	  Z e
 d
  Z d e d e g Z d e d e d e g Z d d  Z d   Z d   Z d   Z d   Z d S(   i    (   t   print_functiont   divisiont   absolute_import(   t   sleep(   t   Client(   t   popent   get_certt   new_config_filet   tls_securityt   tls_only_config(   t   loop(   t   times   tls-ca-cert.pems   tls-cert.pems   tls-key.pems   tls-key-cert.pems   --tls-ca-files
   --tls-certs	   --tls-keyi   c         C` sO   t    } x? t |  j    d k  rJ t d  t    | d k  s t  q Wd  S(   Ni   g?i
   (   R   t   lent   ncoresR   t   AssertionError(   t   cR   t   start(    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt   wait_for_cores   s    	
c         C` ss   t  d d g t  W } t  d d d g t  5 } t d d |  d t    } t |  Wd  QXWd  QXWd  QXd  S(   Ns   dask-schedulers
   --no-bokehs   dask-workers   tls://127.0.0.1:8786R
   t   security(   R   t   tls_argsR   R   R   (   R
   t   st   wR   (    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt
   test_basic#   s    c         C` sv   t  d d g t  Z } t  d d d d g t  5 } t d d |  d t    } t |  Wd  QXWd  QXWd  QXd  S(   Ns   dask-schedulers
   --no-bokehs   dask-workers   --nannys   tls://127.0.0.1:8786R
   R   (   R   R   R   R   R   (   R
   R   R   R   (    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt
   test_nanny.   s    c         C` ss   t  d d g t  W } t  d d d g t  5 } t d d |  d t    } t |  Wd  QXWd  QXWd  QXd  S(   Ns   dask-schedulers
   --no-bokehs   dask-workers   tls://127.0.0.1:8786R
   R   (   R   t
   tls_args_2R   R   R   (   R
   R   R   R   (    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt   test_separate_key_cert9   s    c         C` s   t  t    r t d d d d g  S } t d d d g  5 } t d d |  d t    } t |  Wd  QXWd  QXWd  QXWd  QXd  S(	   Ns   dask-schedulers
   --no-bokehs   --hosts   tls://s   dask-workers   tls://127.0.0.1:8786R
   R   (   R   R	   R   R   R   R   (   R
   R   R   R   (    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt   test_use_config_fileD   s    N(   t
   __future__R    R   R   R   R   t   distributedR   t   distributed.utils_testR   R   R   R   R	   R
   t   distributed.metricst   ca_filet   certt   keyt   keycertR   R   R   R   R   R   R   (    (    (    sA   lib/python2.7/site-packages/distributed/cli/tests/test_tls_cli.pyt   <module>   s    (			