Use unittest and mock instead of testtools