From 2d5c4dae524bacded0ac7a56ab18dd74402dcc4f Mon Sep 17 00:00:00 2001
From: Chris Rossi <chris@archimedeanco.com>
Date: Thu, 12 Dec 2013 17:11:32 -0500
Subject: [PATCH] Test coverage.

---
 .../linkedin/management/commands/mailusers.py |  4 ++
 .../commands/tests/test_mailusers.py          | 39 +++++++++++++++++++
 2 files changed, 43 insertions(+)
 create mode 100644 lms/djangoapps/linkedin/management/commands/tests/test_mailusers.py

diff --git a/lms/djangoapps/linkedin/management/commands/mailusers.py b/lms/djangoapps/linkedin/management/commands/mailusers.py
index 83b03e4114..15043afae1 100644
--- a/lms/djangoapps/linkedin/management/commands/mailusers.py
+++ b/lms/djangoapps/linkedin/management/commands/mailusers.py
@@ -13,6 +13,10 @@ from ...models import LinkedIn
 
 
 class Command(BaseCommand):
+    """
+    Django command for inviting users to add their course certificates to their
+    LinkedIn profiles.
+    """
     args = ''
     help = ('Sends emails to edX users that are on LinkedIn who have completed '
             'course certificates, inviting them to add their certificates to '
diff --git a/lms/djangoapps/linkedin/management/commands/tests/test_mailusers.py b/lms/djangoapps/linkedin/management/commands/tests/test_mailusers.py
new file mode 100644
index 0000000000..4c349236cb
--- /dev/null
+++ b/lms/djangoapps/linkedin/management/commands/tests/test_mailusers.py
@@ -0,0 +1,39 @@
+"""
+Test email scripts.
+"""
+import mock
+import unittest
+
+from linkedin.management.commands import mailusers
+
+
+class MailusersTests(unittest.TestCase):
+    """
+    Test mail users command.
+    """
+
+    @mock.patch('linkedin.management.commands.mailusers.send_email')
+    @mock.patch('linkedin.management.commands.mailusers.GeneratedCertificate')
+    @mock.patch('linkedin.management.commands.mailusers.LinkedIn')
+    def test_mail_users(self, linkedin, certificates, send_email):
+        """
+        Test "happy path" for emailing users.
+        """
+        fut = mailusers.Command().handle
+        fred = mock.Mock(user=mock.Mock(certificates=[1, 2]))
+        barney = mock.Mock(user=mock.Mock(certificates=[3]))
+        linkedin.objects.filter.return_value = [fred, barney]
+
+        def filter_user(user):
+            "Mock querying the database."
+            queryset = mock.Mock()
+            queryset.filter.return_value = user.certificates
+            return queryset
+
+        certificates.objects.filter = filter_user
+        fut()
+        self.assertEqual(
+            send_email.call_args_list,
+            [((fred.user, 1),),
+             ((fred.user, 2),),
+             ((barney.user, 3),)])