From c7d0ba92732d35f6d57b0989c854676c597f48ee Mon Sep 17 00:00:00 2001
From: Daniel <d.brummerloh@gmail.com>
Date: Tue, 21 Nov 2023 22:16:11 +0100
Subject: [PATCH] improve typing

---
 src/test_wrapper.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/test_wrapper.py b/src/test_wrapper.py
index 6327920..8c60908 100644
--- a/src/test_wrapper.py
+++ b/src/test_wrapper.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Protocol
+from typing import Optional, Protocol
 
 from dotenv import dotenv_values
 from icecream import ic
@@ -20,11 +20,13 @@ RUNNER_DICT: dict[str, type[TestRunner]] = {"wordpress": RunnerWordpress}
 class Wrapper:
     def __init__(self, dotenv_path: Path):
         assert dotenv_path.is_file()
-        config = dotenv_values(dotenv_path)
+        config: dict[str, Optional[str]] = dotenv_values(dotenv_path)
         runner = self.load_runner(config, dotenv_path)
         runner.run_tests()
 
-    def load_runner(self, config, dotenv_path):
+    def load_runner(
+        self, config: dict[str, Optional[str]], dotenv_path: Path
+    ) -> TestRunner:
         config_type = config["TYPE"]
         ic(config_type)
         RunnerClass = RUNNER_DICT[config_type]