Add test for admin case

jazzband · Apr 22, 2015 · a013f7f · a013f7f
1 parent a8dae12
commit a013f7f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/simple_history/tests/admin.py b/simple_history/tests/admin.py
@@ -3,7 +3,7 @@
 from django.contrib import admin
 
 from simple_history.admin import SimpleHistoryAdmin
-from .models import Poll, Choice, Person, Book, Document, Paper
+from .models import Poll, Choice, Person, Book, Document, Paper, Employee
 
 
 class PersonAdmin(SimpleHistoryAdmin):
@@ -17,3 +17,4 @@ def has_change_permission(self, request, obj=None):
 admin.site.register(Book, SimpleHistoryAdmin)
 admin.site.register(Document, SimpleHistoryAdmin)
 admin.site.register(Paper, SimpleHistoryAdmin)
+admin.site.register(Employee, SimpleHistoryAdmin)
diff --git a/simple_history/tests/tests/test_admin.py b/simple_history/tests/tests/test_admin.py
@@ -12,7 +12,7 @@
 from django.conf import settings
 from simple_history.models import HistoricalRecords
 
-from ..models import Book, Person, Poll, State
+from ..models import Book, Person, Poll, State, Employee
 
 
 today = datetime(2021, 1, 1, 10, 0)
@@ -234,3 +234,14 @@ def test_deleteting_user(self):
 
         historical_poll = poll.history.all()[0]
         self.assertEqual(historical_poll.history_user, None)
+
+    def test_missing_one_to_one(self):
+        """A relation to a missing one-to-one model should still show history"""
+        self.login()
+        manager = Employee.objects.create()
+        employee = Employee.objects.create(manager=manager)
+        employee.manager = None
+        employee.save()
+        manager.delete()
+        response = self.app.get(get_history_url(employee, 0))
+        self.assertEqual(response.status_code, 200)