#2888: Changes to Applications and Services previously missed.

tests/integration_tests/extensions/applications/extended_application.py

@@ -31,6 +31,8 @@ class ExtendedApplication(Application, identifier="ExtendedApplication"):
     The application requests and loads web pages using its domain name and requesting IP addresses using DNS.
     """
 
+    config: "ExtendedApplication.ConfigSchema"
+
     target_url: Optional[str] = None
 
     domain_name_ip_address: Optional[IPv4Address] = None
@@ -42,6 +44,11 @@ class ExtendedApplication(Application, identifier="ExtendedApplication"):
     history: List["BrowserHistoryItem"] = []
     """Keep a log of visited websites and information about the visit, such as response code."""
 
+    class ConfigSchema(Application.ConfigSchema):
+        """ConfigSchema for ExtendedApplication."""
+
+        type: str = "EXTENDEDAPPLICATION"
+
     def __init__(self, **kwargs):
         kwargs["name"] = "ExtendedApplication"
         kwargs["protocol"] = PROTOCOL_LOOKUP["TCP"]

tests/integration_tests/network/test_broadcast.py

@@ -14,9 +14,16 @@ from primaite.utils.validation.ip_protocol import PROTOCOL_LOOKUP
 from primaite.utils.validation.port import PORT_LOOKUP
 
 
-class BroadcastTestService(Service):
+class BroadcastTestService(Service, identifier="BroadcastTestService"):
     """A service for sending broadcast and unicast messages over a network."""
 
+    config: "BroadcastTestService.ConfigSchema"
+
+    class ConfigSchema(Service.ConfigSchema):
+        """ConfigSchema for BroadcastTestService."""
+
+        type: str = "BROADCASTTESTSERVICE"
+
     def __init__(self, **kwargs):
         # Set default service properties for broadcasting
         kwargs["name"] = "BroadcastService"