Added moderators to Forum

forums/migrations/0003_forum_moderators.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+from django.conf import settings
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('forums', '0002_forum_active'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='forum',
+            name='moderators',
+            field=models.ManyToManyField(to=settings.AUTH_USER_MODEL),
+        ),
+    ]

forums/models.py

@@ -17,6 +17,9 @@ class Forum(models.Model):
     objects = models.Manager()
     visible = VisibleManager()
 
+    # Foreign Keys
+    moderators = models.ManyToManyField('users.User')
+
     def __str__(self):
         return self.name
 
@@ -29,3 +32,6 @@ def save(self, *args, **kwargs):
 
     def threads(self):
         return self.thread_set.all()
+
+    def moderator_list(self):
+        return self.moderators.all()

forums/tests/test_models.py

@@ -1,3 +1,4 @@
+from django.contrib.auth import get_user_model
 from fora.tests.base import BaseTestCase
 from forums.models import Forum
 from threads.models import Thread
@@ -55,3 +56,16 @@ def test_threads(self):
         threads = self.forum.threads()
         self.assertEqual(len(threads), 2)
         self.assertEqual(list(threads), [t1, t2])
+
+    def test_moderator_list_empty(self):
+        mods = self.forum.moderator_list()
+        self.assertEqual(len(mods), 0)
+        self.assertEqual(list(mods), [])
+
+    def test_moderator_list(self):
+        u = get_user_model().objects.create(username='mod1', password='p')
+        self.forum.moderators.add(u)
+
+        mods = self.forum.moderator_list()
+        self.assertEqual(len(mods), 1)
+        self.assertEqual(list(mods), [u])