use unittest.mock instead of mock

[pgajdos]

I see you are still supporting python 2, nevertheless could you please consider to use unittest.mock as a fallback at least?

[a-detiste]

Will provide a PR