Add mock with unittest