fog--fog/lib/fog/compute/requests/ecloud/add_node.rb

module Fog
  module Ecloud
    class Compute
      module Shared
        private

        def generate_node_request(node_data)
          builder = Builder::XmlMarkup.new
          builder.CreateNodeServiceRequest(ecloud_xmlns) {
            builder.IpAddress(node_data[:ip_address])
            builder.Name(node_data[:name])
            builder.Port(node_data[:port])
            builder.Enabled(node_data[:enabled])
            builder.Description(node_data[:description])
          }
        end

        def validate_node_data(node_data, configure=false)
          valid_opts = [:name, :port, :enabled, :description, :ip_address]
          if configure
            valid_opts.delete_if { |opt| ![:name, :enabled, :description].include?(opt) }
          end
          unless valid_opts.all? { |opt| node_data.keys.include?(opt) }
            raise ArgumentError.new("Required data missing: #{(valid_opts - node_data.keys).map(&:inspect).join(", ")}")
          end
        end
      end

      class Real
        include Shared

        def add_node(nodes_uri, node_data)
          validate_node_data(node_data)

          request(
            :body     => generate_node_request(node_data),
            :expects  => 200,
            :headers  => {'Content-Type' => 'application/vnd.tmrk.ecloud.nodeService+xml'},
            :method   => 'POST',
            :uri      => nodes_uri,
            :parse    => true
          )
        end
      end

      class Mock
        include Shared

        def add_node(nodes_uri, node_data)
          validate_node_data(node_data)
          if node_collection = mock_data.public_ip_internet_service_node_collection_from_href(ensure_unparsed(nodes_uri))
            new_node = MockPublicIpInternetServiceNode.new(node_data, node_collection)
            node_collection.items << new_node
            mock_it 200, mock_node_service_response(new_node), { 'Content-Type' => 'application/vnd.tmrk.ecloud.nodeService+xml' }
          else
            mock_error 200, "401 Unauthorized"
          end
        end
      end
    end
  end
end