Database mock connection attribute was removed

This pull request adds it back.