def test_downvote(self):
"""Make sure downvoting Yaks/comments works"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
yak = user.post_yak("Test Message", "Hello")
comment = user.post_comment("Test Comment", yak)
upvote_comment_success = user.downvote(comment)
upvote_yak_success = user.downvote(yak)
self.assertEqual(upvote_comment_success, True)
self.assertEqual(upvote_yak_success, True)
self.assertEqual(yak.liked, True)
self.assertEqual(yak.likes, 1)
self.assertEqual(comment.liked, True)
self.assertEqual(comment.likes, 1)
upvote_comment_success = user.downvote(comment)
upvote_yak_success = user.downvote(yak)
self.assertEqual(upvote_comment_success, True)
self.assertEqual(upvote_yak_success, True)
self.assertEqual(yak.liked, True)
self.assertEqual(yak.likes, 1)
self.assertEqual(comment.liked, True)
self.assertEqual(comment.likes, 1)
delete_success = user.delete(comment)
self.assertEqual(delete_success, True)
self.assertEqual(comment.update(), False)
delete_success = user.delete(yak)
self.assertEqual(delete_success, True)
self.assertEqual(yak.update(), False)
def test_set_basecamp(self):
"""Make sure updating basecamp updates user"""
user = yaklient.User(CLEAR_LOC_1)
basecamp_success = user.set_basecamp("Test Basecamp", CLEAR_LOC_2)
self.assertEqual(basecamp_success, True)
self.assertEqual(user.basecamp_name, "Test Basecamp")
self.assertEqual(user.basecamp_set, True)
def test_new_user_init(self):
"""Make sure a new user is initialized correctly"""
user = yaklient.User(CLEAR_LOC_1)
self.assertEqual(user.location, CLEAR_LOC_1)
self.assertIs(user.basecamp_location, None)
self.assertIs(user.basecamp_name, None)
self.assertEqual(user.basecamp_set, False)
self.assertEqual(user.yakarma, 100)
def test_get_other_peek_locations(self):
"""Make sure get other peek locations works. Also tests get peek
Yaks"""
user = yaklient.User(CLEAR_LOC_1)
peek_locations = user.get_other_peek_locations()
self.assertNotEqual(peek_locations, [])
self.assertIsInstance(peek_locations[0], PeekLocation)
peek_yaks = user.get_peek_yaks(peek_locations[0])
self.assertNotEqual(peek_yaks, [])
self.assertIsInstance(peek_yaks[0], Yak)
def test_report(self):
"""Make sure reporting Yaks/comments doesn't throw any exceptions"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
yak = user.post_yak("Test Message", "Hello")
comment = user.post_comment("Test Comment", yak)
user.report(yak, "Report")
user.report(comment, "Report")
delete_success = user.delete(yak)
self.assertEqual(delete_success, True)
self.assertEqual(yak.update(), False)
def test_get_user_recent_yaks(self):
"""Make sure getting recent Yaks for a user works"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
user.post_yak("Test Message", "Hello")
yak = user.get_user_recent_yaks()[0]
self.assertEqual(yak.message, "Test Message")
self.assertEqual(yak.handle, "Hello")
delete_success = user.delete(yak)
self.assertEqual(delete_success, True)
self.assertEqual(yak.update(), False)
def test_yak(self):
"""Make sure posting/retrieving/updating/deleting a Yak works"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
yak = user.post_yak("Test Message", "Hello")
self.assertEqual(yak.message, "Test Message")
self.assertEqual(yak.handle, "Hello")
yak.update()
self.assertEqual(yak.message, "Test Message")
self.assertEqual(yak.handle, "Hello")
delete_success = user.delete(yak)
self.assertEqual(delete_success, True)
self.assertEqual(yak.update(), False)
def test_comment(self):
"""Make sure posting/retrieving/updating/deleting a comment works"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
yak = user.post_yak("Test Message", "Hello")
comment = user.post_comment("Test Comment", yak)
self.assertEqual(comment.comment, "Test Comment")
comment.update()
self.assertEqual(comment.comment, "Test Comment")
delete_success = user.delete(comment)
self.assertEqual(delete_success, True)
self.assertEqual(comment.update(), False)
delete_success = user.delete(yak)
self.assertEqual(delete_success, True)
self.assertEqual(yak.update(), False)
def test_notifications(self):
"""Make sure notifications can be created, read, and unread"""
users = [yaklient.User(CLEAR_LOC_1, user_id) for user_id in USER_IDS]
yak = users[0].post_yak("Test Message", "Hello")
comment = users[0].post_comment("Test Comment", yak)
for user in users:
comment_downvote_success = user.downvote(comment)
self.assertEqual(comment_downvote_success, True)
yak_downvote_success = user.downvote(yak)
self.assertEqual(yak_downvote_success, True)
notifs = users[0].get_notifications()
self.assertEqual(len(notifs), 2)
self.assertEqual(notifs[0].status, "unread")
self.assertEqual(notifs[1].status, "unread")
mark_read_success = notifs[0].mark_read()
self.assertEqual(mark_read_success, True)
self.assertEqual(notifs[0].status, "read")
mark_read_success = notifs[1].mark_read()
self.assertEqual(mark_read_success, True)
self.assertEqual(notifs[1].status, "read")
mark_unread_success = notifs[0].mark_unread()
self.assertEqual(mark_unread_success, True)
self.assertEqual(notifs[0].status, "unread")
mark_unread_success = notifs[1].mark_unread()
self.assertEqual(mark_unread_success, True)
self.assertEqual(notifs[1].status, "unread")
users[0].mark_notifications_read()
notifs = users[0].get_notifications()
self.assertEqual(len(notifs), 2)
self.assertEqual(notifs[0].status, "read")
self.assertEqual(notifs[1].status, "read")
users[0].mark_notifications_unread()
notifs = users[0].get_notifications()
self.assertEqual(len(notifs), 2)
self.assertEqual(notifs[0].status, "unread")
self.assertEqual(notifs[1].status, "unread")
def test_invalid_basecamp(self):
"""Make sure error is caught when basecamp is not set and it is
used in a get_yaks request"""
user = yaklient.User(CLEAR_LOC_1)
self.assertRaises(NoBasecampSetError, user.get_yaks, basecamp=True)
def test_old_user_init(self):
"""Make sure an existing user is initialized correctly"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
self.assertEqual(user.location, CLEAR_LOC_1)
self.assertIs(user.basecamp_location, None)
self.assertEqual(user.user_id, USER_IDS[0])
def test_no_yaks(self):
"""Make sure a location with no Yaks returns an empty list"""
user = yaklient.User(CLEAR_LOC_1)
self.assertEqual(user.get_yaks(), [])
def test_empty_notifications(self):
"""Make sure a new user has no notifications"""
user = yaklient.User(CLEAR_LOC_1)
self.assertEqual(user.get_notifications(), [])
def test_contact_yikyak(self):
"""Make sure contacting Yik Yak doesn't throw any exceptions"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
user.contact_yikyak("message", "category", "email")
def test_submit_peek_yak(self):
"""Make sure submitting a peek yak doesn't throw any exceptions"""
user = yaklient.User(CLEAR_LOC_1, USER_IDS[0])
peek_locations = user.get_featured_peek_locations()
user.submit_peek_yak("Test Message", peek_locations[0].peek_id, "Test")
def test_get_area_top_yaks(self):
"""Make sure getting area top Yaks at large place is successful"""
user = yaklient.User(BIG_LOC)
yaks = user.get_area_top_yaks()
self.assertNotEqual(yaks, [])
self.assertIsInstance(yaks[0], Yak)
