ó
ø¢TQc           @  sÑ  d  d l  m Z d  d l m Z m Z 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  d l m Z d  d	 l m Z m Z m Z d
 e j f d „  ƒ  YZ d „  Z d e j f d „  ƒ  YZ d e j f d „  ƒ  YZ d e j f d „  ƒ  YZ d e f d „  ƒ  YZ d e j f d „  ƒ  YZ d e j f d „  ƒ  YZ  e j j! e d e  d d ƒZ" d e f d „  ƒ  YZ# d e
 f d „  ƒ  YZ$ d  e
 f d! „  ƒ  YZ% d" e
 f d# „  ƒ  YZ& d$ S(%   iÿÿÿÿ(   t   unicode_literals(   t   formst   http(   t   settings(   t   models(   t   TestCase(   t   TemplateResponse(   t   import_module(   t   User(   t
   WizardViewt   SessionWizardViewt   CookieWizardViewt   DummyRequestc           B  s   e  Z d d  „ Z RS(   c         C  s]   t  t |  ƒ j ƒ  | r d p" d |  _ | d  k	 rG |  j j | ƒ n  i  |  _ t |  _	 d  S(   Nu   POSTu   GET(
   t   superR   t   __init__t   methodt   Nonet   POSTt   updatet   sessiont   Truet   _dont_enforce_csrf_checks(   t   selfR   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR      s    	N(   t   __name__t
   __module__R   R   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR      s   c          O  s4   t  |  | Ž  } t t j ƒ } | j d  ƒ | _ | S(   N(   R   R   R   t   SESSION_ENGINEt   SessionStoreR   R   (   t   argst   kwargst   requestt   engine(    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   get_request   s    t   Step1c           B  s   e  Z e j ƒ  Z RS(    (   R   R   R   t	   CharFieldt   name(    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR    "   s   t   Step2c           B  s   e  Z e j ƒ  Z RS(    (   R   R   R   R!   R"   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR#   &   s   t   Step3c           B  s   e  Z e j ƒ  Z RS(    (   R   R   R   R!   t   data(    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR$   *   s   t   CustomKwargsStep1c           B  s   e  Z d d  „ Z RS(   c         O  s"   | |  _  t t |  ƒ j | | Ž  S(   N(   t   testR   R&   R   (   R   R'   R   R   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR   0   s    	N(   R   R   R   R   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR&   .   s   t	   TestModelc           B  s-   e  Z e j d  d ƒ Z d d d „  ƒ  YZ RS(   t
   max_lengthid   t   Metac           B  s   e  Z d  Z RS(   u	   formtools(   R   R   t	   app_label(    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR*   8   s   (    (   R   R   R   R!   R"   R*   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR(   5   s   t   TestModelFormc           B  s   e  Z d  d d „  ƒ  YZ RS(   R*   c           B  s   e  Z e Z RS(    (   R   R   R(   t   model(    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR*   =   s   (    (   R   R   R*   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR,   <   s   t   formt   extrai   t
   TestWizardc           B  s    e  Z d  Z d „  Z d „  Z RS(   u>   django.contrib.formtools.wizard.storage.session.SessionStoragec         O  s(   t  t |  ƒ j | | | Ž } | |  f S(   N(   R   R0   t   dispatch(   R   R   R   R   t   response(    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR1   G   s    c         O  s;   t  t |  ƒ j | | | Ž } | d k r7 t | d <n  | S(   Nu   kwargs_testu   test(   R   R0   t   get_form_kwargsR   (   R   t   stepR   R   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR3   K   s    (   R   R   t   storage_nameR1   R3   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR0   D   s   	t	   FormTestsc           B  sk   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d „  Z
 d	 „  Z d
 „  Z RS(   c         C  sÅ   t  j t t g ƒ } |  j | d i t d 6t d 6ƒ t  j d t f d t f g ƒ } |  j | d i t d 6t d 6ƒ t  j t t d t f g ƒ } |  j | d i t d 6t d 6t d 6ƒ d  S(   Nu	   form_listu   0u   1u   startu   step2u   finish(   R0   t   get_initkwargsR    R#   t   assertEqualR$   (   R   t   testform(    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_initR   s    "!c         C  s“   t  ƒ  } t j t t g ƒ } | | ƒ \ } } |  j | j j d ƒ t j d t f d t f g ƒ } | | ƒ \ } } |  j | j j d ƒ d  S(   Nu   0u   startu   step2(   R   R0   t   as_viewR    R#   R8   t   stepst   current(   R   R   R9   R2   t   instance(    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_first_step^   s    	!c         C  sÌ   t  j d t f d t f g ƒ } t i d d 6d d 6ƒ } | | ƒ \ } } |  j | j j d ƒ d | j _	 t  j d t f d t f g ƒ } i d d 6| _
 | | ƒ \ } } |  j | j j d ƒ d  S(   Nu   startu   step2u   test_wizard-current_stepu   data1u   name(   R0   R;   R    R#   R   R8   R<   R=   t   storaget   current_stepR   (   R   R9   R   R2   R>   t	   testform2(    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_persistencej   s    !!c         C  sË   t  ƒ  } t j d t f d t f d t f g d i t d 6ƒ} | | ƒ \ } } |  j | j ƒ  d ƒ t j d t f d t f d t f g d i t	 d 6ƒ} | | ƒ \ } } |  j | j ƒ  d ƒ d  S(   Nu   startu   step2u   step3t   condition_dict(
   R   R0   R;   R    R#   R$   R   R8   t   get_next_stept   False(   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_conditionx   s    	!!c         C  s•   t  ƒ  } t j d t f d t f g ƒ } | | ƒ \ } } |  j | j d ƒ i  ƒ |  j | j d ƒ i t d 6ƒ |  j | j d ƒ j	 t ƒ d  S(   Nu   startu   kwargs_testu   test(
   R   R0   R;   R    R&   R8   R3   R   t   get_formR'   (   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_kwargs‡   s    	 c         C  so   t  ƒ  } t j d t f d t f g ƒ } | | ƒ \ } } |  j | j ƒ  d ƒ |  j | j d ƒ d ƒ d  S(   Nu   startu   step2u   another(   R   R0   R;   R    R#   R8   t   get_form_prefix(   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_prefix’   s
    	!c         C  s   t  ƒ  } t j d t f d t f g d i i d d 6d 6ƒ} | | ƒ \ } } |  j | j d ƒ i d d 6ƒ |  j | j d ƒ i  ƒ d  S(   Nu   startu   step2t   initial_dictu   value1u   name(   R   R0   R;   R    R#   R8   t   get_form_initial(   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_initial›   s    	 c         C  sˆ   t  ƒ  } t ƒ  } t j d t f d t f g d i | d 6ƒ} | | ƒ \ } } |  j | j d ƒ | ƒ |  j | j d ƒ d  ƒ d  S(   Nu   startu   step2t   instance_dictu   non_exist_instance(	   R   R(   R0   R;   R,   R#   R8   t   get_form_instanceR   (   R   R   t   the_instanceR9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_form_instance¥   s    		c         C  sé   t  ƒ  } t j j d d ƒ \ } } t j j d d ƒ \ } } t j d t f d t f g d i t j j d d ƒ d 6ƒ} | | ƒ \ } } |  j	 t
 | j d ƒ ƒ | g ƒ |  j	 | j d ƒ d  ƒ |  j	 | j ƒ  j ƒ  d ƒ d  S(	   NR"   u   test object 1u   test object 2u   startu   step2RO   u   non_exist_instancei   (   R   R(   t   objectst   get_or_createR0   R;   t   TestModelFormSetR#   t   filterR8   t   listRP   R   RH   t   initial_form_count(   R   R   t   the_instance1t   createdt   the_instance2R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_formset_instance³   s    	"c         C  sV   t  ƒ  } t j d t f d t f g ƒ } | | ƒ \ } } |  j t | j d  ƒ d  S(   Nu   startu   step2(	   R   R0   R;   R    R#   t   assertRaisest   NotImplementedErrort   doneR   (   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt	   test_doneÂ   s    	!c         C  sc   t  ƒ  } t j d t f d t f g ƒ } | | ƒ \ } } | j d  ƒ |  j | j j	 d ƒ d  S(   Nu   startu   step2(
   R   R0   R;   R    R#   t   render_doneR   R8   R@   RA   (   R   R   R9   R2   R>   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   test_revalidationÊ   s
    	!(   R   R   R:   R?   RC   RG   RI   RK   RN   RR   R\   R`   Rb   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyR6   Q   s   								
			t   SessionFormTestsc           B  s   e  Z d  „  Z RS(   c         C  sA   t  ƒ  } t j d t f g ƒ } |  j t | | ƒ t ƒ ƒ d  S(   Nu   start(   R   R
   R;   R    t
   assertTruet
   isinstanceR   (   R   R   R9   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt	   test_initÔ   s    	(   R   R   Rf   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyRc   Ó   s   t   CookieFormTestsc           B  s   e  Z d  „  Z RS(   c         C  sA   t  ƒ  } t j d t f g ƒ } |  j t | | ƒ t ƒ ƒ d  S(   Nu   start(   R   R   R;   R    Rd   Re   R   (   R   R   R9   (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyRf   Û   s    	(   R   R   Rf   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyRg   Ú   s   N('   t
   __future__R    t   djangoR   R   t   django.confR   t	   django.dbR   t   django.testR   t   django.template.responseR   t   django.utils.importlibR   t   django.contrib.auth.modelsR   t%   django.contrib.formtools.wizard.viewsR	   R
   R   t   HttpRequestR   R   t   FormR    R#   R$   R&   t   ModelR(   t	   ModelFormR,   t   modelformset_factoryRU   R0   R6   Rc   Rg   (    (    (    sD   ../Django//lib/python/django/contrib/formtools/tests/wizard/forms.pyt   <module>   s*   
	‚