added some tests for save_api interface
This commit is contained in:
parent
7844d15d99
commit
d506685f68
20
tests/test_save_api.py
Normal file
20
tests/test_save_api.py
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
import pytest
|
||||||
|
import time
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from waybackpy.save_api import WaybackMachineSaveAPI
|
||||||
|
|
||||||
|
|
||||||
|
def test_save():
|
||||||
|
url = "https://github.com/akamhy/waybackpy"
|
||||||
|
user_agent = "Mozilla/5.0 (MacBook Air; M1 Mac OS X 11_4) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.1.1 Safari/604.1"
|
||||||
|
save_api = WaybackMachineSaveAPI(url, user_agent)
|
||||||
|
save_api.save()
|
||||||
|
archive_url = save_api.archive_url
|
||||||
|
timestamp = save_api.timestamp()
|
||||||
|
headers = save_api.headers # CaseInsensitiveDict
|
||||||
|
cached_save = save_api.cached_save
|
||||||
|
assert cached_save in [True, False]
|
||||||
|
assert archive_url.find("github.com/akamhy/waybackpy") != -1
|
||||||
|
assert str(headers).find("github.com/akamhy/waybackpy") != -1
|
||||||
|
assert type(save_api.timestamp()) == type(datetime(year=2020, month=10, day=2))
|
Loading…
Reference in New Issue
Block a user