http poller fixes

amyasnikov · Feb 27, 2024 · ee00199 · ee00199
1 parent 97810ce
commit ee00199
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/validity/pollers/default_credentials.py b/validity/pollers/default_credentials.py
@@ -27,7 +27,7 @@ class ScrapliNeconfPrivateCreds(BaseModel):
 
 
 class RequestsPublicCreds(BaseModel):
-    url: str = "https://{{device.primary_ip}}/{{command.parameters.url_path.lstrip('/')}}"
+    url: str = "https://{{device.primary_ip.address.ip}}/{{command.parameters.url_path.lstrip('/')}}"
 
 
 class ConnectionTypeCredentials(BaseModel):

diff --git a/validity/pollers/http.py b/validity/pollers/http.py
@@ -14,8 +14,11 @@
 
 
 class RequestParams(BaseModel, extra="allow"):
-    url: str = Field("https://{{device.primary_ip}}/{{command.parameters.url_path.lstrip('/')}}", exclude=True)
+    url: str = Field(
+        "https://{{device.primary_ip.address.ip}}/{{command.parameters.url_path.lstrip('/')}}", exclude=True
+    )
     verify: bool | str = False
+    auth: tuple[str, ...] | None = None
 
     def rendered_url(self, device: "Device", command: "Command") -> str:
         return Environment().from_string(self.url).render(device=device, command=command)