test: replace toml usage with tomli and tomli-w

Replaces toml library calls with `tomli` and `tomli-w` calls and removes
the `toml` library dependency from dev dependencies.

Author: dogukancagatay

pdm.lock

@@ -44,7 +44,7 @@ dev = [
     "urllib3>=2.0.7",
     "pytest-html>=4.1.1",
     "pytest-asyncio>=0.23.3",
-    "toml>=0.10.2",
+    "tomli-w>=1.0.0",
 ]
 [build-system]
 requires = ["pdm-pep517>=1.0"]

pyproject.toml

@@ -1,13 +1,15 @@
 import logging
 import os
+from pathlib import Path
 import subprocess
 import sys
 import tempfile
 import time
 import pytest
-import toml
+import tomli
 from threading import Thread
-from toml import load
+
+import tomli_w
 from test.e2e.utils.database_utils import TestDb
 from test.e2e.env_config import PORT
 from test.e2e.pageobjects.create_paste_page import CreatePastePage
@@ -69,11 +71,14 @@ def create_paste_page(page: Page) -> CreatePastePage:
 def database() -> Generator[None, None, None]:
     log.info("Setting up temp database")
     with tempfile.NamedTemporaryFile(suffix="", delete=False) as temp:
-        props = load(config_path)
+        props = tomli.loads(Path(config_path).read_text())
         props["database_uri"] = f"sqlite:///{temp.name}"
-        props["expiries"] = {"1hour": 4, "1day": 4}  # It expires in 4 seconds instead of an hour
+        props["expiries"] = {
+            "1hour": 4,
+            "1day": 4,
+        }  # It expires in 4 seconds instead of an hour
         with open(config_path, "w") as config_file:
-            toml.dump(props, config_file)
+            tomli_w.dump(props, config_file)
     yield
     log.info("Tearing down temp database")
     close_all_sessions()
@@ -85,4 +90,6 @@ def database() -> Generator[None, None, None]:
 @pytest.fixture
 def clear_db():
     log.info("Clearing temp database")
-    TestDb(load(config_path)["database_uri"]).clear_tables("paste", "file")
+    TestDb(
+        tomli.loads(Path(config_path).read_text())["database_uri"]
+    ).clear_tables("paste", "file")

test/e2e/conftest.py

@@ -1,8 +1,9 @@
 import os
+from pathlib import Path
 
-from toml import load
+from tomli import loads
 
 
 def load_props():
     config_path = os.path.join("test", "e2e", "pinnwand.toml")
-    return load(config_path)
+    return loads(Path(config_path).read_text())

test/e2e/utils/props_utils.py

@@ -1,8 +1,10 @@
 import os
 
+import tomli_w
+
 from pinnwand.configuration import Configuration
 import tempfile
-import toml
+import tomli
 import pytest
 
 
@@ -39,14 +41,13 @@ def test_env_variables_precedence(monkeypatch, temp_dotenv_file):
 def test_load_file_config():
     with tempfile.NamedTemporaryFile(suffix="", delete=False) as temp:
 
-        props = toml.load(temp.name)
+        props = tomli.load(temp)
         url = "sqlite:///database.db"
         props["database_uri"] = url
 
-        with open(temp.name, "w") as config_file:
-            toml.dump(props, config_file)
+        with open(temp.name, "wb") as config_file:
+            tomli_w.dump(props, config_file)
             config: Configuration = Configuration()
 
         config.load_config_file(temp.name)
         assert config.database_uri == url
-

test/integration/test_config.py

@@ -1,13 +1,15 @@
 import asyncio
 import os
+from pathlib import Path
 import tempfile
-import toml
+import tomli
 from unittest.mock import patch, MagicMock
 from sqlalchemy import create_engine
 
 
 import pytest
 from click.testing import CliRunner
+import tomli_w
 
 from pinnwand import command, utility
 from pinnwand.database import utils, manager, models
@@ -19,11 +21,11 @@
 def temp_database_url() -> str:
     """Override the config's database_url with a temporary one."""
     with tempfile.NamedTemporaryFile(suffix="", delete=False) as temp:
-        props = toml.load(config_path)
+        props = tomli.loads(Path(config_path).read_text())
         url = f"sqlite:///{temp.name}"
         props["database_uri"] = url
-        with open(config_path, "w") as config_file:
-            toml.dump(props, config_file)
+        with open(config_path, "wb") as config_file:
+            tomli_w.dump(props, config_file)
 
         yield url