Fix imports for tests

- Also fix optional types for worker

faust/worker.py

@@ -216,20 +216,20 @@ def __init__(
         self,
         app: AppT,
         *services: ServiceT,
-        sensors: Iterable[SensorT] = None,
+        sensors: Optional[Iterable[SensorT]] = None,
         debug: bool = DEBUG,
         quiet: bool = False,
-        loglevel: Union[str, int] = None,
-        logfile: Union[str, IO] = None,
+        loglevel: Union[str, int, None]  = None,
+        logfile: Union[str, IO, None] = None,
         stdout: IO = sys.stdout,
         stderr: IO = sys.stderr,
         blocking_timeout: Optional[float] = None,
-        workdir: Union[Path, str] = None,
+        workdir: Union[Path, str, None] = None,
         console_port: int = CONSOLE_PORT,
         loop: Optional[asyncio.AbstractEventLoop] = None,
         redirect_stdouts: Optional[bool] = None,
-        redirect_stdouts_level: Severity = None,
-        logging_config: Dict = None,
+        redirect_stdouts_level: Optional[Severity] = None,
+        logging_config: Optional[Dict] = None,
         **kwargs: Any,
     ) -> None:
         self.app = app

tests/conftest.py

@@ -2,7 +2,7 @@
 import threading
 import time
 from http import HTTPStatus
-from typing import Any, NamedTuple
+from typing import Any, NamedTuple, Optional
 
 import pytest
 from _pytest.assertion.util import _compare_eq_dict, _compare_eq_set

tests/integration/app.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-from typing import List
+from typing import List, Optional
 
 import faust
 

tests/stress/models.py

@@ -1,6 +1,7 @@
 import random
 from datetime import datetime, timezone
 from itertools import count
+from typing import Optional
 
 import faust
 
@@ -9,7 +10,7 @@ class Withdrawal(faust.Record, isodates=True, serializer="json"):
     user: str
     country: str
     amount: float
-    date: datetime = None
+    date: Optional[datetime] = None
 
 
 def generate_withdrawals(n: Optional[int] = None):

tests/stress/reports/models.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import faust
 
 __all__ = ["Error", "Status"]