Doctest for publish_subscribe

Author: gyermolenko

patterns/behavioral/publish_subscribe.py

@@ -50,41 +50,43 @@ def run(self, msg):
 
 
 def main():
-    message_center = Provider()
-
-    fftv = Publisher(message_center)
-
-    jim = Subscriber("jim", message_center)
-    jim.subscribe("cartoon")
-    jack = Subscriber("jack", message_center)
-    jack.subscribe("music")
-    gee = Subscriber("gee", message_center)
-    gee.subscribe("movie")
-    vani = Subscriber("vani", message_center)
-    vani.subscribe("movie")
-    vani.unsubscribe("movie")
-
-    fftv.publish("cartoon")
-    fftv.publish("music")
-    fftv.publish("ads")
-    fftv.publish("movie")
-    fftv.publish("cartoon")
-    fftv.publish("cartoon")
-    fftv.publish("movie")
-    fftv.publish("blank")
-
-    message_center.update()
+    """
+    >>> message_center = Provider()
+
+    >>> fftv = Publisher(message_center)
+
+    >>> jim = Subscriber("jim", message_center)
+    >>> jim.subscribe("cartoon")
+    >>> jack = Subscriber("jack", message_center)
+    >>> jack.subscribe("music")
+    >>> gee = Subscriber("gee", message_center)
+    >>> gee.subscribe("movie")
+    >>> vani = Subscriber("vani", message_center)
+    >>> vani.subscribe("movie")
+    >>> vani.unsubscribe("movie")
+
+    # Note that no one subscirbed to `ads`
+    # and that vani changed their mind
+
+    >>> fftv.publish("cartoon")
+    >>> fftv.publish("music")
+    >>> fftv.publish("ads")
+    >>> fftv.publish("movie")
+    >>> fftv.publish("cartoon")
+    >>> fftv.publish("cartoon")
+    >>> fftv.publish("movie")
+    >>> fftv.publish("blank")
+
+    >>> message_center.update()
+    jim got cartoon
+    jack got music
+    gee got movie
+    jim got cartoon
+    jim got cartoon
+    gee got movie
+    """
 
 
 if __name__ == "__main__":
-    main()
-
-
-OUTPUT = """
-jim got cartoon
-jack got music
-gee got movie
-jim got cartoon
-jim got cartoon
-gee got movie
-"""
+    import doctest
+    doctest.testmod()