require 'fog/collection'
require 'fog/bluebox/models/flavor'

module Fog
  module Bluebox

    class Mock
      def flavors
        Fog::Bluebox::Flavors.new(:connection => self)
      end
    end

    class Real
      def flavors
        Fog::Bluebox::Flavors.new(:connection => self)
      end
    end

    class Flavors < Fog::Collection

      model Fog::Bluebox::Flavor

      def all
        data = connection.get_flavors.body['flavors']
        load(data)
      end

      def get(product_id)
        response = connection.get_flavor(product_id)
        model.new(response.body)
      rescue Excon::Errors::NotFound
        # Whoops - no image with that id.
        nil
      rescue Excon::Errors::Forbidden
        nil
      end

    end

  end
end