Water companies are under pressure to upgrade infrastructure to handle population growth and climate change, with improvements expected to cost billions.