Remove unittest and switch to pytest fully

This commit is contained in:
Patrick Robertson
2025-01-14 16:28:39 +01:00
parent cef4037ad5
commit 6f10270baf
9 changed files with 148 additions and 195 deletions

View File

@@ -1,32 +1,22 @@
import tempfile
import os
import unittest
from auto_archiver.databases.csv_db import CSVDb
from auto_archiver.core import Metadata
def test_store_item(tmp_path):
"""Tests storing an item in the CSV database"""
class TestCSVdb(unittest.TestCase):
temp_db = tmp_path / "temp_db.csv"
db = CSVDb({
"csv_db": {"csv_file": temp_db.as_posix()}
})
def setUp(self):
_, temp_db = tempfile.mkstemp(suffix="csv")
self.temp_db = temp_db
item = Metadata().set_url("http://example.com").set_title("Example").set_content("Example content").success("my-archiver")
def tearDown(self):
os.remove(self.temp_db)
db.done(item)
def test_store_item(self):
db = CSVDb({
"csv_db": {"csv_file": self.temp_db}
})
with open(temp_db, "r", encoding="utf-8") as f:
assert f.read().strip() == f"status,metadata,media\nmy-archiver: success,\"{{'_processed_at': {repr(item.get('_processed_at'))}, 'url': 'http://example.com', 'title': 'Example', 'content': 'Example content'}}\",[]"
item = Metadata().set_url("http://example.com").set_title("Example").set_content("Example content").success("my-archiver")
db.done(item)
with open(self.temp_db, "r") as f:
assert f.read().strip() == f"status,metadata,media\nmy-archiver: success,\"{{'_processed_at': {repr(item.get('_processed_at'))}, 'url': 'http://example.com', 'title': 'Example', 'content': 'Example content'}}\",[]"
# TODO: csv db doesn't have a fetch method - need to add it (?)
# assert db.fetch(item) == item
# TODO: csv db doesn't have a fetch method - need to add it (?)
# assert db.fetch(item) == item