-
Notifications
You must be signed in to change notification settings - Fork 51
/
test_errors.rb
82 lines (70 loc) · 2.73 KB
/
test_errors.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# Author:: Couchbase <info@couchbase.com>
# Copyright:: 2011, 2012 Couchbase, Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
require File.join(File.dirname(__FILE__), 'setup')
require 'digest/md5'
class TestErrors < MiniTest::Unit::TestCase
def setup
@mock = start_mock
end
def teardown
stop_mock(@mock)
end
def genkey(item)
tuple = [item["author"], item["message"]]
Digest::MD5.hexdigest(tuple.join('-'))
end
def test_graceful_add_with_collision
connection = Couchbase.new(:hostname => @mock.host, :port => @mock.port)
msg1 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:09"}
key1 = uniq_id(genkey(msg1))
msg2 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:29:30"}
key2 = uniq_id(genkey(msg2))
connection.add(key1, msg1)
begin
connection.add(key2, msg2)
rescue Couchbase::Error::KeyExists => ex
# using info from exception
# it could be done with cas operation, but we can save one request
# here (in real world cas operation will be more consistent because it
# fetch fresh version from the cluster)
#
# connection.cas(key2) do |msg|
# msg.merge("time" => [msg["time"], msg2["time"]])
# end
msg2 = msg1.merge("time" => [msg1["time"], msg2["time"]])
connection.set(key2, msg2, :cas => ex.cas)
end
msg3 = {"author" => "foo", "message" => "hi all",
"time" => ["2012-01-12 11:29:09", "2012-01-12 11:29:30"]}
key3 = uniq_id(genkey(msg3))
assert_equal msg3, connection.get(key3)
connection.run do |conn|
msg4 = {"author" => "foo", "message" => "hi all", "time" => "2012-01-12 11:45:34"}
key4 = uniq_id(genkey(msg4))
connection.add(key4, msg4) do |ret|
assert_equal :add, ret.operation
assert_equal key4, ret.key
msg4 = msg3.merge("time" => msg3["time"] + [msg4["time"]])
connection.set(ret.key, msg4, :cas => ret.cas)
end
end
msg5 = {"author" => "foo", "message" => "hi all",
"time" => ["2012-01-12 11:29:09", "2012-01-12 11:29:30", "2012-01-12 11:45:34"]}
key5 = uniq_id(genkey(msg5))
assert_equal msg5, connection.get(key5)
end
end