backporting test fix solutions branch to master

Author: pybites

04/test_usertweets.py

@@ -1,42 +1,59 @@
+from collections import namedtuple
 import csv
-import datetime
 import unittest
+from unittest.mock import patch
 
-from usertweets import UserTweets
-from usertweets import NUM_TWEETS
 
-DT = datetime.datetime(2017, 1, 13, 9, 0, 5)
+from tweets import TWEETS  # mock data
+from usertweets import UserTweets, NUM_TWEETS
+
 HANDLE = 'pybites'
 MAX_ID = '819831370113351680'
-TWEETS = (
-    """5 cool things you can do with itertools https://t.co/Nk4s3yL6zL #python""",
-    """How to create a nice-looking HTML page of your #Kindle book highlights (notes) https://t.co/HKFK7inhUa #python""",
-)
-USER = UserTweets(HANDLE, max_id=MAX_ID)
-
-def read_csv():
-    with open(USER.output_file) as f:
+
+Tweet = namedtuple('Tweet', ['id_str', 'created_at', 'text'])
+
+
+def read_csv(fname):
+    with open(fname) as f:
+        has_header = csv.Sniffer().has_header(f.readline())
+        f.seek(0)
         r = csv.reader(f)
-        next(r, None)  # skip the headers
-        return list(r) 
+        if has_header:
+            next(r, None)  # skip the header
+        return [Tweet(*tw) for tw in r]  # list(r)
+
 
 class TestUserTweets(unittest.TestCase):
+    def setUp(self):
+        super().setUp()
+        with patch('tweepy.API.user_timeline') as mock_timeline:
+            mock_timeline.return_value = TWEETS
+            self.user = UserTweets(HANDLE, max_id=MAX_ID)
+
+    def tearDown(self):
+        self.user = None
+        super().tearDown()
 
     def test_num_tweets(self):
-        self.assertEqual(len(USER), NUM_TWEETS)
+        self.assertEqual(len(self.user), NUM_TWEETS)
 
     def test_first_tweet_returned_by_api(self):
-        self.assertEqual(USER[0].id_str, MAX_ID)
-        self.assertEqual(USER[0].created_at, DT)
-        self.assertEqual(USER[0].text, TWEETS[0])
+        tw_n = 0
+        self.assertEqual(self.user[tw_n].id_str, MAX_ID)
+        self.assertEqual(self.user[tw_n].created_at, TWEETS[tw_n].created_at)
+        self.assertEqual(self.user[tw_n].text, TWEETS[tw_n].text)
 
     def test_read_back_from_cached_csv(self):
-        csv_tweets = read_csv()
+        csv_tweets = read_csv(self.user.output_file)
         self.assertEqual(len(csv_tweets), NUM_TWEETS)
-        self.assertEqual(csv_tweets[0][0], MAX_ID)
-        self.assertEqual(csv_tweets[0][1], str(DT))
-        self.assertEqual(csv_tweets[0][2], TWEETS[0])
-        self.assertEqual(csv_tweets[-1][2], TWEETS[1])
+        tw_n = 0  # first
+        self.assertEqual(csv_tweets[tw_n].id_str, MAX_ID)
+        self.assertEqual(csv_tweets[tw_n].created_at,
+                         str(TWEETS[tw_n].created_at))
+        self.assertEqual(csv_tweets[tw_n].text, TWEETS[tw_n].text)
+        tw_n = -1  # last
+        self.assertEqual(csv_tweets[tw_n].text, TWEETS[tw_n].text)
+
 
 if __name__ == "__main__":
     unittest.main()