Add polling tests