ó
D()Qc           @   ss   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   Z	 d e
 f d     YZ d	 S(
   i˙˙˙˙(   t   datetime(   t   HttpRequest(   t   settings(   t   import_module(   t   Userc          C   s.   t    }  t t j  } | j d   |  _ |  S(   N(   R   R   R   t   SESSION_ENGINEt   SessionStoret   Nonet   session(   t   requestt   engine(    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   get_request
   s    	t   TestStoragec           B   s5   e  Z d    Z d   Z d   Z d   Z d   Z RS(   c         C   s"   t  j j d d  \ |  _ } d  S(   Nt   usernamet	   testuser1(   R   t   objectst   get_or_createt   testuser(   t   selft   created(    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   setUp   s    c         C   sŤ   t    } |  j   d | d   } d } |  j | j d   | | _ |  j | j |  | j   |  j | j d   | | _ |  j   d | d   } |  j | j d   d  S(   Nt   wizard1i   t   wizard2(   R   t   get_storageR   t   assertEqualt   current_stept   reset(   R   R	   t   storaget   my_stept   storage2(    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   test_current_step   s    		
	c         C   sü   t    } |  j   d | d   } d } i d d 6d d 6t j   d 6|  j d 6} |  j | j |  d   | j | |  |  j | j |  |  | j	   |  j | j |  d   | j | |  |  j   d	 | d   } |  j | j |  d   d  S(
   NR   t   startt   data1t   field1t   data2t   field2t   field3t   field4R   (
   R   R   R   R    t   nowR   R   t   get_step_datat   set_step_dataR   (   R   R	   R   t   step1t
   step_data1R   (    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   test_step_data&   s    	

c         C   sĐ   t    } |  j   d | d   } i d d 6d d 6t j   d 6|  j d 6} |  j | j i   | | _ |  j | j |  | j   |  j | j i   | | _ |  j   d | d   } |  j | j i   d  S(	   NR   R    t   key1R"   t   key2t   key3t   key4R   (	   R   R   R   R    R&   R   R   t
   extra_dataR   (   R   R	   R   t   extra_contextR   (    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   test_extra_context;   s    	
	
	c         C   s^   t    } |  j   d | d   } |  j d | j k  t | j d <|  j d | j k  d  S(   NR   t   test(   R   R   R   t   assertFalseR0   t   Truet
   assertTrue(   R   R	   R   (    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt"   test_extra_context_key_persistenceO   s
    	(   t   __name__t
   __module__R   R   R+   R2   R7   (    (    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyR      s
   				N(   R    t   django.httpR   t   django.confR   t   django.utils.importlibR   t   django.contrib.auth.modelsR   R   t   objectR   (    (    (    sF   ../Django//lib/python/django/contrib/formtools/tests/wizard/storage.pyt   <module>   s   	