Ticket #9071: add-r9017-tests.diff

tests/regressiontests/admin_widgets/models.py

@@ -39 +39 @@
 >>> from django.contrib.admin.widgets import FilteredSelectMultiple, AdminSplitDateTime
 >>> from django.contrib.admin.widgets import AdminFileWidget, ForeignKeyRawIdWidget, ManyToManyRawIdWidget
 >>> from django.contrib.admin.widgets import RelatedFieldWidgetWrapper
+>>> from django.contrib.admin import site
+>>> from django.forms.widgets import Select as SelectWidget
 
 Calling conditional_escape on the output of widget.render will simulate what
 happens in the template. This is easier than setting up a template and context
@@ -101 +103 @@
 >>> w = ForeignKeyRawIdWidget(rel)
 >>> print w.render('test', core.parent_id, attrs={})
 <input type="text" name="test" value="86" class="vForeignKeyRawIdAdminField" /><a href="../../../admin_widgets/inventory/?t=barcode" class="related-lookup" id="lookup_id_test" onclick="return showRelatedObjectLookupPopup(this);"> <img src="%(ADMIN_MEDIA_PREFIX)simg/admin/selector-search.gif" width="16" height="16" alt="Lookup" /></a>&nbsp;<strong>Apple</strong>
+
+
+# Check that RelatedFieldWidgetWrapper doesn'tshow the add image when explicitely
+# requested
+>>> site.register(Band)
+>>> rel = Album._meta.get_field('band').rel
+>>> w = RelatedFieldWidgetWrapper(SelectWidget(), rel, site, True)
+>>> print w.render('test', 'test')
+<select name="test">
+</select><a href="../../../admin_widgets/band/add/" class="add-another" id="add_id_test" onclick="return showAddAnotherPopup(this);"> <img src="/media/img/admin/icon_addlink.gif" width="10" height="10" alt="Add Another"/></a>
+
+>>> w = RelatedFieldWidgetWrapper(SelectWidget(), rel, site, False)
+>>> print w.render('test', 'test')
+<select name="test">
+</select>
+
 """ % {
     'ADMIN_MEDIA_PREFIX': settings.ADMIN_MEDIA_PREFIX,
     'STORAGE_URL': default_storage.url(''),
 }}
+