This commit refactors the function assertNumConnections to use wait.NoError. Prior to this commit, `make lint` will fail on this function. While fixing it, it's noticed that wait.NoError suits the case so it's refactored to use it.