Return `id` assumption for simple polymorphic `Collection#model` factories.

Also, add tests to make sure it doesn't break in the future.
Make new tests more thorough