ó
ÉrRc           @   sj   d  d l  Z  d  d l Z d  d l m Z d  d l Z d e j f d „  ƒ  YZ e d k rf e j ƒ  n  d S(   iÿÿÿÿN(   t   OSErrort   TestMigratec           B   s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C   sž   t  j t  j j t  j j t ƒ ƒ d ƒ y t  j d ƒ Wn t k
 rM n Xy t j	 d ƒ Wn t k
 rr n Xt  j
 d ƒ t  j
 d ƒ t  j
 d ƒ d  S(   Ni    s   app.dbt
   migrationss   python app.py db inits   python app.py db migrates   python app.py db upgrade(   t   ost   chdirt   patht   splitt   abspatht   __file__t   removeR    t   shutilt   rmtreet   system(   t   self(    (    sD   /home/miguel/Documents/dev/flask/Flask-Migrate/tests/test_migrate.pyt   setUp   s    )c         C   sN   y t  j d ƒ Wn t k
 r$ n Xy t j d ƒ Wn t k
 rI n Xd  S(   Ns   app.dbR   (   R   R	   R    R
   R   (   R   (    (    sD   /home/miguel/Documents/dev/flask/Flask-Migrate/tests/test_migrate.pyt   tearDown   s    c         C   s@   d d l  m } m } | j j | d d ƒ ƒ | j j ƒ  d  S(   Niÿÿÿÿ(   t   dbt   Usert   namet   test(   t   appR   R   t   sessiont   addt   commit(   R   R   R   (    (    sD   /home/miguel/Documents/dev/flask/Flask-Migrate/tests/test_migrate.pyt   test_migrate_upgrade    s    (   t   __name__t
   __module__R   R   R   (    (    (    sD   /home/miguel/Documents/dev/flask/Flask-Migrate/tests/test_migrate.pyR      s   		
t   __main__(	   R   R
   t
   exceptionsR    t   unittestt   TestCaseR   R   t   main(    (    (    sD   /home/miguel/Documents/dev/flask/Flask-Migrate/tests/test_migrate.pyt   <module>   s   