integration-cli: update comments referencing waitAndAssert()

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>

integration-cli/docker_api_swarm_test.go

@@ -737,7 +737,7 @@ func checkClusterHealth(c *testing.T, cl []*daemon.Daemon, managerCount, workerC
 			info swarm.Info
 		)
 
-		// check info in a waitAndAssert, because if the cluster doesn't have a leader, `info` will return an error
+		// check info in a poll.WaitOn(), because if the cluster doesn't have a leader, `info` will return an error
 		checkInfo := func(c *testing.T) (interface{}, string) {
 			client := d.NewClientT(c)
 			daemonInfo, err := client.Info(context.Background())

integration-cli/docker_cli_service_logs_test.go

@@ -51,7 +51,7 @@ func (s *DockerSwarmSuite) TestServiceLogs(c *testing.T) {
 	}
 }
 
-// countLogLines returns a closure that can be used with waitAndAssert to
+// countLogLines returns a closure that can be used with poll.WaitOn() to
 // verify that a minimum number of expected container log messages have been
 // output.
 func countLogLines(d *daemon.Daemon, name string) func(*testing.T) (interface{}, string) {