allow provider default none

This commit is contained in:
Diogo Sousa
2023-07-17 12:49:17 +01:00
parent d2b57cda86
commit 869001fdeb
2 changed files with 3 additions and 2 deletions

View File

@@ -150,6 +150,7 @@ class ParametersBuilder:
provider = (
random_choice
if route_defaults is None
or route_defaults.get("provider", None) is None
else route_defaults.get("provider", random_choice)
)
kwargs["provider_choices"] = {"provider": provider}

View File

@@ -1,4 +1,4 @@
from typing import Dict
from typing import Dict, Optional
from pydantic import BaseModel, Field
@@ -9,7 +9,7 @@ class Defaults(BaseModel):
class Config:
validate_assignment = True
routes: Dict[str, Dict[str, str]] = Field(default_factory=dict)
routes: Dict[str, Dict[str, Optional[str]]] = Field(default_factory=dict)
def __repr__(self) -> str:
return (