Mock Database

Salve io sto creando dei test unit.
Vi spiego un attimo il funzionamento in breve del programma.
C'è un merge astratto che chiama il metodo getEntityById (che viene implementato nelle varie classi che estendono il merge astratto) che tramite un repository effettua una query a db e ritorna una lista di entità(in realtà una sola, altrimenti lancia errori) o null se non è presente, se non è presente crea una nuova entita e fa un certo comportamento, se la trova ne fa un altro.
Ora se io faccio il mock della classe che chiama il merge astratto (e si porta dietro il mock del repository che fa la query) essendo "simulato " il Db tornerà sempre una lista vuota e si comporterà sempre come una nuova entity, non posso simulare e creare dei test per il comportamento in caso di entità trovata, esiste un modo per aggirare il problema?