mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-12 21:28:29 +03:00
Fix up unit tests - dataclass + subclasses not having @dataclass was breaking it
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
from typing import Type
|
||||
|
||||
import pytest
|
||||
|
||||
from auto_archiver.core.metadata import Metadata
|
||||
from auto_archiver.core.extractor import Extractor
|
||||
|
||||
|
||||
class TestExtractorBase(object):
|
||||
|
||||
extractor_module: str = None
|
||||
@@ -13,7 +16,7 @@ class TestExtractorBase(object):
|
||||
assert self.extractor_module is not None, "self.extractor_module must be set on the subclass"
|
||||
assert self.config is not None, "self.config must be a dict set on the subclass"
|
||||
|
||||
self.extractor: Extractor = setup_module(self.extractor_module, self.config)
|
||||
self.extractor: Type[Extractor] = setup_module(self.extractor_module, self.config)
|
||||
|
||||
def assertValidResponseMetadata(self, test_response: Metadata, title: str, timestamp: str, status: str = ""):
|
||||
assert test_response is not False
|
||||
|
||||
Reference in New Issue
Block a user