AZ wants legal assurances CA won't take its stored water