Merge pull request #350 from UffizziCloud/feature/#339_sleeping_mode

[339] added cluster sleep/wake commands

lib/uffizzi/cli/cluster.rb

@@ -22,7 +22,7 @@ def list
       run('list')
     end
 
-    desc 'create [NAME]', 'Create a cluster'
+    desc 'create [CLUSTER_NAME]', 'Create a cluster'
     method_option :name, type: :string, required: false, aliases: '-n'
     method_option :kubeconfig, type: :string, required: false, aliases: '-k'
     method_option :manifest, type: :string, required: false, aliases: '-m'
@@ -34,13 +34,13 @@ def create(name = nil)
       run('create', { name: name })
     end
 
-    desc 'describe [NAME]', 'Describe a cluster'
+    desc 'describe [CLUSTER_NAME]', 'Describe a cluster'
     method_option :output, required: false, type: :string, aliases: '-o', enum: ['json', 'pretty-json']
     def describe(name)
       run('describe', cluster_name: name)
     end
 
-    desc 'delete [NAME]', 'Delete a cluster'
+    desc 'delete [CLUSTER_NAME]', 'Delete a cluster'
     method_option :'delete-config', required: false, type: :boolean, default: true
     def delete(name)
       run('delete', cluster_name: name)
@@ -61,6 +61,16 @@ def disconnect
       run('disconnect')
     end
 
+    desc 'sleep [CLUSTER_NAME]', 'Scales a Uffizzi cluster down to zero resource utilization'
+    def sleep(name = nil)
+      run('sleep', cluster_name: name)
+    end
+
+    desc 'wake [CLUSTER_NAME]', 'Scales up a Uffizzi cluster to its original resource'
+    def wake(name = nil)
+      run('wake', cluster_name: name)
+    end
+
     private
 
     def run(command, command_args = {})
@@ -81,6 +91,10 @@ def run(command, command_args = {})
         handle_update_kubeconfig_command(project_slug, command_args)
       when 'disconnect'
         ClusterDisconnectService.handle(options)
+      when 'sleep'
+        handle_sleep_command(project_slug, command_args)
+      when 'wake'
+        handle_wake_command(project_slug, command_args)
       end
     end
 
@@ -131,7 +145,7 @@ def handle_create_command(project_slug, command_args)
 
       if ClusterService.failed?(cluster_data[:state])
         spinner.error
-        Uffizzi.ui.say_error_and_exit("Cluster with name: #{cluster_name} failed to be created.")
+        Uffizzi.ui.say_error_and_exit("Cluster #{cluster_name} failed to be created.")
       end
 
       spinner.success
@@ -196,7 +210,8 @@ def handle_delete_cluster(project_slug, cluster_name)
 
     def handle_update_kubeconfig_command(project_slug, command_args)
       kubeconfig_path = options[:kubeconfig] || KubeconfigService.default_path
-      cluster_data = fetch_cluster_data(project_slug, command_args[:cluster_name])
+      cluster_name = command_args[:cluster_name]
+      cluster_data = fetch_cluster_data(project_slug, cluster_name)
 
       unless cluster_data[:kubeconfig].present?
         say_error_update_kubeconfig(cluster_data)
@@ -218,13 +233,48 @@ def handle_update_kubeconfig_command(project_slug, command_args)
         new_kubeconfig
       end
 
-      update_clusters_config(cluster_data[:id], kubeconfig_path: kubeconfig_path)
+      update_clusters_config(cluster_data[:id], name: cluster_name, kubeconfig_path: kubeconfig_path)
 
       return if options[:quiet]
 
       Uffizzi.ui.say("Kubeconfig was updated by the path: #{kubeconfig_path}")
     end
 
+    def handle_sleep_command(project_slug, command_args)
+      cluster_name = command_args[:cluster_name] || ConfigFile.read_option(:current_cluster)&.fetch(:name)
+      return handle_missing_cluster_name_error if cluster_name.nil?
+
+      response = scale_down_cluster(ConfigFile.read_option(:server), project_slug, cluster_name)
+      return ResponseHelper.handle_failed_response(response) unless ResponseHelper.ok?(response)
+
+      spinner = TTY::Spinner.new("[:spinner] Scaling down cluster #{cluster_name}...", format: :dots)
+      spinner.auto_spin
+      ClusterService.wait_cluster_scale_down(project_slug, cluster_name)
+
+      spinner.success
+      Uffizzi.ui.say("Cluster #{cluster_name} was successfully scaled down")
+    end
+
+    def handle_wake_command(project_slug, command_args)
+      cluster_name = command_args[:cluster_name] || ConfigFile.read_option(:current_cluster)&.fetch(:name)
+      return handle_missing_cluster_name_error if cluster_name.nil?
+
+      response = scale_up_cluster(ConfigFile.read_option(:server), project_slug, cluster_name)
+      return ResponseHelper.handle_failed_response(response) unless ResponseHelper.ok?(response)
+
+      spinner = TTY::Spinner.new("[:spinner] Waking up cluster #{cluster_name}...", format: :dots)
+      spinner.auto_spin
+      cluster_data = ClusterService.wait_cluster_scale_up(project_slug, cluster_name)
+
+      if ClusterService.failed_scaling_up?(cluster_data[:state])
+        spinner.error
+        Uffizzi.ui.say_error_and_exit("Failed to wake up cluster #{cluster_name}.")
+      end
+
+      spinner.success
+      Uffizzi.ui.say("Cluster #{cluster_name} was successfully scaled up")
+    end
+
     def say_error_update_kubeconfig(cluster_data)
       if ClusterService.failed?(cluster_data[:state])
         Uffizzi.ui.say_error_and_exit('Kubeconfig is empty because cluster failed to be created.')
@@ -321,6 +371,7 @@ def handle_succeed_create_response(cluster_data)
       is_update_current_context = options[:'update-current-context']
       parsed_kubeconfig = parse_kubeconfig(cluster_data[:kubeconfig])
       rendered_cluster_data = render_cluster_data(cluster_data)
+      cluster_name = cluster_data[:name]
 
       Uffizzi.ui.enable_stdout
       Uffizzi.ui.say("Cluster with name: #{rendered_cluster_data[:name]} was created.")
@@ -332,7 +383,7 @@ def handle_succeed_create_response(cluster_data)
       Uffizzi.ui.say(rendered_cluster_data) if Uffizzi.ui.output_format
 
       save_kubeconfig(parsed_kubeconfig, kubeconfig_path)
-      update_clusters_config(cluster_data[:id], kubeconfig_path: kubeconfig_path)
+      update_clusters_config(cluster_data[:id], name: cluster_name, kubeconfig_path: kubeconfig_path)
       GithubService.write_to_github_env(rendered_cluster_data) if GithubService.github_actions_exists?
     end
 
@@ -362,6 +413,7 @@ def save_kubeconfig(kubeconfig, kubeconfig_path)
     def update_clusters_config(id, params)
       clusters_config = Uffizzi::ConfigHelper.update_clusters_config_by_id(id, params)
       ConfigFile.write_option(:clusters, clusters_config)
+      ConfigFile.write_option(:current_cluster, ConfigHelper.cluster_config_by_id(id))
     end
 
     def render_cluster_data(cluster_data)
@@ -400,5 +452,11 @@ def save_previous_current_context(kubeconfig_path, current_context)
       previous_current_contexts = Uffizzi::ConfigHelper.set_previous_current_context_by_path(kubeconfig_path, current_context)
       ConfigFile.write_option(:previous_current_contexts, previous_current_contexts)
     end
+
+    def handle_missing_cluster_name_error
+      Uffizzi.ui.say("No kubeconfig found at #{KubeconfigService.default_path}")
+      Uffizzi.ui.say('Please update the current context or provide a cluster name.')
+      Uffizzi.ui.say('$uffizzi cluster sleep my-cluster')
+    end
   end
 end

lib/uffizzi/cli/dev.rb

@@ -81,11 +81,12 @@ def wait_cluster_creation(cluster_name)
     def handle_succeed_cluster_creation(cluster_data)
       kubeconfig_path = options[:kubeconfig] || KubeconfigService.default_path
       parsed_kubeconfig = parse_kubeconfig(cluster_data[:kubeconfig])
+      cluster_name = cluster_data[:name]
 
-      Uffizzi.ui.say("Cluster with name: #{cluster_data[:name]} was created.")
+      Uffizzi.ui.say("Cluster with name: #{cluster_name} was created.")
 
       save_kubeconfig(parsed_kubeconfig, kubeconfig_path)
-      update_clusters_config(cluster_data[:id], kubeconfig_path: kubeconfig_path)
+      update_clusters_config(cluster_data[:id], name: cluster_name, kubeconfig_path: kubeconfig_path)
     end
 
     def save_kubeconfig(kubeconfig, kubeconfig_path)

lib/uffizzi/clients/api/api_client.rb

@@ -262,6 +262,20 @@ def create_cluster(server, project_slug, params)
     build_response(response)
   end
 
+  def scale_down_cluster(server, project_slug, cluster_name)
+    uri = scale_down_cluster_uri(server, project_slug, cluster_name)
+    response = http_client.make_put_request(uri)
+
+    build_response(response)
+  end
+
+  def scale_up_cluster(server, project_slug, cluster_name)
+    uri = scale_up_cluster_uri(server, project_slug, cluster_name)
+    response = http_client.make_put_request(uri)
+
+    build_response(response)
+  end
+
   def create_access_token(server, session_id)
     uri = access_tokens_url(server)
 

lib/uffizzi/clients/api/api_routes.rb

@@ -110,6 +110,14 @@ def cluster_uri(server, project_slug, cluster_name:, oidc_token:)
     "#{server}/api/cli/v1/projects/#{project_slug}/clusters/#{cluster_name}?token=#{oidc_token}"
   end
 
+  def scale_up_cluster_uri(server, project_slug, cluster_name)
+    "#{server}/api/cli/v1/projects/#{project_slug}/clusters/#{cluster_name}/scale_up"
+  end
+
+  def scale_down_cluster_uri(server, project_slug, cluster_name)
+    "#{server}/api/cli/v1/projects/#{project_slug}/clusters/#{cluster_name}/scale_down"
+  end
+
   def access_token_url(server, code)
     "#{server}/api/cli/v1/access_tokens/#{code}"
   end

lib/uffizzi/helpers/config_helper.rb

@@ -2,7 +2,7 @@
 
 module Uffizzi
   module ConfigHelper
-    CLUSTER_PARAMS = [:kubeconfig_path].freeze
+    CLUSTER_PARAMS = [:kubeconfig_path, :name].freeze
 
     class ConfigParamsError < StandardError
       def initialize(unavailable_params, key)

lib/uffizzi/services/cluster_service.rb

@@ -6,6 +6,9 @@ class ClusterService
   CLUSTER_STATE_DEPLOYING_NAMESPACE = 'deploying_namespace'
   CLUSTER_STATE_DEPLOYING = 'deploying'
   CLUSTER_STATE_DEPLOYED = 'deployed'
+  CLUSTER_STATE_SCALING_DOWN = 'scaling_down'
+  CLUSTER_STATE_SCALING_UP = 'scaling_up'
+  CLUSTER_FAILED_SCALING_UP = 'failed_scaling_up'
   CLUSTER_STATE_FAILED_DEPLOY_NAMESPACE = 'failed_deploy_namespace'
   CLUSTER_STATE_FAILED = 'failed'
   CLUSTER_NAME_MAX_LENGTH = 15
@@ -26,6 +29,18 @@ def failed?(cluster_state)
       [CLUSTER_STATE_FAILED_DEPLOY_NAMESPACE, CLUSTER_STATE_FAILED].include?(cluster_state)
     end
 
+    def scaling_up?(cluster_state)
+      cluster_state === CLUSTER_STATE_SCALING_UP
+    end
+
+    def scaling_down?(cluster_state)
+      cluster_state === CLUSTER_STATE_SCALING_DOWN
+    end
+
+    def failed_scaling_up?(cluster_state)
+      cluster_state === CLUSTER_FAILED_SCALING_UP
+    end
+
     def wait_cluster_deploy(project_slug, cluster_name, oidc_token)
       loop do
         params = {
@@ -43,6 +58,38 @@ def wait_cluster_deploy(project_slug, cluster_name, oidc_token)
       end
     end
 
+    def wait_cluster_scale_up(project_slug, cluster_name)
+      loop do
+        params = {
+          cluster_name: cluster_name,
+        }
+        response = get_cluster(Uffizzi::ConfigFile.read_option(:server), project_slug, params)
+        return Uffizzi::ResponseHelper.handle_failed_response(response) unless Uffizzi::ResponseHelper.ok?(response)
+
+        cluster_data = response.dig(:body, :cluster)
+
+        return cluster_data unless scaling_up?(cluster_data[:state])
+
+        sleep(5)
+      end
+    end
+
+    def wait_cluster_scale_down(project_slug, cluster_name)
+      loop do
+        params = {
+          cluster_name: cluster_name,
+        }
+        response = get_cluster(Uffizzi::ConfigFile.read_option(:server), project_slug, params)
+        return Uffizzi::ResponseHelper.handle_failed_response(response) unless Uffizzi::ResponseHelper.ok?(response)
+
+        cluster_data = response.dig(:body, :cluster)
+
+        return unless scaling_down?(cluster_data[:state])
+
+        sleep(3)
+      end
+    end
+
     def generate_name
       name = Faker::Internet.domain_word[0..CLUSTER_NAME_MAX_LENGTH]
 

man/uffizzi-cluster-sleep.ronn

@@ -0,0 +1,30 @@
+$ uffizzi cluster sleep -h
+uffizzi-cluster-sleep - put a cluster to sleep (non-destructive)
+================================================================
+
+## SYNOPSIS
+    uffizzi cluster sleep [CLUSTER_NAME]
+
+## DESCRIPTION
+    Scales a Uffizzi cluster down to zero resource utilization 
+    while keeping the namespace and any stateful resources,
+    like persistent volume claims. If no CLUSTER_NAME is 
+    specified, the kubeconfig current context is used.
+
+    For more information on Uffizzi clusters, see:
+    https://docs.uffizzi.com/references/cli/
+
+## OPTIONS
+    CLUSTER_NAME
+    The name of the target Uffizzi cluster
+
+## EXAMPLES
+    To put the Uffizzi cluster in the current context to 
+    sleep, run:
+
+        $ uffizzi cluster sleep
+
+    To put a Uffizzi cluster outside the current context to 
+    sleep, run:
+
+        $ uffizzi cluster sleep my-cluster

man/uffizzi-cluster-wake.ronn

@@ -0,0 +1,28 @@
+$ uffizzi cluster wake -h
+uffizzi-cluster-wake - wake a cluster that is sleeping
+================================================================
+
+## SYNOPSIS
+    uffizzi cluster wake [CLUSTER_NAME]
+
+## DESCRIPTION
+    Scales up a Uffizzi cluster to its original resource 
+    utilization from zero (see 'uffizzi cluster sleep -h'). 
+    If no CLUSTER_NAME is specified, the kubeconfig current 
+    context is used.
+
+    For more information on Uffizzi clusters, see:
+    https://docs.uffizzi.com/references/cli/
+
+## OPTIONS
+    CLUSTER_NAME
+    The name of the target Uffizzi cluster
+
+## EXAMPLES
+    To wake the Uffizzi cluster in the current context, run:
+
+        $ uffizzi cluster wake
+
+    To wake a Uffizzi cluster outside the current context, run:
+
+        $ uffizzi cluster wake my-cluster

man/uffizzi-cluster.ronn

@@ -25,6 +25,12 @@ uffizzi-cluster - manage clusters
      list
         List all clusters
 
+     sleep
+        Put a cluster to sleep (non-destructive)
+
+     wake
+        Wake a cluster that is sleeping
+
      update-kubeconfig
         Update kubeconfig file
 

test/fixtures/files/uffizzi/uffizzi_cluster_scaled_down.json

@@ -0,0 +1,10 @@
+{
+  "cluster": {
+    "id": 1,
+    "name": "uffizzi-test-cluster-vcluster",
+    "state": "scaled_down",
+    "created_at": "2023-07-15T15:33:41.590Z",
+    "host": "some-host.cluster.com",
+    "kubeconfig": "YXBpVmVyc2lvbjogdjEKY2x1c3RlcnM6Ci0gY2x1c3RlcjoKICAgIGNlcnRpZmljYXRlLWF1dGhvcml0eS1kYXRhOiBMUzB0TFMxQ1JVZEpUaUJEUlZKVVNVWkpRMEZVUlMwdExTMHRDazFKU1VKa2VrTkRRVkl5WjBGM1NVSkJaMGxDUVVSQlMwSm5aM0ZvYTJwUFVGRlJSRUZxUVdwTlUwVjNTSGRaUkZaUlVVUkVRbWh5VFROTmRHTXlWbmtLWkcxV2VVeFhUbWhSUkVVeVQwUlpORTVFUlhkTlJFRjNTR2hqVGsxcVRYZE9ha1V4VFZSUk1VNXFVWGRYYUdOT1RYcE5kMDVxUlhsTlZGRXhUbXBSZHdwWGFrRnFUVk5GZDBoM1dVUldVVkZFUkVKb2NrMHpUWFJqTWxaNVpHMVdlVXhYVG1oUlJFVXlUMFJaTkU1RVJYZE5SRUYzVjFSQlZFSm5ZM0ZvYTJwUENsQlJTVUpDWjJkeGFHdHFUMUJSVFVKQ2QwNURRVUZTZEhac2FqUjJlakl4T0VSMWVFdFNkVlp6Y0VFNVZ6WkpNSHBTTkZweEwxcHVTWEk0T0hsaGJYVUtVMHRZUlRsa1QzbGtlRlZzWVRWeVUyVjBXVkUyTmsxeGFtWkJlVFp4TkhSbFpXWXdjMjFwVkhGYWVtdHZNRWwzVVVSQlQwSm5UbFpJVVRoQ1FXWTRSUXBDUVUxRFFYRlJkMFIzV1VSV1VqQlVRVkZJTDBKQlZYZEJkMFZDTDNwQlpFSm5UbFpJVVRSRlJtZFJWVFpIWWxGU05rbHhNVTFrU2twS2FFdHVkREZVQ25FMU1XWkNhSGQzUTJkWlNVdHZXa2w2YWpCRlFYZEpSRk5CUVhkU1VVbG9RVWs1Y0N0UlRXWjJibU5UZFRnd1JrUnBjblZ6U1ZsUEswNW9aSFZ4TjNNS2QyOHdNRFJWYzJ4NmJYTkhRV2xCZUdKNVZVUTVSMDQzYjJKbVMzWTNTa3g1TDAxMFFUbFhMMDU2T0dFdlVrNDBaSFJaYWxWdk5qYzVRVDA5Q2kwdExTMHRSVTVFSUVORlVsUkpSa2xEUVZSRkxTMHRMUzBLCiAgICBzZXJ2ZXI6IGh0dHBzOi8vbG9jYWxob3N0Ojg0NDMKICBuYW1lOiBteS12Y2x1c3Rlcgpjb250ZXh0czoKLSBjb250ZXh0OgogICAgY2x1c3RlcjogbXktdmNsdXN0ZXIKICAgIG5hbWVzcGFjZTogZGVmYXVsdAogICAgdXNlcjogbXktdmNsdXN0ZXIKICBuYW1lOiBteS12Y2x1c3RlcgpjdXJyZW50LWNvbnRleHQ6IG15LXZjbHVzdGVyCmtpbmQ6IENvbmZpZwpwcmVmZXJlbmNlczoge30KdXNlcnM6Ci0gbmFtZTogbXktdmNsdXN0ZXIKICB1c2VyOgogICAgY2xpZW50LWNlcnRpZmljYXRlLWRhdGE6IExTMHRMUzFDUlVkSlRpQkRSVkpVU1VaSlEwRlVSUzB0TFMwdENrMUpTVUpyYWtORFFWUmxaMEYzU1VKQlowbEpRMkZqWmpCRksySnNTRzkzUTJkWlNVdHZXa2w2YWpCRlFYZEpkMGw2UldoTlFqaEhRVEZWUlVGM2Qxa0tZWHBPZWt4WFRuTmhWMVoxWkVNeGFsbFZRWGhPYW1jeVQwUlJlRTFFUVhkTlFqUllSRlJKZWsxRVdYaE9WRVV3VGxSWk1FMUdiMWhFVkVrd1RVUlplQXBPUkVVd1RsUlpNRTFHYjNkTlJFVllUVUpWUjBFeFZVVkRhRTFQWXpOc2VtUkhWblJQYlRGb1l6TlNiR051VFhoR1ZFRlVRbWRPVmtKQlRWUkVTRTQxQ21NelVteGlWSEJvV2tjeGNHSnFRbHBOUWsxSFFubHhSMU5OTkRsQlowVkhRME54UjFOTk5EbEJkMFZJUVRCSlFVSkVjMHBJTWtsck5XMUlPVkJVVW1VS2NuQmxlbFJHZVZWWk1GaGxRbVJxSzFkTE9HMVZNWE51U21KTWJGbG9ibEJHU201dlpFbG1lVVJ6UzFaVmJ6bHFiRmhTYlZweGFtOWhUMUI0UzNZMFFncGFaVXd2UWpWdGFsTkVRa2ROUVRSSFFURlZaRVIzUlVJdmQxRkZRWGRKUm05RVFWUkNaMDVXU0ZOVlJVUkVRVXRDWjJkeVFtZEZSa0pSWTBSQmFrRm1Da0puVGxaSVUwMUZSMFJCVjJkQ1VVbDBaRUZIUm1SSWVEZzBUbFZRYTBWTllUSTNXRWxIWkdoS1ZFRkxRbWRuY1docmFrOVFVVkZFUVdkT1NrRkVRa2NLUVdsRlFYaHZSVU5EV0haNFJrWnhhSEpaYUZoeVZFMURWVlprVFVkaUsybFVZbkYyU1hKVE1TOVlaRFI2ZUhORFNWRkVXRzV5WmxOd2NsY3phVkJHVVFwa01XMUJiVzlLVkROVGQybHZUVTR4UlZCb00ya3lhMHRCVUdWcWNXYzlQUW90TFMwdExVVk9SQ0JEUlZKVVNVWkpRMEZVUlMwdExTMHRDaTB0TFMwdFFrVkhTVTRnUTBWU1ZFbEdTVU5CVkVVdExTMHRMUXBOU1VsQ1pIcERRMEZTTW1kQmQwbENRV2RKUWtGRVFVdENaMmR4YUd0cVQxQlJVVVJCYWtGcVRWTkZkMGgzV1VSV1VWRkVSRUpvY2swelRYUlpNbmh3Q2xwWE5UQk1WMDVvVVVSRk1rOUVXVFJPUkVWM1RVUkJkMGhvWTA1TmFrMTNUbXBGTVUxVVVURk9hbEYzVjJoalRrMTZUWGRPYWtWNVRWUlJNVTVxVVhjS1YycEJhazFUUlhkSWQxbEVWbEZSUkVSQ2FISk5NMDEwV1RKNGNGcFhOVEJNVjA1b1VVUkZNazlFV1RST1JFVjNUVVJCZDFkVVFWUkNaMk54YUd0cVR3cFFVVWxDUW1kbmNXaHJhazlRVVUxQ1FuZE9RMEZCVVZVM2NGUXllRkp3U2pBd1lVdEVUMXB2WVZaWVZqSXZhR0Z1TDJsRGEzZ3pZM2t3WWpOUmF6aHpDalZTWTBSVlNFNURXVGRPUVZOaEt6VlNRekZNVGpOd1VFTm9hREl5WkRWSGNEWXhiVTVvUzBaYWNtMXNiekJKZDFGRVFVOUNaMDVXU0ZFNFFrRm1PRVVLUWtGTlEwRnhVWGRFZDFsRVZsSXdWRUZSU0M5Q1FWVjNRWGRGUWk5NlFXUkNaMDVXU0ZFMFJVWm5VVlZEVEZoUlFtaFlVamhtVDBSV1JEVkNSRWQwZFFveGVVSnVXVk5WZDBObldVbExiMXBKZW1vd1JVRjNTVVJUUVVGM1VsRkphRUZLVFc5SGRISlllWEF2ZWtwNWNYRTROVkZEV1VaVVpWb3dTVzF3TnpJNENrbFRRMDVSZFNzelQwVmpSMEZwUW1OdWFYZERVMGRWT1c1M09VTmpkRWRvUlZWeFQyWlROUzlhVVRKSU5sUk5RMGxqWmxseFVFVk5XWGM5UFFvdExTMHRMVVZPUkNCRFJWSlVTVVpKUTBGVVJTMHRMUzB0Q2c9PQogICAgY2xpZW50LWtleS1kYXRhOiBMUzB0TFMxQ1JVZEpUaUJGUXlCUVVrbFdRVlJGSUV0RldTMHRMUzB0Q2sxSVkwTkJVVVZGU1VJM1ZGaE1NblpMVVdZNFRVNXhhR3B2VUhKdVJrTkhPVEZyVldSa2NYcGhPQ3M0WkRsSVpERTVVemR2UVc5SFEwTnhSMU5OTkRrS1FYZEZTRzlWVVVSUlowRkZUM2RyWmxscFZHMVpaakE1VGtZMmRXdzNUazFZU2xKcVVtUTBSakpRTlZseWVWcFVWM2xqYkhOMVZtbEhZemhWYldWb01BcG9MMGxQZDNCV1Uyb3lUMVprUjFwdGNVOW9ielF2UlhFdlowWnNOSFk0U0cxUlBUMEtMUzB0TFMxRlRrUWdSVU1nVUZKSlZrRlVSU0JMUlZrdExTMHRMUW89Cg=="
+  }
+}

test/support/uffizzi_stub_support.rb

@@ -200,6 +200,18 @@ def stub_uffizzi_delete_cluster(project_slug)
     stub_request(:delete, uri).to_return(status: 204)
   end
 
+  def stub_uffizzi_scale_down_cluster(body, project_slug, cluster_name:)
+    uri = scale_down_cluster_uri(Uffizzi.configuration.server, project_slug, cluster_name)
+
+    stub_request(:put, uri).to_return(status: 200, body: body.to_json)
+  end
+
+  def stub_uffizzi_scale_up_cluster(body, project_slug, cluster_name:)
+    uri = scale_up_cluster_uri(Uffizzi.configuration.server, project_slug, cluster_name)
+
+    stub_request(:put, uri).to_return(status: 200, body: body.to_json)
+  end
+
   def stub_get_token_request(body)
     uri = %r{#{Uffizzi.configuration.server}/api/cli/v1/access_tokens/([A-Za-z0-9\-_]+)}