From f4fd3ab88f0b4816a77be47381f1a7eec6c284e6 Mon Sep 17 00:00:00 2001
From: Tom Talbot <tom.talbot@torchbox.com>
Date: Wed, 9 Jul 2014 15:37:22 +0100
Subject: [PATCH] Add document usage count tests

---
 wagtail/wagtailadmin/tests/tests.py | 26 +++++++++++++++++++++++---
 wagtail/wagtailadmin/utils.py       |  2 +-
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/wagtail/wagtailadmin/tests/tests.py b/wagtail/wagtailadmin/tests/tests.py
index 4d5491f1d..2b5d379e4 100644
--- a/wagtail/wagtailadmin/tests/tests.py
+++ b/wagtail/wagtailadmin/tests/tests.py
@@ -1,10 +1,13 @@
 from django.test import TestCase
-from wagtail.tests.utils import WagtailTestUtils
-from wagtail.wagtailcore.models import Page
-from wagtail.wagtailadmin.tasks import send_email_task
 from django.core.urlresolvers import reverse
 from django.core import mail
 
+from wagtail.tests.utils import WagtailTestUtils
+from wagtail.tests.models import EventPage, EventPageRelatedLink
+from wagtail.wagtailcore.models import Page
+from wagtail.wagtailadmin.tasks import send_email_task
+from wagtail.wagtaildocs.models import Document
+
 
 class TestHome(TestCase, WagtailTestUtils):
     def setUp(self):
@@ -43,3 +46,20 @@ class TestSendEmailTask(TestCase):
         self.assertEqual(mail.outbox[0].subject, "Test subject")
         self.assertEqual(mail.outbox[0].body, "Test content")
         self.assertEqual(mail.outbox[0].to, ["nobody@email.com"])
+
+
+class TestUsageCount(TestCase):
+    fixtures = ['wagtail/tests/fixtures/test.json']
+
+    def test_unused_document_usage_count(self):
+        doc = Document.objects.get(id=1)
+        self.assertEqual(doc.usage_count, 0)
+
+    def test_used_document_usage_count(self):
+        doc = Document.objects.get(id=1)
+        page = EventPage.objects.get(id=4)
+        event_page_related_link = EventPageRelatedLink()
+        event_page_related_link.page = page
+        event_page_related_link.link_document = doc
+        event_page_related_link.save()
+        self.assertEqual(doc.usage_count, 1)
diff --git a/wagtail/wagtailadmin/utils.py b/wagtail/wagtailadmin/utils.py
index a252d52aa..a2dd39b8f 100644
--- a/wagtail/wagtailadmin/utils.py
+++ b/wagtail/wagtailadmin/utils.py
@@ -2,7 +2,7 @@ from wagtail.wagtailcore.models import Page
 
 
 def usage_count(self):
-    """The number of times that an selfect has been used"""
+    """The number of times that an obect has been used"""
     count = 0
     relations = self._meta.get_all_related_objects(
         include_hidden=True,