Tests for scaleset size validation

src/pytypes/tests/test_models.py

@@ -7,7 +7,7 @@
 
 from pydantic import ValidationError
 
-from onefuzztypes.models import TeamsTemplate
+from onefuzztypes.models import Scaleset, TeamsTemplate
 from onefuzztypes.requests import NotificationCreate
 
 
@@ -28,5 +28,41 @@ def test_model(self) -> None:
             NotificationCreate.parse_obj(missing_container)
 
 
+class TestScaleset(unittest.TestCase):
+    def test_scaleset_size_lt_0(self) -> None:
+        self.assertRaises(
+            ValueError,
+            Scaleset,
+            pool_name="test_pool",
+            vm_sku="Standard_D2ds_v4",
+            image="Canonical:UbuntuServer:18.04-LTS:latest",
+            region="westus2",
+            size=-1,
+            spot_instances=False,
+        )
+
+    def test_scaleset_size_eq_0(self) -> None:
+        scaleset = Scaleset(
+            pool_name="test_pool",
+            vm_sku="Standard_D2ds_v4",
+            image="Canonical:UbuntuServer:18.04-LTS:latest",
+            region="westus2",
+            size=0,
+            spot_instances=False,
+        )
+        self.assertEqual(scaleset.size, 0)
+
+    def test_scaleset_size_gt_0(self) -> None:
+        scaleset = Scaleset(
+            pool_name="test_pool",
+            vm_sku="Standard_D2ds_v4",
+            image="Canonical:UbuntuServer:18.04-LTS:latest",
+            region="westus2",
+            size=80,
+            spot_instances=False,
+        )
+        self.assertEqual(scaleset.size, 80)
+
+
 if __name__ == "__main__":
     unittest.main()