Replaced the uses of "mock" with the standard library's unittest.mock