JezK
Edit File: replication_ack_test.rb
$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib')) require './test/replica_sets/rs_test_helper' class ReplicaSetAckTest < Test::Unit::TestCase def setup ensure_rs @conn = ReplSetConnection.new(build_seeds(1)) @slave1 = Connection.new(@conn.secondary_pools[0].host, @conn.secondary_pools[0].port, :slave_ok => true) assert !@slave1.read_primary? @db = @conn.db(MONGO_TEST_DB) @db.drop_collection("test-sets") @col = @db.collection("test-sets") end def teardown @rs.restart_killed_nodes @conn.close if @conn end def test_safe_mode_with_w_failure assert_raise_error OperationFailure, "timeout" do @col.insert({:foo => 1}, :safe => {:w => 4, :wtimeout => 1, :fsync => true}) end assert_raise_error OperationFailure, "timeout" do @col.update({:foo => 1}, {:foo => 2}, :safe => {:w => 4, :wtimeout => 1, :fsync => true}) end assert_raise_error OperationFailure, "timeout" do @col.remove({:foo => 2}, :safe => {:w => 4, :wtimeout => 1, :fsync => true}) end end def test_safe_mode_replication_ack @col.insert({:baz => "bar"}, :safe => {:w => 3, :wtimeout => 5000}) assert @col.insert({:foo => "0" * 5000}, :safe => {:w => 3, :wtimeout => 5000}) assert_equal 2, @slave1[MONGO_TEST_DB]["test-sets"].count assert @col.update({:baz => "bar"}, {:baz => "foo"}, :safe => {:w => 3, :wtimeout => 5000}) assert @slave1[MONGO_TEST_DB]["test-sets"].find_one({:baz => "foo"}) assert @col.remove({}, :safe => {:w => 3, :wtimeout => 5000}) assert_equal 0, @slave1[MONGO_TEST_DB]["test-sets"].count end def test_last_error_responses 20.times { @col.insert({:baz => "bar"}) } response = @db.get_last_error(:w => 2, :wtimeout => 5000) assert response['ok'] == 1 assert response['lastOp'] @col.update({}, {:baz => "foo"}, :multi => true) response = @db.get_last_error(:w => 2, :wtimeout => 5000) assert response['ok'] == 1 assert response['lastOp'] @col.remove({}) response = @db.get_last_error(:w => 2, :wtimeout => 5000) assert response['ok'] == 1 assert response['n'] == 20 assert response['lastOp'] end end