Reworked unit tests to use a different implementation of the context so that properties of entities can be checked.