U
d/
@ s d dl mZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ d dlmZ d ejifd
d ZG d d
d
eZdS ) )parse)modelform_factory)TestCase)
safestring)escape) countriesfieldswidgets)settings)Personcountryc K s t tfdg| d|S )Nr )r r )r r )r kwargs r G/tmp/pip-unpacked-wheel-dnjn05s9/django_countries/tests/test_widgets.py person_form
s r c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dS )TestCountrySelectWidgetc C s t ` d S Nr selfr r r setUp s z TestCountrySelectWidget.setUpc C s t ` d S r r r r r r tearDown s z TestCountrySelectWidget.tearDownc C s. t i d}| jd j}| t|tj d S )N)r r )r r widgetZ assertFalse
isinstancer CountrySelectWidget)r
PersonFormr r r r test_not_default_widget s
z/TestCountrySelectWidget.test_not_default_widgetc C sL | j ddid2 t }| }| ttt jt j| W 5 Q R X d S )NAUDesertZCOUNTRIES_ONLY) r
r as_passertInr urlparseurljoinZ
STATIC_URLZCOUNTRIES_FLAG_URLr r htmlr r r test_render_contains_flag_url s
z5TestCountrySelectWidget.test_render_contains_flag_urlc C sh | j ddidN t }| }| jd|dd | tdj| | tdj| W 5 Q R X d S )Nr r r z" )count__) r
r r Z assertInHTMLr! r Countryflag assertNotInr$ r r r test_render( s
z#TestCountrySelectWidget.test_renderc C s` | j ddidF t }|ddid }| tdj| | tdj| W 5 Q R X d S )Nr r r r )initialr) )r
r r r! r r* r+ r, r$ r r r test_render_initial0 s
z+TestCountrySelectWidget.test_render_initialc C s> t dd}| d| | d| | t|tj d S )Ntestz