Add tests for pubbot.trac

pubbothq · Aug 24, 2014 · aec26b4 · aec26b4
1 parent 0d1cd60
commit aec26b4
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/pubbot/trac/receivers.py b/pubbot/trac/receivers.py
@@ -19,7 +19,7 @@
 
 
 @chat_receiver(r'^newticket: (?P<ticket>.*)$')
-def raise_ticket(sender, ticket, **kwargs):
+def raise_ticket(sender, ticket, user, **kwargs):
     instance = "http://localhost/sometrac"
     username = "username"
     password = "password"
@@ -43,7 +43,7 @@ def raise_ticket(sender, ticket, **kwargs):
         "__FORM_TOKEN": token,
         "field_summary": ticket,
         "field_status": "new",
-        "field_reporter": kwargs['source'],
+        "field_reporter": user,
         "field_owner": "",
     })
 

diff --git a/pubbot/trac/tests.py b/pubbot/trac/tests.py
@@ -0,0 +1,22 @@
+import mock
+import unittest
+
+from pubbot.trac import receivers
+
+
+DUMMY_FORM = """
+<html><head><title>Trac</title></head><body>
+<input type="hidden" name="__FORM_TOKEN" value="FOOBAR" />
+</body></html>
+"""
+
+
+class TestRaiseTicket(unittest.TestCase):
+
+    def test_raise_ticket(self):
+        with mock.patch("requests.Session") as Session:
+            Session.return_value.get.return_value.text = DUMMY_FORM
+            Session.return_value.post.return_value.url = 'http://localhost/sometrac/ticket/1'
+
+            r = receivers.raise_ticket(None, user='fred', content="newticket: Raise a ticket")
+            self.assertEqual(r['content'], 'http://localhost/sometrac/ticket/1')