tests: tidy up the imports, in particular import mock from support.

it's a less code, more predicatable and in theory a tiny bit faster as
only one import is attempted for Py2 per test file.