ó
ç'ÊUc           @   sj   d  d l  m Z d  d l m Z d  d l m Z d  d l m Z d  d l m	 Z	 d e f d „  ƒ  YZ
 d S(	   iÿÿÿÿ(   t   TestCase(   t   Spider(   t   Request(   t   UserAgentMiddleware(   t   get_crawlert   UserAgentMiddlewareTestc           B   s>   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C   s8   t  t i | d 6ƒ } | j d ƒ } | t j | ƒ f S(   Nt
   USER_AGENTt   foo(   R   R   t   _create_spiderR   t   from_crawler(   t   selft   default_useragentt   crawlert   spider(    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   get_spider_and_mw   s    c         C   sZ   |  j  d ƒ \ } } t d ƒ } | j | | ƒ d  k s? t ‚ |  j | j d d ƒ d  S(   NR   s   http://scrapytest.org/s
   User-Agent(   R   R   t   process_requestt   Nonet   AssertionErrort   assertEqualst   headers(   R
   R   t   mwt   req(    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   test_default_agent   s    c         C   sw   |  j  d ƒ \ } } d  | _ | j | ƒ t d ƒ } | j | | ƒ d  k sU t ‚ | j j d ƒ d  k ss t ‚ d  S(   NR   s   http://scrapytest.org/s
   User-Agent(	   R   R   t
   user_agentt   spider_openedR   R   R   R   t   get(   R
   R   R   R   (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   test_remove_agent   s    	c         C   sp   |  j  d ƒ \ } } d | _ | j | ƒ t d ƒ } | j | | ƒ d  k sU t ‚ |  j | j d d ƒ d  S(   NR   t   spider_useragents   http://scrapytest.org/s
   User-Agent(	   R   R   R   R   R   R   R   R   R   (   R
   R   R   R   (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   test_spider_agent   s    	c         C   s}   |  j  d ƒ \ } } d | _ | j | ƒ t d d i d d 6ƒ} | j | | ƒ d  k sb t ‚ |  j | j d d ƒ d  S(   NR   R   s   http://scrapytest.org/R   t   header_useragents
   User-Agent(	   R   R   R   R   R   R   R   R   R   (   R
   R   R   R   (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   test_header_agent'   s    	c         C   sn   |  j  d  ƒ \ } } d  | _ | j | ƒ t d ƒ } | j | | ƒ d  k sU t ‚ d | j k sj t ‚ d  S(   Ns   http://scrapytest.org/s
   User-Agent(   R   R   R   R   R   R   R   R   (   R
   R   R   R   (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   test_no_agent/   s    	(   t   __name__t
   __module__R   R   R   R   R   R   (    (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyR   	   s   						N(   t   unittestR    t   scrapy.spidersR   t   scrapy.httpR   t&   scrapy.downloadermiddlewares.useragentR   t   scrapy.utils.testR   R   (    (    (    sM   /home/travis/build/scrapy/scrapy/tests/test_downloadermiddleware_useragent.pyt   <module>   s
   