Add callback tests