ó
G()Qc           @   s‚   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	 m
 Z
 d „  Z d „  Z d e f d	 „  ƒ  YZ d
 S(   iÿÿÿÿ(   t	   constants(   t   BaseTest(   t   Message(   t   SessionStorage(   t   SafeDatat	   mark_safec         C   s/   | |  j  j |  j <t |  d ƒ r+ |  ` n  d S(   sn   
    Sets the messages into the backend request's session and remove the
    backend's loaded data cache.
    t   _loaded_dataN(   t   requestt   sessiont   session_keyt   hasattrR   (   t   storaget   messages(    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   set_session_data   s    c         C   s%   |  j  j j |  j g  ƒ } t | ƒ S(   N(   R   R   t   getR	   t   len(   R   t   data(    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   stored_session_messages_count   s    t   SessionTestc           B   s2   e  Z e Z d  „  Z d „  Z d „  Z d „  Z RS(   c         C   s.   i  |  _  t t |  ƒ j ƒ  } |  j  | _  | S(   N(   R   t   superR   t   get_request(   t   selfR   (    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyR      s    	c         C   s
   t  | ƒ S(   N(   R   (   R   R   t   response(    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   stored_messages_count    s    c         C   sH   |  j  |  j ƒ  ƒ } d d g } t | | ƒ |  j t | ƒ | ƒ d  S(   Nt   testt   me(   t   storage_classR   R   t   assertEqualt   list(   R   R   t   example_messages(    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   test_get#   s    c         C   sU   |  j  ƒ  } t t j t d ƒ ƒ } t | | g ƒ |  j t | ƒ d j t	 ƒ d S(   s†   
        Tests that a message containing SafeData is keeping its safe status when
        retrieved from the message storage.
        s   <b>Hello Django!</b>i    N(
   t   get_storageR   R    t   DEBUGR   R   t   assertIsInstanceR   t   messageR   (   R   R   R"   (    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   test_safedata+   s    (   t   __name__t
   __module__R   R   R   R   R   R#   (    (    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyR      s
   			N(   t   django.contrib.messagesR    t"   django.contrib.messages.tests.baseR   t$   django.contrib.messages.storage.baseR   t'   django.contrib.messages.storage.sessionR   t   django.utils.safestringR   R   R   R   R   (    (    (    s>   ../Django//lib/python/django/contrib/messages/tests/session.pyt   <module>   s   	
	