Fix subscriber's get_object_display

Closes-Bug: #1287751 Change-Id: I84b36b98e51a19645e1bc47af2a205aec4cbbd2c
kickstandproject · Mar 6, 2014 · 0d6403a · 0d6403a
1 parent ca8eb96
commit 0d6403a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/wildcard/dashboards/project/subscribers/tables.py b/wildcard/dashboards/project/subscribers/tables.py
@@ -104,3 +104,6 @@ def _domains(self):
 
     def get_object_id(self, datum):
         return datum.uuid
+
+    def get_object_display(self, datum):
+        return datum.username
diff --git a/wildcard/dashboards/project/subscribers/tests.py b/wildcard/dashboards/project/subscribers/tests.py
@@ -20,6 +20,7 @@
 
 from wildcard import api
 from wildcard.dashboards.project.subscribers import forms
+from wildcard.dashboards.project.subscribers import tables
 from wildcard.test import helpers as test
 
 
@@ -176,6 +177,12 @@ def test_disable_subscriber(self):
 
         self.assertRedirectsNoFollow(res, INDEX_URL)
 
+    def test_object_display(self):
+        subscriber = self.subscribers.get(uuid="1")
+        table = tables.SubscribersTable(self.request, [subscriber])
+        object_display = table.get_object_display(subscriber)
+        self.assertEquals(object_display, subscriber.username)
+
 
 class TextInputWithGenerator(test.TestCase):