def setUp(self) -> None:
from db.entities.note import note
self.mockDbAdapter = MagicMock()
self.mockDbSession = MagicMock()
self.someNote: note = MagicMock()
self.mockLog = MagicMock()
self.noteRepository: NoteRepository = NoteRepository(self.mockDbAdapter, self.mockLog)
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 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)