arangodb/UnitTests/HttpInterface/avocadodb.rb

require 'httparty'

class AvocadoDB
  include HTTParty

  base_uri 'http://localhost:8529'
  format :json

################################################################################
## create a collection
################################################################################

  def self.create_collection (name, wait_for_sync = true)
    body = "{ \"name\" : \"#{name}\", \"waitForSync\" : #{wait_for_sync} }"

    doc = self.post("/_api/collection", :body => body)

    if doc.code == 409
      return doc.parsed_response['id']
    end

    if doc.code != 200
      return nil
    end

    return doc.parsed_response['id']
  end

################################################################################
## drop a collection
################################################################################

  def self.drop_collection (name)
    cmd = "/_api/collection/#{name}"
    self.delete(cmd)
  end

################################################################################
## size of a collection
################################################################################

  def self.size_collection (name)
    doc = self.get("/_api/collection/#{name}/count") # TODO use api call

    return doc.parsed_response['count']
  end

################################################################################
## issues a get
################################################################################

  def self.log_get (output, url, args = {})
    doc = self.get(url, args);
    self.log(:method => :get, :url => url, :body => args[:body], :headers => args[:headers], :result => doc, :output => output);
    return doc
  end

################################################################################
## issues a head
################################################################################

  def self.log_head (output, url, args = {})
    doc = self.head(url, args);
    self.log(:method => :head, :url => url, :body => args[:body], :headers => args[:headers], :result => doc, :output => output);
    return doc
  end

################################################################################
## issues a post
################################################################################

  def self.log_post (output, url, args = {})
    doc = self.post(url, args);
    self.log(:method => :post, :url => url, :body => args[:body], :headers => args[:headers], :result => doc, :output => output);
    return doc
  end

################################################################################
## issues a put
################################################################################

  def self.log_put (output, url, args = {})
    doc = self.put(url, args);
    self.log(:method => :put, :url => url, :body => args[:body], :headers => args[:headers], :result => doc, :output => output);
    return doc
  end

################################################################################
## issues a delete
################################################################################

  def self.log_delete (output, url, args = {})
    doc = self.delete(url, args);
    self.log(:method => :delete, :url => url, :body => args[:body], :headers => args[:headers], :result => doc, :output => output);
    return doc
  end

################################################################################
## generate log file
################################################################################

  def self.log (args)
    if args.key?(:output)
      logfile = File.new("logs/#{args[:output]}", "a")
    else
      logfile = File.new("output.log", "a")
    end

    method = args[:method] || :get
    url = args[:url]
    body = args[:body]
    headers = args[:headers]
    result = args[:result]
    response = result.parsed_response

    logfile.puts '-' * 80

    h_option = ""
    h_sep = ""

    if headers
      for k in [ "if-match", "if-none-match" ] do
	if headers.key?(k)
	  h_option = h_option + h_sep + "'-H #{k}: #{headers[k]}'"
	  h_sep = " "
	end
      end
      h_option = h_option + h_sep
    end

    if method == :get
      logfile.puts "> curl -X GET #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts
    elsif method == :head
      logfile.puts "> curl -X HEAD #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts
    elsif method == :delete
      logfile.puts "> curl -X DELETE #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts
    elsif method == :post
      if body == nil
	logfile.puts "> curl -X POST #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts
      else
	logfile.puts "> curl --data @- -X POST #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts body
	logfile.puts
      end
    elsif method == :put
      if body == nil
	logfile.puts "> curl -X PUT #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts
      else
	logfile.puts "> curl --data @- -X PUT #{h_option}--dump - http://localhost:8529#{url}"
	logfile.puts body
	logfile.puts
      end
    else
      logfile.puts "MISSING"
    end

    logfile.puts "HTTP/1.1 #{result.code} #{result.message}"

    if result.headers.key?('content-type')
      logfile.puts "content-type: #{result.headers['content-type']}"
    end

    if result.headers.key?('location')
      logfile.puts "location: #{result.headers['location']}"
    end

    if result.headers.key?('etag')
      logfile.puts "etag: #{result.headers['etag']}"
    end

    if response != nil
      logfile.puts
      logfile.puts JSON.pretty_generate(response)
    end

    logfile.close
  end
end