From dcb706fa6abbf747fd0911fecc1bfa80d009ae9f Mon Sep 17 00:00:00 2001
From: Noah McGregor Harper <74685766+nharper285@users.noreply.github.com>
Date: Fri, 26 Aug 2022 09:57:51 -0700
Subject: [PATCH] Changing CLI Scaleset `size` parameter to `max_size`. (#2293)

* Changing size parameter to max_size.

* Adding default param for clarity.

* FIxing param

* Setting size to 1.

* Adding comment.

* Trailing white spaces.

* Adding initialized_size param.

* Using initialized_at.

* Changing variable name.
---
 src/cli/onefuzz/api.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/cli/onefuzz/api.py b/src/cli/onefuzz/api.py
index 8b36b85cb..0038855b6 100644
--- a/src/cli/onefuzz/api.py
+++ b/src/cli/onefuzz/api.py
@@ -1393,8 +1393,9 @@ class Scaleset(Endpoint):
     def create(
         self,
         pool_name: primitives.PoolName,
-        size: int,
+        max_size: int,
         *,
+        initial_size: Optional[int] = 1,
         image: Optional[str] = None,
         vm_sku: Optional[str] = "Standard_D2s_v3",
         region: Optional[primitives.Region] = None,
@@ -1423,14 +1424,16 @@ class Scaleset(Endpoint):
 
         auto_scale = requests.AutoScaleOptions(
             min=min_instances,
-            max=size,
-            default=size,
+            max=max_size,
+            default=max_size,
             scale_out_amount=scale_out_amount,
             scale_out_cooldown=scale_out_cooldown,
             scale_in_amount=scale_in_amount,
             scale_in_cooldown=scale_in_cooldown,
         )
 
+        # Setting size=1 so that the scaleset is intialized with only 1 node.
+        # The default and max are defined above
         return self._req_model(
             "POST",
             models.Scaleset,
@@ -1439,7 +1442,7 @@ class Scaleset(Endpoint):
                 vm_sku=vm_sku,
                 image=image,
                 region=region,
-                size=size,
+                size=initial_size,
                 spot_instances=spot_instances,
                 ephemeral_os_disks=ephemeral_os_disks,
                 tags=tags,