diff options
Diffstat (limited to 'tests/admin_widgets/tests.py')
| -rw-r--r-- | tests/admin_widgets/tests.py | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/tests/admin_widgets/tests.py b/tests/admin_widgets/tests.py index 133be0a5d2..677ee18cff 100644 --- a/tests/admin_widgets/tests.py +++ b/tests/admin_widgets/tests.py @@ -26,7 +26,7 @@ from django.utils import translation from .models import ( Advisor, Album, Band, Bee, Car, Company, Event, Honeycomb, Individual, - Inventory, Member, MyFileField, Profile, School, Student, + Inventory, Member, MyFileField, Profile, ReleaseEvent, School, Student, UnsafeLimitChoicesTo, VideoStream, ) from .widgetadmin import site as widget_admin_site @@ -538,19 +538,27 @@ class ForeignKeyRawIdWidgetTest(TestCase): band.album_set.create( name='Hybrid Theory', cover_art=r'albums\hybrid_theory.jpg' ) - rel = Album._meta.get_field('band').remote_field - - w = widgets.ForeignKeyRawIdWidget(rel, widget_admin_site) + rel_uuid = Album._meta.get_field('band').remote_field + w = widgets.ForeignKeyRawIdWidget(rel_uuid, widget_admin_site) self.assertHTMLEqual( w.render('test', band.uuid, attrs={}), '<input type="text" name="test" value="%(banduuid)s" ' - 'class="vForeignKeyRawIdAdminField">' + 'class="vForeignKeyRawIdAdminField vUUIDField">' '<a href="/admin_widgets/band/?_to_field=uuid" class="related-lookup" ' 'id="lookup_id_test" title="Lookup"></a> <strong>' '<a href="/admin_widgets/band/%(bandpk)s/change/">Linkin Park</a>' '</strong>' % {'banduuid': band.uuid, 'bandpk': band.pk} ) + rel_id = ReleaseEvent._meta.get_field('album').remote_field + w = widgets.ForeignKeyRawIdWidget(rel_id, widget_admin_site) + self.assertHTMLEqual( + w.render('test', None, attrs={}), + '<input type="text" name="test" class="vForeignKeyRawIdAdminField">' + '<a href="/admin_widgets/album/?_to_field=id" class="related-lookup" ' + 'id="lookup_id_test" title="Lookup"></a>', + ) + def test_relations_to_non_primary_key(self): # ForeignKeyRawIdWidget works with fields which aren't related to # the model's primary key. |