def setUp(self):
self.oauth = OAuthInfo()
self.pyny = PynyTwitter(self.oauth)
class TestPynyTwitter(unittest.TestCase):
def setUp(self):
self.oauth = OAuthInfo()
self.pyny = PynyTwitter(self.oauth)
def test_time_line_def(self):
tweets = self.pyny.home_timeline()
self.assertEqual(len(tweets), 20)
def test_time_line_5(self):
tweets = self.pyny.home_timeline(count=5)
self.assertEqual(len(tweets), 5)
def test_user_timeline(self):
tweets = self.pyny.user_timeline(count=3)
self.assertEqual(len(tweets), 3)
def test_update_status(self):
n = random.randint(1, 999999999)
tweet_text = "this is a test-tweet rand=" + str(n)
self.pyny.update_status(tweet_text)
time.sleep(3) # waits for consistence in API calls
tweet = self.pyny.user_timeline(count=1)[0]
self.assertEqual(tweet.text, tweet_text)
def test_mentions(self):
tweets = self.pyny.mentions()
self.assertEqual(len(tweets), 1)
# def test_plainTextWriter_stdout(self): #prints to screen
# w = PlainTextWriter()
# ui = PynyTwitterUI(w)
# ui.get_home_timeline()
def test_plainTextWriter_file_wipe_mode(self):
f = "./log.txt"
open(f, "w").close()
times = 3
w = PlainTextWriter(f, "w")
ui = PynyTwitterUI(w)
for i in range(times):
ui.get_home_timeline()
expected_lines = 1 * 2 * 20
actual_lines = len(open(f, "r").readlines())
self.assertEqual(expected_lines, actual_lines)
def test_plainTextWriter_file_append_mode(self):
f = "./log.txt"
open(f, "w").close()
times = 3
w = PlainTextWriter(f, "a")
ui = PynyTwitterUI(w)
for i in range(times):
ui.get_home_timeline()
expected_lines = times * 2 * 20
actual_lines = len(open(f, "r").readlines())
self.assertEqual(expected_lines, actual_lines)
def test_plainTextWriter_file_update_status(self):
n = random.randint(1, 999999999)
tweet_text = "this is a test-tweet rand=" + str(n)
f = "./log.txt"
open(f, "w").close()
w = PlainTextWriter(f, "w")
ui = PynyTwitterUI(w)
ui.update_status(tweet_text)
actual_lines = open(f, "r").readlines()
self.assertEqual(tweet_text, actual_lines[0])
def test_plainTextWriter_file_user_timeline_5(self):
f = "./log.txt"
open(f, "w").close()
count = 5
w = PlainTextWriter(f, "w")
ui = PynyTwitterUI(w)
ui.get_user_timeline(count)
expected_lines = 1 * 2 * count
actual_lines = len(open(f, "r").readlines())
self.assertEqual(expected_lines, actual_lines)
