Added a reference to the driver mock in the element tests

This avoids getting tons of issues reported on Scrutinizer telling that ``expects`` does not exist on the ``DriverInterface``.
It will also make the diff of the 2.0 changes a bit smaller (elements depend on the driver directly in 2.0 so this change was done in it)