Continue to rewrite poller unittest