21 lines
807 B
Python
21 lines
807 B
Python
|
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))
|