def test_getNoteByHostId_WhenProvidedHostId_ReturnsNote(self):
self.mockDbAdapter.session.return_value = self.mockDbSession
self.mockDbSession.query.return_value = mockQueryWithFilterBy(
mockFirstByReturnValue("some-note"))
note = self.noteRepository.getNoteByHostId("some-host-id")
self.assertEqual("some-note", note)
def test_deleteHost_InvokedWithAHostId_DeletesProcess(self):
self.mockDbSession.query.return_value = mockQueryWithFilterBy(
mockFirstByReturnValue(self.mockProcess))
self.hostRepository.deleteHost("some-host-id")
self.mockDbSession.delete.assert_called_once_with(self.mockProcess)
self.mockDbSession.commit.assert_called_once()
def test_toggleHostCheckStatus_WhenHostIsSetToFalse_TogglesToTrue(self):
self.mockProcess.checked = 'False'
self.mockDbSession.query.return_value = mockQueryWithFilterBy(mockFirstByReturnValue(self.mockProcess))
self.hostRepository.toggleHostCheckStatus("some-ip-address")
self.assertEqual('True', self.mockProcess.checked)
self.mockDbSession.add.assert_called_once_with(self.mockProcess)
self.mockDbAdapter.commit.assert_called_once()
def test_storeProcessOutput_WhenProvidedProcessIdDoesNotExist_DoesNotPerformAnyUpdate(self):
self.mockDbAdapter.session.return_value = self.mockDbSession
self.mockDbSession.query.return_value = mockQueryWithFilterBy(mockFirstByReturnValue(False))
self.processRepository.storeProcessOutput("some_non_existent_process_id", "this is some cool output")
self.mockDbSession.add.assert_not_called()
self.mockDbAdapter.commit.assert_not_called()
def test_storeCloseStatus_WhenProvidedProcessId_StoresCloseStatus(self):
self.mockProcess.closed = 'False'
self.mockDbSession.query.return_value = mockQueryWithFilterBy(mockFirstByReturnValue(self.mockProcess))
self.processRepository.storeCloseStatus("some-process-id")
self.assertEqual('True', self.mockProcess.closed)
self.mockDbSession.add.assert_called_once_with(self.mockProcess)
self.mockDbAdapter.commit.assert_called_once()
def test_storeProcessRunningElapsedTime_WhenProvidedProcessId_StoresProcessRunningElapsedTime(self):
self.mockProcess.elapsed = "some-time"
self.mockDbSession.query.return_value = mockQueryWithFilterBy(mockFirstByReturnValue(self.mockProcess))
self.processRepository.storeProcessRunningElapsedTime("some-process-id", "another-time")
self.assertEqual("another-time", self.mockProcess.elapsed)
self.mockDbSession.add.assert_called_once_with(self.mockProcess)
self.mockDbAdapter.commit.assert_called_once()
def test_storeNotes_WhenProvidedHostIdAndNoteAndNoteDoesNotExist_SavesNewNote(
self):
self.mockDbAdapter.session.return_value = self.mockDbSession
self.mockDbSession.query.return_value = mockQueryWithFilterBy(
mockFirstByReturnValue(None))
self.noteRepository.storeNotes("some-host-id", "some-note")
self.mockDbSession.add.assert_called_once()
self.mockDbAdapter.commit.assert_called_once()
def test_getHostInfo_WhenProvidedHostIpAddress_FetchesHostInformation(self):
from db.entities.host import hostObj
expected_host_info: hostObj = MagicMock()
self.mockDbSession.query.return_value = mockQueryWithFilterBy(mockFirstByReturnValue(expected_host_info))
actual_host_info = self.hostRepository.getHostInformation("127.0.0.1")
self.assertEqual(actual_host_info, expected_host_info)
def whenProcessDoesNotFinishGracefully(self, process_status: str):
from db.entities.process import process
from db.entities.processOutput import process_output
expected_process: process = MagicMock()
expected_process.status = process_status
expected_process_output: process_output = MagicMock()
self.mockDbSession.query.return_value = mockQueryWithFilterBy(
mockFirstBySideEffect([expected_process, expected_process_output]))
self.processRepository.storeProcessOutput("some_process_id", "this is some cool output")
self.mockDbSession.add.assert_called_once_with(expected_process_output)
self.mockDbAdapter.commit.assert_called_once()
def mockProcessStatusAndReturnSingle(self, processStatus: str):
self.mockProcess.status = processStatus
self.mockDbSession.query.return_value = mockQueryWithFilterBy(
mockFirstByReturnValue(self.mockProcess))
