move test models into helper file