dashboard: doctest for autocomplete.highlight

circle/dashboard/autocomplete_light_registry.py

@@ -7,15 +7,12 @@ from .views import AclUpdateView
 from .models import Profile
 
 
-class AclUserGroupAutocomplete(autocomplete_light.AutocompleteGenericBase):
-    search_fields = (
-        ('first_name', 'last_name', 'username', 'email', 'profile__org_id'),
-        ('name', 'groupprofile__org_id'),
-    )
-    choice_html_format = (u'<span data-value="%s"><span style="display:none"'
-                          u'>%s</span>%s</span>')
+def highlight(field, q, none_wo_match=True):
+    """
+    >>> highlight('<b>Akkount Krokodil', 'kro', False)
+    u'&lt;b&gt;Akkount <span class="autocomplete-hl">Kro</span>kodil'
+    """
 
-    def highlight(self, field, q, none_wo_match=True):
     if not field:
         return None
     try:
@@ -33,20 +30,29 @@ class AclUserGroupAutocomplete(autocomplete_light.AutocompleteGenericBase):
     else:
         return escape(field)
 
+
+class AclUserGroupAutocomplete(autocomplete_light.AutocompleteGenericBase):
+    search_fields = (
+        ('first_name', 'last_name', 'username', 'email', 'profile__org_id'),
+        ('name', 'groupprofile__org_id'),
+    )
+    choice_html_format = (u'<span data-value="%s"><span style="display:none"'
+                          u'>%s</span>%s</span>')
+
     def choice_displayed_text(self, choice):
         q = unicode(self.request.GET.get('q', ''))
-        name = self.highlight(unicode(choice), q, False)
+        name = highlight(unicode(choice), q, False)
         if isinstance(choice, User):
-            extra_fields = [self.highlight(choice.get_full_name(), q, False),
-                            self.highlight(choice.email, q)]
+            extra_fields = [highlight(choice.get_full_name(), q, False),
+                            highlight(choice.email, q)]
             try:
-                extra_fields.append(self.highlight(choice.profile.org_id, q))
+                extra_fields.append(highlight(choice.profile.org_id, q))
             except Profile.DoesNotExist:
                 pass
             return '%s (%s)' % (name, ', '.join(f for f in extra_fields
                                                 if f))
         else:
-            return '%s (%s)' % (name, _('group'))
+            return _('%s (group)') % name
 
     def choice_html(self, choice):
         return self.choice_html_format % (