#2706 - Correcting whitespace change in database_service.py and actioning some review comments

This commit is contained in:
Charlie Crane
2024-07-30 09:57:47 +01:00
parent e492f19a43
commit bb0ecb93a4
3 changed files with 3 additions and 6 deletions

View File

@@ -7,9 +7,6 @@ from typing import Optional
from primaite.interface.request import RequestResponse
from primaite.simulator.network.protocols.packet import DataPacket
# TODO: Elaborate / Confirm / Validate - See 2709.
# Placeholder implementation for Terminal Class implementation.
class SSHTransportMessage(IntEnum):
"""

View File

@@ -21,7 +21,7 @@ class DatabaseService(Service):
"""
A class for simulating a generic SQL Server service.
This class inherits from the `Service` class and provides methods to simulate a SQL database.
This class inherits from the `Service` class and provides methods to simulate a SQL database.
"""
password: Optional[str] = None

View File

@@ -117,7 +117,7 @@ class Terminal(Service):
def _login(request: List[Any], context: Any) -> RequestResponse:
login = self._process_local_login(username=request[0], password=request[1])
if login == True:
if login:
return RequestResponse(status="success", data={})
else:
return RequestResponse(status="failure", data={})
@@ -140,7 +140,7 @@ class Terminal(Service):
self.parent.UserSessionManager.logoff(self.connection_uuid)
self.disconnect(self.connection_uuid)
return RequestResponse(status="success")
return RequestResponse(status="success", data={})
rm.add_request(
"Login",