From eb3daf84f1b099b8521b172974de95ac017f00a6 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 4 Oct 2022 12:13:04 -0400 Subject: [PATCH] Common: Use strings for NetworkService Enum values --- monkey/common/types.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/monkey/common/types.py b/monkey/common/types.py index 00b77ca10..55ebb4620 100644 --- a/monkey/common/types.py +++ b/monkey/common/types.py @@ -1,7 +1,7 @@ from __future__ import annotations from dataclasses import dataclass -from enum import Enum, auto +from enum import Enum from ipaddress import IPv4Address from typing import Dict, List, Optional, Union from uuid import UUID @@ -29,7 +29,14 @@ JSONSerializable = Union[ # type: ignore[misc] class NetworkService(Enum): - UNKNOWN = auto() + """ + An Enum representing network services + + This Enum represents all network services that Infection Monkey supports. The value of each + member is the member's name in all lower-case characters. + """ + + UNKNOWN = "unknown" class NetworkPort(ConstrainedInt):