Add test for keeping work after rpc client timeout