Added a reference to the driver mock in the element tests
Merge pull request #538 from stof/refactor_document_tests

Added a reference to the driver mock in the element tests