Adding support for DB access from unit tests