move mock to it's own pkg