class Aws::Partitions::PartitionList

def add_partition(partition)
  if Partition === partition
    @partitions[partition.name] = partition
  else
    raise ArgumentError, "expected Partition, got #{partition.class}"
  end
end

def build(partitions)
  partitions['partitions'].inject(PartitionList.new) do |list, partition|
    list.add_partition(Partition.build(partition))
    list
  end
end

def clear
  @partitions = {}
end

def each(&block)
  @partitions.each_value(&block)
end

def initialize
  @partitions = {}
end

def partition(partition_name)
  if @partitions.key?(partition_name)
    @partitions[partition_name]
  else
    msg = "invalid partition name #{partition_name.inspect}; valid "
    msg << "partition names include %s" % [@partitions.keys.join(', ')]
    raise ArgumentError, msg
  end
end

def partitions
  @partitions.values
end