Add option to mock test.