Move mock into actual test.