From 725f79466de65743df0c3498a1bb08e106b5209e Mon Sep 17 00:00:00 2001 From: halcy Date: Thu, 17 Nov 2022 22:00:37 +0200 Subject: Fix tests --- tests/test_timeline.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'tests/test_timeline.py') diff --git a/tests/test_timeline.py b/tests/test_timeline.py index fbeae8e..9d63025 100644 --- a/tests/test_timeline.py +++ b/tests/test_timeline.py @@ -2,6 +2,8 @@ import pytest import time from mastodon.Mastodon import MastodonAPIError, MastodonIllegalArgumentError, MastodonUnauthorizedError import datetime +import pickle +import os @pytest.mark.vcr() def test_public_tl_anonymous(api_anonymous, status3): @@ -78,9 +80,13 @@ def test_min_max_id(api, status): @pytest.mark.vcr() def test_min_max_id_datetimes(api, status): - the_past = datetime.datetime.now() - datetime.timedelta(seconds=20) - the_future = datetime.datetime.now() + datetime.timedelta(seconds=20) - the_far_future = datetime.datetime.now() + datetime.timedelta(seconds=40) + if os.path.exists("tests/cassettes/test_min_max_id_datetimes_datetimeobjects.pkl"): + the_past, the_future, the_far_future = pickle.load(open("tests/cassettes/test_min_max_id_datetimes_datetimeobjects.pkl", 'rb')) + else: + the_past = datetime.datetime.now() - datetime.timedelta(seconds=20) + the_future = datetime.datetime.now() + datetime.timedelta(seconds=20) + the_far_future = datetime.datetime.now() + datetime.timedelta(seconds=40) + pickle.dump((the_past, the_future, the_far_future), open("tests/cassettes/test_min_max_id_datetimes_datetimeobjects.pkl", 'wb')) time.sleep(3) tl = api.timeline_home(min_id = the_past, max_id = the_future) -- cgit v1.2.3