ó
CÿWc           @  sý   d  d l  m Z d  d l m Z y d  d l m Z Wn! e k
 rW d  d l m Z n Xd  d l Z d  d l Z d  d l	 Z	 d  d l
 Z
 d  d l m Z d  d l m Z d  d l m Z m Z d „  Z d	 e	 j f d
 „  ƒ  YZ e d k rù e	 j ƒ  n  d S(   iÿÿÿÿ(   t   unicode_literals(   t   ArgumentParser(   t   StringION(   t   add_subparser_modules(   t   print_colors(   t   Colort   ColorStreamc         C  s6   |  j  d ƒ } | j d ƒ | j d t j ƒ  ƒ d  S(   Nu   hellou   worldt   main(   t
   add_parsert   add_argumentt   set_defaultst   mockt   Mock(   t
   subparserst	   subparser(    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   add_hello_parser   s    t   Testc           B  sY   e  Z d  „  Z e j e j d k d ƒ d „  ƒ Z e j e j d k d ƒ d „  ƒ Z RS(   c         C  s   t  ƒ  } t j d ƒ ; } t j ƒ  } t | j _ | g | _ t | d  d ƒ Wd  QX| j	 d d g ƒ } |  j
 | j d ƒ d  S(   Nu   clyent.iter_entry_pointsu   entry_point_nameu   hellou   world(   R   R   t   patchR   R   t   loadt   return_valueR   t   Nonet
   parse_argst   assertEqualt   world(   t   selft   parsert   iter_entry_pointst   ept   args(    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   test_add_subparser_modules   s    	u   ntu#   Cannot colorize StringIO on Windowsc         C  sa   t  ƒ  } d „  | _ t | ƒ } t d d d d | ƒ| j ƒ  } | j ƒ  |  j d | ƒ d  S(   Nc           S  s   d S(   Niÿÿÿÿ(    (    (    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   <lambda>(   s    u   Are youu   {=okay!c:green}u   Annie?t   fileu   Are you [92mokay[0m Annie?
(   R   t   filenoR   R   t   getvaluet   closeR   (   R   t   outputt   streamt   value(    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   test_color_format%   s    	
c         C  sq   t  ƒ  } d „  | _ t | ƒ } t d | ƒ  t d d | ƒWd  QX| j ƒ  } | j ƒ  |  j d | ƒ d  S(   Nc           S  s   d S(   Niÿÿÿÿ(    (    (    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyR   5   s    u   redu   ERROR!R   u   [91mERROR!
[0m(   R   R    R   R   R   R!   R"   R   (   R   R#   R$   R%   (    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   test_color_context2   s    	
(	   t   __name__t
   __module__R   t   unittestt   skipIft   ost   nameR&   R'   (    (    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyR      s   	$u   __main__(   t
   __future__R    t   argparseR   R   t   ImportErrort   ioR,   t   sysR*   R   t   clyentR   t   clyent.colors.color_formatterR   t   clyent.colorsR   R   R   t   TestCaseR   R(   R   (    (    (    s7   lib/python2.7/site-packages/clyent/tests/test_clyent.pyt   <module>   s    	+