tft-cli 0.0.18__py3-none-any.whl → 0.0.20__py3-none-any.whl

tft/cli/commands.py

@@ -6,6 +6,7 @@ import json
 import os
 import re
 import shutil
+import stat
 import subprocess
 import textwrap
 import time
@@ -519,6 +520,7 @@ def request(
     fedora_copr_build: List[str] = OPTION_FEDORA_COPR_BUILD,
     repository: List[str] = OPTION_REPOSITORY,
     repository_file: List[str] = OPTION_REPOSITORY_FILE,
+    sanity: bool = typer.Option(False, help="Run Testing Farm sanity test.", rich_help_panel=RESERVE_PANEL_GENERAL),
     tags: Optional[List[str]] = typer.Option(
         None,
         "-t",
@@ -566,6 +568,16 @@ def request(
             "Only 'x86_64' architecture supported in this case."
         )
 
+    if sanity:
+        if git_url or tmt_plan_name:
+            exit_error(
+                "The option [underline]--sanity[/underline] is mutually exclusive with "
+                "[underline]--git-url[/underline] and [underline]--plan[/underline]."
+            )
+
+        git_url = str(settings.TESTING_FARM_TESTS_GIT_URL)
+        tmt_plan_name = str(settings.TESTING_FARM_SANITY_PLAN)
+
     # resolve git repository details from the current repository
     if not git_url:
         if not git_available:
@@ -1177,6 +1189,28 @@ def reserve(
         if not print_only_request_id:
             console.print(message)
 
+    # Sanity checks for ssh-agent
+
+    # Check of SSH_AUTH_SOCK is defined
+    ssh_auth_sock = os.getenv("SSH_AUTH_SOCK")
+    if not ssh_auth_sock:
+        exit_error("SSH_AUTH_SOCK is not defined, make sure the ssh-agent is running by executing 'eval `ssh-agent`'.")
+
+    # Check if SSH_AUTH_SOCK exists
+    if not os.path.exists(ssh_auth_sock):
+        exit_error(
+            "SSH_AUTH_SOCK socket does not exist, make sure the ssh-agent is running by executing 'eval `ssh-agent`'."
+        )
+
+    # Check if value of SSH_AUTH_SOCK is socket
+    if not stat.S_ISSOCK(os.stat(ssh_auth_sock).st_mode):
+        exit_error("SSH_AUTH_SOCK is not a socket, make sure the ssh-agent is running by executing 'eval `ssh-agent`'.")
+
+    # Check if ssh-add -L is not empty
+    ssh_add_output = subprocess.run(["ssh-add", "-L"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if ssh_add_output.returncode != 0:
+        exit_error("No SSH identities found in the SSH agent. Please run `ssh-add`.")
+
     # check for token
     if not settings.API_TOKEN:
         exit_error("No API token found, export `TESTING_FARM_API_TOKEN` environment variable.")
@@ -1404,10 +1438,27 @@ def reserve(
 
             time.sleep(1)
 
-    console.print(f"🌎 ssh root@{guest}")
+    sshproxy_url = urllib.parse.urljoin(str(settings.API_URL), f"v0.1/sshproxy?api_key={settings.API_TOKEN}")
+    response = session.get(sshproxy_url)
+
+    content = response.json()
+
+    ssh_private_key = ""
+    if content.get('ssh_private_key_base_64'):
+        ssh_private_key = base64.b64decode(content['ssh_private_key_base_64']).decode()
+
+    ssh_proxy_option = f" -J {content['ssh_proxy']}" if content.get('ssh_proxy') else ""
+
+    if ssh_private_key:
+        console.print("🔑 [blue]Adding SSH proxy key[/blue]")
+        subprocess.run(["ssh-add", "-"], input=ssh_private_key.encode())
+
+    console.print(f"🌎 ssh{ssh_proxy_option} root@{guest}")
 
     if autoconnect:
-        os.system(f"ssh -oStrictHostKeyChecking=no -oUserKnownHostsFile=/dev/null root@{guest}")
+        os.system(
+            f"ssh -oStrictHostKeyChecking=no -oUserKnownHostsFile=/dev/null{ssh_proxy_option} root@{guest}"  # noqa: E501
+        )
 
 
 def update():

tft/cli/config.py

@@ -21,4 +21,7 @@ settings = LazySettings(
     DEFAULT_RETRY_BACKOFF_FACTOR=1,
     # system CA certificates path, default for RHEL variants
     REQUESTS_CA_BUNDLE="/etc/ssl/certs/ca-bundle.crt",
+    # Testing Farm sanity test,
+    TESTING_FARM_TESTS_GIT_URL="https://gitlab.com/testing-farm/tests",
+    TESTING_FARM_SANITY_PLAN="/testing-farm/sanity",
 )

{tft_cli-0.0.18.dist-info → tft_cli-0.0.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tft-cli
-Version: 0.0.18
+Version: 0.0.20
 Summary: Testing Farm CLI tool
 License: Apache-2.0
 Author: Miroslav Vadkerti
@@ -16,4 +16,5 @@ Requires-Dist: colorama (>=0.4.4,<0.5.0)
 Requires-Dist: dynaconf (>=3.1.7,<4.0.0)
 Requires-Dist: requests (>=2.27.1,<3.0.0)
 Requires-Dist: ruamel-yaml (>=0.18.6,<0.19.0)
+Requires-Dist: setuptools
 Requires-Dist: typer[all] (>=0.7.0,<0.8.0)

tft_cli-0.0.20.dist-info/RECORD

@@ -0,0 +1,10 @@
+tft/cli/__init__.py,sha256=uEJkNJbqC583PBtNI30kxWdeOr3Wj6zJzIYKf0AD72I,92
+tft/cli/commands.py,sha256=nGzFCuKTFrwRAJwhm7Fd-uPmTqh5ONEBoa-2ubH9Fpg,56699
+tft/cli/config.py,sha256=lJ9TtsBAdcNDbh4xZd0x1b48V7IsGl3t7kALmNjCqNs,1115
+tft/cli/tool.py,sha256=wFcVxe1NRGW8stputOZlKMasZHjpysas7f0sgpEzipQ,865
+tft/cli/utils.py,sha256=9s7zY_k1MYYPTF4Gr2AMH2DMcySUCIgXbF3LjYa7bzY,7404
+tft_cli-0.0.20.dist-info/LICENSE,sha256=YpVAQfXkIyzQAdm5LZkI6L5UWqLppa6O8_tgDSdoabQ,574
+tft_cli-0.0.20.dist-info/METADATA,sha256=dVw-eLgGMMRxiN055vL_nIX59bDxInpZ_XSRsbtLjOI,731
+tft_cli-0.0.20.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+tft_cli-0.0.20.dist-info/entry_points.txt,sha256=xzdebHkH5Bx-YRf-XPMsIoVpvgfUqqcRQGuo8DFkiao,49
+tft_cli-0.0.20.dist-info/RECORD,,

tft_cli-0.0.18.dist-info/RECORD

@@ -1,10 +0,0 @@
-tft/cli/__init__.py,sha256=uEJkNJbqC583PBtNI30kxWdeOr3Wj6zJzIYKf0AD72I,92
-tft/cli/commands.py,sha256=Io1U_zo1ZWtqdfhf4WpXpsLlyBpXRh0XuEEoTop2mIE,54520
-tft/cli/config.py,sha256=rIY8y42pCruDtWAp8eCs08wm7gcWDKxU14Bs4ULkGqg,958
-tft/cli/tool.py,sha256=wFcVxe1NRGW8stputOZlKMasZHjpysas7f0sgpEzipQ,865
-tft/cli/utils.py,sha256=9s7zY_k1MYYPTF4Gr2AMH2DMcySUCIgXbF3LjYa7bzY,7404
-tft_cli-0.0.18.dist-info/LICENSE,sha256=YpVAQfXkIyzQAdm5LZkI6L5UWqLppa6O8_tgDSdoabQ,574
-tft_cli-0.0.18.dist-info/METADATA,sha256=b-bIarMMld3dufVt5pFI114lofPEvQ8ybg7oI5YOTuM,705
-tft_cli-0.0.18.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-tft_cli-0.0.18.dist-info/entry_points.txt,sha256=xzdebHkH5Bx-YRf-XPMsIoVpvgfUqqcRQGuo8DFkiao,49
-tft_cli-0.0.18.dist-info/RECORD,,