From f3530249673a28e61decab626d3dfd91a7318720 Mon Sep 17 00:00:00 2001 From: Federico Ressi Date: Wed, 13 Oct 2021 07:38:51 +0200 Subject: [PATCH] Update curl.execute_curl function Change-Id: If43ee5fbdd34a366f9b8ab907e469b0a8d17ae9b --- tobiko/shell/curl/_execute.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/tobiko/shell/curl/_execute.py b/tobiko/shell/curl/_execute.py index fba7d8e9a..740b909dc 100644 --- a/tobiko/shell/curl/_execute.py +++ b/tobiko/shell/curl/_execute.py @@ -33,14 +33,14 @@ CURL_CONNECTION_ERRORS = { def execute_curl( - hostname: typing.Union[str, netaddr.IPAddress, None] = None, - port: typing.Optional[int] = None, - path: typing.Optional[str] = None, - scheme: typing.Optional[str] = None, - ssh_client: typing.Optional[ssh.SSHClientFixture] = None, + hostname: typing.Union[str, netaddr.IPAddress] = None, + port: int = None, + path: str = None, + scheme: str = None, + ssh_client: ssh.SSHClientType = None, connect_timeout: tobiko.Seconds = None, fail_silently: bool = True, - retry_count: typing.Optional[int] = None, + retry_count: int = None, retry_timeout: tobiko.Seconds = None, retry_interval: tobiko.Seconds = None, **execute_params) -> str: @@ -85,7 +85,8 @@ def make_netloc( try: ip_address = netaddr.IPAddress(hostname) except netaddr.AddrFormatError: - netloc = str(hostname).lower() + tobiko.check_valid_type(hostname, str) + netloc = hostname else: if ip_address.version == 6: # Add square brackets around IPv6 address to please curl @@ -105,7 +106,7 @@ def make_netloc( def make_url(scheme: typing.Optional[str] = None, netloc: typing.Optional[str] = None, path: typing.Optional[str] = None) -> str: - return parse.SplitResult(scheme=scheme or '', + return parse.SplitResult(scheme=(scheme or 'http').lower(), netloc=netloc or '', path=path or '', query='',