~ubuntu-branches/debian/squeeze/python-django/squeeze

 

from django.conf import settings

from django.db import models

from django.core.files.storage import default_storage

 

class Member(models.Model):

    name = models.CharField(max_length=100)

 

    def __unicode__(self):

        return self.name

 

class Band(models.Model):

    name = models.CharField(max_length=100)

    members = models.ManyToManyField(Member)

 

    def __unicode__(self):

        return self.name

 

class Album(models.Model):

    band = models.ForeignKey(Band)

    name = models.CharField(max_length=100)

    cover_art = models.FileField(upload_to='albums')

 

    def __unicode__(self):

        return self.name

 

class Inventory(models.Model):

   barcode = models.PositiveIntegerField(unique=True)

   parent = models.ForeignKey('self', to_field='barcode', blank=True, null=True)

   name = models.CharField(blank=False, max_length=20)

 

   def __unicode__(self):

      return self.name

 

__test__ = {'WIDGETS_TESTS': """

>>> from datetime import datetime

>>> from django.utils.html import escape, conditional_escape

>>> from django.core.files.uploadedfile import SimpleUploadedFile

>>> from django.contrib.admin.widgets import FilteredSelectMultiple, AdminSplitDateTime

>>> from django.contrib.admin.widgets import AdminFileWidget, ForeignKeyRawIdWidget, ManyToManyRawIdWidget

>>> from django.contrib.admin.widgets import RelatedFieldWidgetWrapper

 

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

for each test.

 

Make sure that the Admin widgets render properly, that is, without their extra

HTML escaped.

 

>>> w = FilteredSelectMultiple('test', False)

>>> print conditional_escape(w.render('test', 'test'))

<select multiple="multiple" name="test">

</select><script type="text/javascript">addEvent(window, "load", function(e) {SelectFilter.init("id_test", "test", 0, "%(ADMIN_MEDIA_PREFIX)s"); });</script>

<BLANKLINE>

 

>>> w = AdminSplitDateTime()

>>> print conditional_escape(w.render('test', datetime(2007, 12, 1, 9, 30)))

<p class="datetime">Date: <input value="2007-12-01" type="text" class="vDateField" name="test_0" size="10" /><br />Time: <input value="09:30:00" type="text" class="vTimeField" name="test_1" size="8" /></p>

 

>>> band = Band.objects.create(pk=1, name='Linkin Park')

>>> album = band.album_set.create(name='Hybrid Theory', cover_art=r'albums\hybrid_theory.jpg')

 

>>> w = AdminFileWidget()

>>> print conditional_escape(w.render('test', album.cover_art))

Currently: <a target="_blank" href="%(STORAGE_URL)salbums/hybrid_theory.jpg">albums\hybrid_theory.jpg</a> <br />Change: <input type="file" name="test" />

>>> print conditional_escape(w.render('test', SimpleUploadedFile('test', 'content')))

<input type="file" name="test" />

 

>>> rel = Album._meta.get_field('band').rel

>>> w = ForeignKeyRawIdWidget(rel)

>>> print conditional_escape(w.render('test', band.pk, attrs={}))

<input type="text" name="test" value="1" class="vForeignKeyRawIdAdminField" /><a href="../../../admin_widgets/band/?t=id" 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>Linkin Park</strong>

 

>>> m1 = Member.objects.create(pk=1, name='Chester')

>>> m2 = Member.objects.create(pk=2, name='Mike')

>>> band.members.add(m1, m2)

 

>>> rel = Band._meta.get_field('members').rel

>>> w = ManyToManyRawIdWidget(rel)

>>> print conditional_escape(w.render('test', [m1.pk, m2.pk], attrs={}))

<input type="text" name="test" value="1,2" class="vManyToManyRawIdAdminField" /><a href="../../../admin_widgets/member/" 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>

>>> w._has_changed(None, None)

False

>>> w._has_changed([], None)

False

>>> w._has_changed(None, [u'1'])

True

>>> w._has_changed([1, 2], [u'1', u'2'])

False

>>> w._has_changed([1, 2], [u'1'])

True

>>> w._has_changed([1, 2], [u'1', u'3'])

True

 

# Check that ForeignKeyRawIdWidget works with fields which aren't related to

# the model's primary key.

>>> apple = Inventory.objects.create(barcode=86, name='Apple')

>>> pear = Inventory.objects.create(barcode=22, name='Pear')

>>> core = Inventory.objects.create(barcode=87, name='Core', parent=apple)

>>> rel = Inventory._meta.get_field('parent').rel

>>> 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>

""" % {

    'ADMIN_MEDIA_PREFIX': settings.ADMIN_MEDIA_PREFIX,

    'STORAGE_URL': default_storage.url(''),

}}