add option to use custom image for the demo alert (#1101)

src/robusta/cli/main.py

@@ -1,20 +1,20 @@
 import base64
 import json
+import os
 import random
 import subprocess
 import time
 import traceback
 import uuid
 from typing import Dict, List, Optional, Union
 
+import certifi
 import typer
 import yaml
 from hikaru.model.rel_1_26 import Container, Job, JobSpec, ObjectMeta, PodSpec, PodTemplateSpec
 from kubernetes import client, config
 from pydantic import BaseModel, Extra
 
-import certifi
-import os
 from robusta._version import __version__
 from robusta.cli.auth import RSAKeyPair
 from robusta.cli.auth import app as auth_commands
@@ -32,6 +32,7 @@
 from robusta.core.sinks.robusta.robusta_sink_params import RobustaSinkConfigWrapper, RobustaSinkParams
 from robusta.core.sinks.slack.slack_sink_params import SlackSinkConfigWrapper, SlackSinkParams
 from robusta.integrations.prometheus.utils import AlertManagerDiscovery
+
 ADDITIONAL_CERTIFICATE: str = os.environ.get("CERTIFICATE", "")
 # TODO - separate shared classes to a separated shared repo, to remove dependencies between the cli and runner
 
@@ -413,6 +414,7 @@ def demo_alert(
         help="Additional alert labels. Comma separated list. For example: env=prod,team=infra ",
     ),
     kube_config: str = typer.Option(None, help="Kube config file path override."),
+    image: str = typer.Option("curlimages/curl", help="Docker image with curl support."),
 ):
     """
     Create a demo alert on AlertManager.
@@ -489,7 +491,7 @@ def demo_alert(
                     containers=[
                         Container(
                             name="alert-curl",
-                            image="curlimages/curl",
+                            image=image,
                             command=command,
                         )
                     ],