As we dropped Python 2 support we don't have to use the third party
mock library with Python 2 compatibility and can move to Python 3
standard library. The change updates the import statements to reflect
that.
Change-Id: I39d9410ae43a2d9e6f6d6eaf74d06a82f2c33524