[TASK] Add Quit message.

2015-10-25 20:52:09 +01:00
parent 35f358cccf
commit 1c99d74f02
2 changed files with 26 additions and 1 deletions
@@ -156,4 +156,17 @@ class Message(unittest.TestCase):

    def test_construct_topic(self):
        msg = irc.Topic(channel="#politics", topic="Yes, we can!")
-        self.assertEqual(str(msg), "TOPIC #politics :Yes, we can!")
+        self.assertEqual(str(msg), "TOPIC #politics :Yes, we can!")
+
+
+    def test_parse_quit(self):
+        raw = ":worker!~frustrated@company.com QUIT :This is it!"
+        msg = irc.Message.from_string(raw)
+        self.assertIsInstance(msg, irc.Quit, msg="Not a Quit!")
+        self.assertEqual(msg.nick, "worker")
+        self.assertEqual(msg.message, "This is it!")
+
+    def test_construct_quit(self):
+        msg = irc.Quit(message="Good bye!")
+        self.assertEqual(str(msg), "QUIT :Good bye!")
+