mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-11 04:38:29 +03:00
4f2b9baa73
1. Extract more metadata 2. Better extract thumbnail 3. Setup framework for specific sites to provide more granular metadata processing
28 lines
958 B
Python
28 lines
958 B
Python
import pytest
|
|
|
|
from auto_archiver.core import Metadata
|
|
from auto_archiver.core import Step
|
|
from auto_archiver.core.metadata import Metadata
|
|
|
|
class TestArchiverBase(object):
|
|
|
|
archiver_class = None
|
|
config = None
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_archiver(self):
|
|
assert self.archiver_class is not None, "self.archiver_class must be set on the subclass"
|
|
assert self.config is not None, "self.config must be a dict set on the subclass"
|
|
self.archiver = self.archiver_class({self.archiver_class.name: self.config})
|
|
|
|
def assertValidResponseMetadata(self, test_response: Metadata, title: str, timestamp: str, status: str = ""):
|
|
assert test_response is not False
|
|
|
|
if not status:
|
|
assert test_response.is_success()
|
|
else:
|
|
assert status == test_response.status
|
|
|
|
assert title == test_response.get_title()
|
|
assert timestamp, test_response.get("timestamp")
|