Added tests for getTags()

Author: turbodog

test_rss2email.py

@@ -7,8 +7,10 @@
 
 import unittest
 from rss2email import *
+import rss2email
+import feedparser
 
-class Test_validateEmail(unittest.TestCase):
+class test_validateEmail(unittest.TestCase):
 	""""""
 	def test_valid_email(self):
 		email = validateEmail("[email protected]", "[email protected]")
@@ -26,5 +28,76 @@ def test_no_at(self):
 		email = validateEmail("invalid", "[email protected]")
 		self.assertEqual(email, "[email protected]")
 
+class test_getName(unittest.TestCase):
+    """"""
+    def setUp(self):
+        self.feed = feedparser.parse("""
+<feed xmlns="http://www.w3.org/2005/Atom">
+<entry>
+  <author>
+    <name>Example author</name>
+    <email>[email protected]</email>
+    <url>http://example.com/</url>
+  </author>
+</entry>
+</feed>
+        """)
+        self.entry = self.feed.entries[0]
+
+
+    def test_no_friendly_name(self):
+        rss2email.NO_FRIENDLY_NAME = 1
+        name = getName(0, 0)
+        rss2email.NO_FRIENDLY_NAME = 0
+        self.assertEqual(name, '')
+        
+    def test_override_from(self):
+        # have to fake url attribute because it is only set on downloaded feeds
+        urlToOverride = 'http://example.com/feed/'
+        self.feed['url'] = urlToOverride
+        rss2email.OVERRIDE_FROM = {urlToOverride:'override'}
+        name = getName(self.feed, self.entry)
+        self.assertEqual(name, 'override')
+
+    def test_no_friendly_name_negative(self):
+        rss2email.NO_FRIENDLY_NAME = 0
+        name=getName(self.feed, self.entry)
+        self.assertEqual(name, 'Example author')
+
+class test_getTags(unittest.TestCase):
+    """"""
+    def test_valid_tags(self):
+        entry = {'tags': [{'term': u'tag1', 'scheme': None, 'label': None}]}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "tag1")
+
+    def test_no_tags(self):
+        entry = {}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "")
+
+    def test_empty_tags(self):
+        entry = {'tags': []}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "")
+
+    def test_no_term(self):
+        entry = {'tags': [{'scheme': None, 'label': None}]}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "")
+
+    def test_empty_term(self):
+        entry = {'tags': [{'term': u'', 'scheme': None, 'label': None}]}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "")
+
+    def test_multiple_tags(self):
+        entry = {'tags': [{'term': u'tag1', 'scheme': None, 'label': None}, {'term': u'tag2', 'scheme': None, 'label': None}]}
+        tagline = getTags(entry)
+        self.assertEqual(tagline, "tag1,tag2")
+
+    
+
 if __name__ == '__main__':
     unittest.main()
+