mirror of
https://github.com/instaloader/instaloader.git
synced 2024-08-17 04:09:39 +02:00
If Instaloader was called with --fast-update and encountered a post where nothing has been downloaded because of flags that disable download of certain types of media, the download was aborted. This behavior is now fixed. Closes #183.
178 lines
6.4 KiB
Python
178 lines
6.4 KiB
Python
"""Unit Tests for Instaloader"""
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
from itertools import islice
|
|
|
|
import instaloader
|
|
|
|
PROFILE_WITH_HIGHLIGHTS = 325732271
|
|
PUBLIC_PROFILE = "selenagomez"
|
|
PUBLIC_PROFILE_ID = 460563723
|
|
HASHTAG = "kitten"
|
|
OWN_USERNAME = "aandergr"
|
|
NORMAL_MAX_COUNT = 2
|
|
PAGING_MAX_COUNT = 15
|
|
PRIVATE_PROFILE = "aandergr"
|
|
|
|
# Preserve query timestamps (rate control) between tests to not get rate limited
|
|
instaloadercontext_query_timestamps = list()
|
|
|
|
|
|
class TestInstaloaderAnonymously(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.dir = tempfile.mkdtemp()
|
|
print("Testing in {}".format(self.dir))
|
|
os.chdir(self.dir)
|
|
self.L = instaloader.Instaloader(download_geotags=True,
|
|
download_comments=True,
|
|
save_metadata=True)
|
|
self.L.context.raise_all_errors = True
|
|
self.L.context.query_timestamps = instaloadercontext_query_timestamps
|
|
|
|
def tearDown(self):
|
|
global instaloadercontext_query_timestamps
|
|
instaloadercontext_query_timestamps = self.L.context.query_timestamps
|
|
self.L.close()
|
|
os.chdir('/')
|
|
print("Removing {}".format(self.dir))
|
|
shutil.rmtree(self.dir)
|
|
|
|
def post_paging_test(self, iterator):
|
|
previous_post = None
|
|
for post in islice(iterator, PAGING_MAX_COUNT):
|
|
print(post)
|
|
if previous_post:
|
|
self.assertTrue(post.date_utc < previous_post.date_utc)
|
|
previous_post = post
|
|
|
|
@unittest.SkipTest
|
|
def test_public_profile_download(self):
|
|
self.L.download_profile(PUBLIC_PROFILE, profile_pic=False, fast_update=True)
|
|
self.L.download_profile(PUBLIC_PROFILE, profile_pic=False, fast_update=True)
|
|
|
|
def test_public_profile_paging(self):
|
|
self.post_paging_test(instaloader.Profile.from_username(self.L.context, PUBLIC_PROFILE).get_posts())
|
|
|
|
def test_profile_pic_download(self):
|
|
self.L.download_profile(PUBLIC_PROFILE, profile_pic_only=True)
|
|
|
|
def test_hashtag_download(self):
|
|
self.L.download_hashtag(HASHTAG, NORMAL_MAX_COUNT)
|
|
|
|
def test_hashtag_paging(self):
|
|
for count, post in enumerate(self.L.get_hashtag_posts(HASHTAG)):
|
|
print(post)
|
|
if count == PAGING_MAX_COUNT:
|
|
break
|
|
|
|
def test_get_id_by_username(self):
|
|
self.assertEqual(PUBLIC_PROFILE_ID,
|
|
instaloader.Profile.from_username(self.L.context, PUBLIC_PROFILE).userid)
|
|
|
|
def test_get_username_by_id(self):
|
|
self.assertEqual(PUBLIC_PROFILE.lower(),
|
|
instaloader.Profile.from_id(self.L.context, PUBLIC_PROFILE_ID).username)
|
|
|
|
def test_post_from_mediaid(self):
|
|
for post in instaloader.Profile.from_username(self.L.context, PUBLIC_PROFILE).get_posts():
|
|
post2 = instaloader.Post.from_mediaid(self.L.context, post.mediaid)
|
|
self.assertEqual(post, post2)
|
|
break
|
|
|
|
def test_public_profile_tagged_paging(self):
|
|
for post in islice(instaloader.Profile.from_username(self.L.context, PUBLIC_PROFILE).get_tagged_posts(),
|
|
PAGING_MAX_COUNT):
|
|
print(post)
|
|
|
|
|
|
class TestInstaloaderLoggedIn(TestInstaloaderAnonymously):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.L.load_session_from_file(OWN_USERNAME)
|
|
|
|
@unittest.SkipTest
|
|
def test_stories_download(self):
|
|
self.L.download_stories()
|
|
|
|
@unittest.SkipTest
|
|
def test_private_profile_download(self):
|
|
self.L.download_profile(PRIVATE_PROFILE, download_stories=True)
|
|
|
|
def test_stories_paging(self):
|
|
for user_story in self.L.get_stories():
|
|
print("profile {}.".format(user_story.owner_username))
|
|
for item in user_story.get_items():
|
|
print(item)
|
|
|
|
def test_highlights_paging(self):
|
|
for user_highlight in self.L.get_highlights(PROFILE_WITH_HIGHLIGHTS):
|
|
print("Retrieving {} highlights \"{}\" from profile {}".format(user_highlight.itemcount,
|
|
user_highlight.title,
|
|
user_highlight.owner_username))
|
|
for item in user_highlight.get_items():
|
|
print(item)
|
|
|
|
def test_private_profile_paging(self):
|
|
self.post_paging_test(instaloader.Profile.from_username(self.L.context, PRIVATE_PROFILE).get_posts())
|
|
|
|
def test_feed_download(self):
|
|
self.L.download_feed_posts(NORMAL_MAX_COUNT)
|
|
|
|
def test_feed_paging(self):
|
|
for count, post in enumerate(self.L.get_feed_posts()):
|
|
print(post)
|
|
if count == PAGING_MAX_COUNT:
|
|
break
|
|
|
|
def test_saved_download(self):
|
|
self.L.download_saved_posts(NORMAL_MAX_COUNT)
|
|
|
|
def test_saved_paging(self):
|
|
for count, post in enumerate(instaloader.Profile.from_username(self.L.context, OWN_USERNAME).get_saved_posts()):
|
|
print(post)
|
|
if count == PAGING_MAX_COUNT:
|
|
break
|
|
|
|
def test_test_login(self):
|
|
self.assertEqual(OWN_USERNAME, self.L.test_login())
|
|
|
|
def test_followees_and_stories(self):
|
|
profile = instaloader.Profile.from_username(self.L.context, OWN_USERNAME)
|
|
for f in islice(profile.get_followees(), PAGING_MAX_COUNT):
|
|
self.L.download_profile(f.username, profile_pic=False, download_stories_only=True)
|
|
|
|
def test_get_followees(self):
|
|
profile = instaloader.Profile.from_username(self.L.context, OWN_USERNAME)
|
|
for f in profile.get_followees():
|
|
print(f.username)
|
|
|
|
def test_get_followers(self):
|
|
profile = instaloader.Profile.from_username(self.L.context, OWN_USERNAME)
|
|
for f in profile.get_followers():
|
|
print(f.username)
|
|
|
|
def test_get_username_by_id(self):
|
|
self.assertEqual(PUBLIC_PROFILE.lower(),
|
|
instaloader.Profile.from_id(self.L.context, PUBLIC_PROFILE_ID).username)
|
|
|
|
def test_get_likes(self):
|
|
for post in instaloader.Profile.from_username(self.L.context, OWN_USERNAME).get_posts():
|
|
for like in post.get_likes():
|
|
print(like.username)
|
|
break
|
|
|
|
def test_explore_paging(self):
|
|
for count, post in enumerate(self.L.get_explore_posts()):
|
|
print(post)
|
|
if count == PAGING_MAX_COUNT:
|
|
break
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|