Simplify test case by mocking out models

No need to test the models so mock out how they're accessed.