move mock dbi code to setup function