From fb3071ad4ac8fe5265500b984e4008bbdf28066d Mon Sep 17 00:00:00 2001 From: Anshuman Goel Date: Thu, 15 Oct 2020 09:05:57 -0700 Subject: [PATCH] size validation (#156) --- src/pytypes/onefuzztypes/models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/pytypes/onefuzztypes/models.py b/src/pytypes/onefuzztypes/models.py index 6c82bf714..39399434d 100644 --- a/src/pytypes/onefuzztypes/models.py +++ b/src/pytypes/onefuzztypes/models.py @@ -562,6 +562,12 @@ class Scaleset(BaseModel): client_object_id: Optional[UUID] tags: Dict[str, str] = Field(default_factory=lambda: {}) + @validator("size", allow_reuse=True) + def check_size(cls, value: int) -> int: + if value < 0: + raise ValueError("Invalid scaleset size") + return value + class NotificationConfig(BaseModel): config: NotificationTemplate