# coding: utf-8

require 'rspec'
require 'arangodb.rb'

describe ArangoDB do
  api = "/_api/cursor"
  prefix = "api-cursor"

################################################################################
## query cache
################################################################################

  context "testing the query cache:" do
    before do
      doc = ArangoDB.get("/_api/query-cache/properties")
      @mode = doc.parsed_response['mode']
      ArangoDB.put("/_api/query-cache/properties", :body => "{ \"mode\" : \"demand\" }")

      ArangoDB.delete("/_api/query-cache")
    end

    after do
      ArangoDB.put("/_api/query-cache/properties", :body => "{ \"mode\" : \"#{@mode}\" }")
    end

    it "testing without cache attribute set" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..5 RETURN i\" }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-disabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []

      # should see same result, but not from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end

    it "testing explicitly disabled cache" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..5 RETURN i\", \"cache\" : false }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-disabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []

      # should see same result, but not from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end

    it "testing enabled cache" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..5 RETURN i\", \"cache\" : true }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-enabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      stats = doc.parsed_response['extra']['stats']
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []

      # should see same result, but now from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end

    it "testing clearing the cache" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..5 RETURN i\", \"cache\" : true }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-enabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []

      # should see same result, but now from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []

      # now clear cache
      ArangoDB.delete("/_api/query-cache")

      # query again. now response should not come from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
      stats = doc.parsed_response['extra']['stats']

      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end
    
    it "testing fullCount off" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..10000 LIMIT 5 RETURN i\", \"cache\" : true, \"fullCount\" : false }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-enabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should_not have_key('fullCount')
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
      stats = doc.parsed_response['extra']['stats']

      # should see same result, but now from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should_not have_key('fullCount')
      doc.parsed_response['extra']['stats'].should eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end
    
    it "testing fullCount on" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..10000 LIMIT 5 RETURN i\", \"cache\" : true, \"options\": { \"fullCount\" : true } }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-enabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should have_key('fullCount')
      doc.parsed_response['extra']['stats']['fullCount'].should eq(10000)
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
      stats = doc.parsed_response['extra']['stats']

      # should see same result, but now from cache
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should have_key('fullCount')
      doc.parsed_response['extra']['stats']['fullCount'].should eq(10000)
      doc.parsed_response['extra']['stats'].should eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end
    
    it "testing fullCount on/off" do
      cmd = api
      body = "{ \"query\" : \"FOR i IN 1..10000 LIMIT 5 RETURN i\", \"cache\" : true, \"options\": { \"fullCount\" : true } }"
      doc = ArangoDB.log_post("#{prefix}-query-cache-enabled", cmd, :body => body)
      
      doc.code.should eq(201)
      doc.headers['content-type'].should eq("application/json; charset=utf-8")
      doc.parsed_response['error'].should eq(false)
      doc.parsed_response['code'].should eq(201)
      doc.parsed_response['id'].should be_nil
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should have_key('fullCount')
      doc.parsed_response['extra']['stats']['fullCount'].should eq(10000)
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
      stats = doc.parsed_response['extra']['stats']

      # toggle fullcount value
      body = "{ \"query\" : \"FOR i IN 1..10000 LIMIT 5 RETURN i\", \"cache\" : true, \"options\": { \"fullCount\" : false } }"
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(false)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should_not have_key('fullCount')
      doc.parsed_response['extra']['stats'].should_not eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
      
      # toggle fullcount value yet once more
      body = "{ \"query\" : \"FOR i IN 1..10000 LIMIT 5 RETURN i\", \"cache\" : true, \"options\": { \"fullCount\" : true } }"
      doc = ArangoDB.log_post("#{prefix}-query-cache", cmd, :body => body)
      doc.code.should eq(201)
      result = doc.parsed_response['result']
      result.should eq([ 1, 2, 3, 4, 5 ])
      doc.parsed_response['cached'].should eq(true)
      doc.parsed_response['extra'].should have_key('stats')
      doc.parsed_response['extra']['stats'].should have_key('fullCount')
      doc.parsed_response['extra']['stats'].should eq stats
      doc.parsed_response['extra'].should have_key('warnings')
      doc.parsed_response['extra']['warnings'].should eq []
    end

  end
end