diff options
Diffstat (limited to 'tests/test_status.py')
-rw-r--r-- | tests/test_status.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tests/test_status.py b/tests/test_status.py index dff9812..7e30362 100644 --- a/tests/test_status.py +++ b/tests/test_status.py | |||
@@ -4,6 +4,8 @@ import datetime | |||
4 | import pytz | 4 | import pytz |
5 | import vcr | 5 | import vcr |
6 | import time | 6 | import time |
7 | import pickle | ||
8 | import os | ||
7 | 9 | ||
8 | @pytest.mark.vcr() | 10 | @pytest.mark.vcr() |
9 | def test_status(status, api): | 11 | def test_status(status, api): |
@@ -171,8 +173,12 @@ def test_scheduled_status(api): | |||
171 | api.scheduled_status_delete(scheduled_toot_2) | 173 | api.scheduled_status_delete(scheduled_toot_2) |
172 | scheduled_toot_list_2 = api.scheduled_statuses() | 174 | scheduled_toot_list_2 = api.scheduled_statuses() |
173 | assert not scheduled_toot_2.id in map(lambda x: x.id, scheduled_toot_list_2) | 175 | assert not scheduled_toot_2.id in map(lambda x: x.id, scheduled_toot_list_2) |
174 | 176 | ||
175 | the_very_immediate_future = datetime.datetime.now() + datetime.timedelta(seconds=5) | 177 | if os.path.exists("tests/cassettes/test_scheduled_status_datetimeobjects.pkl"): |
178 | the_very_immediate_future = pickle.load(open("tests/cassettes/test_scheduled_status_datetimeobjects.pkl", 'rb')) | ||
179 | else: | ||
180 | the_very_immediate_future = datetime.datetime.now() + datetime.timedelta(seconds=5) | ||
181 | pickle.dump(the_very_immediate_future, open("tests/cassettes/test_scheduled_status_datetimeobjects.pkl", 'wb')) | ||
176 | scheduled_toot_4 = api.status_post("please ensure adequate headroom", scheduled_at=the_very_immediate_future) | 182 | scheduled_toot_4 = api.status_post("please ensure adequate headroom", scheduled_at=the_very_immediate_future) |
177 | time.sleep(15) | 183 | time.sleep(15) |
178 | statuses = api.timeline_home() | 184 | statuses = api.timeline_home() |