Added ForeignKey test case & example, coverage == 100%

Author: and3rson

example/db.sqlite3

@@ -1,15 +1,16 @@
 from django.contrib import admin
 from django import forms
 from searchableselect.widgets import SearchableSelect
-from .models import Cat, Food
+from .models import Cat, Food, Person
 
 
 class CatAdminForm(forms.ModelForm):
     class Meta:
         model = Cat
         exclude = ()
         widgets = {
-            'favorite_foods': SearchableSelect(model='example.Food', search_field='name', many=True)
+            'favorite_foods': SearchableSelect(model='example.Food', search_field='name', many=True),
+            'owner': SearchableSelect(model='example.Person', search_field='name', many=False),
         }
 
 
@@ -21,5 +22,10 @@ class FoodAdmin(admin.ModelAdmin):
     pass
 
 
+class PersonAdmin(admin.ModelAdmin):
+    pass
+
+
 admin.site.register(Cat, CatAdmin)
 admin.site.register(Food, FoodAdmin)
+admin.site.register(Person, PersonAdmin)

example/example/admin.py

@@ -1,8 +1,9 @@
 # -*- coding: utf-8 -*-
-# Generated by Django 1.10.2 on 2016-10-28 14:52
+# Generated by Django 1.10.2 on 2016-10-30 16:15
 from __future__ import unicode_literals
 
 from django.db import migrations, models
+import django.db.models.deletion
 
 
 class Migration(migrations.Migration):
@@ -27,9 +28,21 @@ class Migration(migrations.Migration):
                 ('name', models.CharField(max_length=64)),
             ],
         ),
+        migrations.CreateModel(
+            name='Person',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=64)),
+            ],
+        ),
         migrations.AddField(
             model_name='cat',
             name='favorite_foods',
             field=models.ManyToManyField(related_name='loved_by', to='example.Food'),
         ),
+        migrations.AddField(
+            model_name='cat',
+            name='owner',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='example.Person'),
+        ),
     ]

example/example/migrations/0001_initial.py

@@ -4,6 +4,7 @@
 class Cat(models.Model):
     name = models.CharField(max_length=64, null=False, blank=False)
     favorite_foods = models.ManyToManyField('Food', related_name='loved_by')
+    owner = models.ForeignKey('Person', null=False, blank=False)
 
     def __unicode__(self):
         return self.name
@@ -14,3 +15,10 @@ class Food(models.Model):
 
     def __unicode__(self):
         return self.name
+
+
+class Person(models.Model):
+    name = models.CharField(max_length=64, null=False, blank=False)
+
+    def __unicode__(self):
+        return self.name

example/example/models.py

@@ -1,7 +1,7 @@
 from django import test
 from django.test.utils import override_settings
 from django.contrib.auth.models import User
-from example.models import Cat, Food
+from example.models import Cat, Food, Person
 from selenium import webdriver
 import selenium.webdriver.support.ui as ui
 import time
@@ -32,6 +32,15 @@ def setUp(self):
             Food(name='Whiskas')
         ])
 
+        Person.objects.bulk_create([
+            Person(name='Andrew'),
+            Person(name='David'),
+            Person(name='Mary'),
+            Person(name='Matthew'),
+            Person(name='Mike'),
+            Person(name='Victoria')
+        ])
+
     @override_settings(DEBUG=True)
     def test_foo(self):
         self.selenium.get(self.live_server_url + '/admin/')
@@ -55,18 +64,38 @@ def select_food(text):
             field = self.selenium.find_element_by_id('id_favorite_foods')
             field.send_keys(text)
 
-            wait.until(lambda driver: driver.find_elements_by_class_name('tt-suggestion'))
+            wait.until(lambda driver: driver.find_elements_by_css_selector('.field-favorite_foods .tt-suggestion'))
 
-            suggestions = self.selenium.find_elements_by_class_name('tt-suggestion')
+            suggestions = self.selenium.find_elements_by_css_selector('.field-favorite_foods .tt-suggestion')
             suggestions[0].click()
 
-            wait.until(lambda driver: driver.find_elements_by_class_name('chip'))
+            wait.until(lambda driver: driver.find_elements_by_css_selector('.field-favorite_foods .chip'))
+
             # Wait for suggestion animation to finish
             time.sleep(0.5)
 
         select_food('M')
         select_food('W')
 
+        # Select person
+
+        owner_field = self.selenium.find_element_by_id('id_owner')
+        owner_field.send_keys('An')
+
+        wait.until(lambda driver: driver.find_elements_by_css_selector('.field-owner .tt-suggestion'))
+
+        owner = self.selenium.find_elements_by_css_selector('.field-owner .tt-suggestion')
+        owner[0].click()
+
+        wait.until(lambda driver: driver.find_elements_by_css_selector('.field-owner .chip'))
+
+        # Wait for suggestion animation to finish
+        time.sleep(0.5)
+
+        wait.until(lambda driver: driver.find_elements_by_class_name('tt-suggestion'))
+
+        # Save cat record
+
         cat_name_input.submit()
 
         wait.until(lambda driver: driver.find_elements_by_class_name('success'))
@@ -77,3 +106,4 @@ def select_food(text):
         self.assertEqual(favorite_foods.count(), 2)
         self.assertIn(favorite_foods[0].name, ('Milk', 'Meat'))
         self.assertEqual(favorite_foods[1].name, 'Whiskas')
+        self.assertEqual(marusia.owner.name, 'Andrew')