mirror of https://gitee.com/bigwinds/arangodb
248 lines
6.6 KiB
Ruby
248 lines
6.6 KiB
Ruby
# coding: utf-8
|
|
|
|
require 'rspec'
|
|
require './avocadodb.rb'
|
|
|
|
describe AvocadoDB do
|
|
api = "/_api/index"
|
|
prefix = "api-index-unique-constraint"
|
|
|
|
################################################################################
|
|
## unique constraints during create
|
|
################################################################################
|
|
|
|
context "creating:" do
|
|
context "dealing with unique constraints:" do
|
|
before do
|
|
@cn = "UnitTestsCollectionIndexes"
|
|
AvocadoDB.drop_collection(@cn)
|
|
@cid = AvocadoDB.create_collection(@cn)
|
|
end
|
|
|
|
after do
|
|
AvocadoDB.drop_collection(@cn)
|
|
end
|
|
|
|
it "rolls back in case of violation" do
|
|
cmd = "/_api/index/#{@cid}"
|
|
body = "{ \"type\" : \"hash\", \"unique\" : true, \"fields\" : [ \"a\" ] }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd, :body => body)
|
|
|
|
doc.code.should eq(201)
|
|
doc.parsed_response['type'].should eq("hash")
|
|
doc.parsed_response['unique'].should eq(true)
|
|
|
|
# create a document
|
|
cmd1 = "/document?collection=#{@cid}"
|
|
body = "{ \"a\" : 1, \"b\" : 1 }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd1, :body => body)
|
|
|
|
doc.code.should eq(201)
|
|
|
|
id1 = doc.parsed_response['_id']
|
|
id1.should be_kind_of(String)
|
|
|
|
rev1 = doc.parsed_response['_rev']
|
|
rev1.should be_kind_of(Integer)
|
|
|
|
# check it
|
|
cmd2 = "/document/#{id1}"
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should eq(rev1)
|
|
|
|
# create a unique constraint violation
|
|
body = "{ \"a\" : 1, \"b\" : 2 }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd1, :body => body)
|
|
|
|
doc.code.should eq(409)
|
|
|
|
# check it again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should eq(rev1)
|
|
|
|
# third try (make sure the rollback has not destroyed anything)
|
|
body = "{ \"a\" : 1, \"b\" : 3 }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd1, :body => body)
|
|
|
|
doc.code.should eq(409)
|
|
|
|
# check it again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should eq(rev1)
|
|
|
|
# unload collection
|
|
cmd3 = "/_api/collection/#{@cid}/unload"
|
|
doc = AvocadoDB.log_put("#{prefix}", cmd3)
|
|
|
|
doc.code.should eq(200)
|
|
|
|
cmd3 = "/_api/collection/#{@cid}"
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd3)
|
|
doc.code.should eq(200)
|
|
|
|
while doc.parsed_response['status'] != 2
|
|
doc = AvocadoDB.get(cmd3)
|
|
doc.code.should eq(200)
|
|
end
|
|
|
|
# check it again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should eq(rev1)
|
|
end
|
|
end
|
|
end
|
|
|
|
################################################################################
|
|
## unique constraints during update
|
|
################################################################################
|
|
|
|
context "updating:" do
|
|
context "dealing with unique constraints:" do
|
|
before do
|
|
@cn = "UnitTestsCollectionIndexes"
|
|
AvocadoDB.drop_collection(@cn)
|
|
@cid = AvocadoDB.create_collection(@cn)
|
|
end
|
|
|
|
after do
|
|
AvocadoDB.drop_collection(@cn)
|
|
end
|
|
|
|
it "rolls back in case of violation" do
|
|
cmd = "/_api/index/#{@cid}"
|
|
body = "{ \"type\" : \"hash\", \"unique\" : true, \"fields\" : [ \"a\" ] }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd, :body => body)
|
|
|
|
doc.code.should eq(201)
|
|
doc.parsed_response['type'].should eq("hash")
|
|
doc.parsed_response['unique'].should eq(true)
|
|
|
|
# create a document
|
|
cmd1 = "/document?collection=#{@cid}"
|
|
body = "{ \"a\" : 1, \"b\" : 1 }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd1, :body => body)
|
|
|
|
doc.code.should eq(201)
|
|
|
|
id1 = doc.parsed_response['_id']
|
|
id1.should be_kind_of(String)
|
|
|
|
rev1 = doc.parsed_response['_rev']
|
|
rev1.should be_kind_of(Integer)
|
|
|
|
# check it
|
|
cmd2 = "/document/#{id1}"
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should eq(rev1)
|
|
|
|
# create a second document
|
|
body = "{ \"a\" : 2, \"b\" : 2 }"
|
|
doc = AvocadoDB.log_post("#{prefix}", cmd1, :body => body)
|
|
|
|
doc.code.should eq(201)
|
|
|
|
id2 = doc.parsed_response['_id']
|
|
id2.should be_kind_of(String)
|
|
|
|
rev2 = doc.parsed_response['_rev']
|
|
rev2.should be_kind_of(Integer)
|
|
|
|
# create a unique constraint violation during update
|
|
body = "{ \"a\" : 2, \"b\" : 3 }"
|
|
doc = AvocadoDB.log_put("#{prefix}", cmd2, :body => body)
|
|
|
|
doc.code.should eq(409)
|
|
|
|
# check first document again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should_not eq(rev1)
|
|
|
|
rev3 = doc.parsed_response['_rev']
|
|
rev3.should be_kind_of(Integer)
|
|
|
|
# check second document again
|
|
cmd3 = "/document/#{id2}"
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd3)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(2)
|
|
doc.parsed_response['b'].should eq(2)
|
|
doc.parsed_response['_id'].should eq(id2)
|
|
doc.parsed_response['_rev'].should eq(rev2)
|
|
|
|
# third try (make sure the rollback has not destroyed anything)
|
|
body = "{ \"a\" : 2, \"b\" : 4 }"
|
|
doc = AvocadoDB.log_put("#{prefix}", cmd2, :body => body)
|
|
|
|
doc.code.should eq(409)
|
|
|
|
# check the first document again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should_not eq(rev1)
|
|
doc.parsed_response['_rev'].should_not eq(rev3)
|
|
|
|
# unload collection
|
|
cmd4 = "/_api/collection/#{@cid}/unload"
|
|
doc = AvocadoDB.log_put("#{prefix}", cmd4)
|
|
|
|
doc.code.should eq(200)
|
|
|
|
cmd4 = "/_api/collection/#{@cid}"
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd4)
|
|
doc.code.should eq(200)
|
|
|
|
while doc.parsed_response['status'] != 2
|
|
doc = AvocadoDB.get(cmd4)
|
|
doc.code.should eq(200)
|
|
end
|
|
|
|
# check the first document again
|
|
doc = AvocadoDB.log_get("#{prefix}", cmd2)
|
|
|
|
doc.code.should eq(200)
|
|
doc.parsed_response['a'].should eq(1)
|
|
doc.parsed_response['b'].should eq(1)
|
|
doc.parsed_response['_id'].should eq(id1)
|
|
doc.parsed_response['_rev'].should_not eq(rev1)
|
|
doc.parsed_response['_rev'].should_not eq(rev3)
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|