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))