def reinitialize(self, db: Database, hostRepository: HostRepository):
self.serviceRepository: ServiceRepository = ServiceRepository(db)
self.processRepository: ProcessRepository = ProcessRepository(db, log)
self.hostRepository: HostRepository = hostRepository
self.portRepository: PortRepository = PortRepository(db)
self.cveRepository: CVERepository = CVERepository(db)
self.noteRepository: NoteRepository = NoteRepository(db, log)
self.scriptRepository: ScriptRepository = ScriptRepository(db)
def setUp(self, get_logger) -> None:
from db.repositories.ProcessRepository import ProcessRepository
self.mockProcess = MagicMock()
self.mockDbSession = MagicMock()
self.mockDbAdapter = MagicMock()
self.mockLogger = MagicMock()
self.mockFilters = MagicMock()
self.mockDbAdapter.session.return_value = self.mockDbSession
self.processRepository = ProcessRepository(self.mockDbAdapter, self.mockLogger)
def buildRepositories(self, database: Database) -> RepositoryContainer:
hostRepository = HostRepository(database)
processRepository = ProcessRepository(database, self.logger)
serviceRepository = ServiceRepository(database)
portRepository: PortRepository = PortRepository(database)
cveRepository: CVERepository = CVERepository(database)
noteRepository: NoteRepository = NoteRepository(database, self.logger)
scriptRepository: ScriptRepository = ScriptRepository(database)
return RepositoryContainer(serviceRepository, processRepository,
hostRepository, portRepository,
cveRepository, noteRepository,
scriptRepository)