JezK
Edit File: threaded_authentication_test.rb
$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib')) require 'mongo' require 'thread' require 'test/unit' require File.expand_path("../../test_helper", __FILE__) # NOTE: This test requires bouncing the server. # It also requires that a user exists on the admin database. class AuthenticationTest < Test::Unit::TestCase include Mongo def setup @conn = standard_connection(:pool_size => 10) @db1 = @conn.db('mongo-ruby-test-auth1') @db2 = @conn.db('mongo-ruby-test-auth2') @admin = @conn.db('admin') end def teardown @db1.authenticate('user1', 'secret') @db2.authenticate('user2', 'secret') @conn.drop_database('mongo-ruby-test-auth1') @conn.drop_database('mongo-ruby-test-auth2') end def threaded_exec threads = [] 100.times do threads << Thread.new do yield end end 100.times do |n| threads[n].join end end def test_authenticate @admin.authenticate('bob', 'secret') @db1.add_user('user1', 'secret') @db2.add_user('user2', 'secret') @admin.logout threaded_exec do assert_raise Mongo::OperationFailure do @db1['stuff'].insert({:a => 2}, :safe => true) end end threaded_exec do assert_raise Mongo::OperationFailure do @db2['stuff'].insert({:a => 2}, :safe => true) end end @db1.authenticate('user1', 'secret') @db2.authenticate('user2', 'secret') threaded_exec do assert @db1['stuff'].insert({:a => 2}, :safe => true) end threaded_exec do assert @db2['stuff'].insert({:a => 2}, :safe => true) end puts "Please bounce the server." gets # Here we reconnect. begin @db1['stuff'].find.to_a rescue Mongo::ConnectionFailure end threaded_exec do assert @db1['stuff'].insert({:a => 2}, :safe => true) end threaded_exec do assert @db2['stuff'].insert({:a => 2}, :safe => true) end @db1.logout threaded_exec do assert_raise Mongo::OperationFailure do @db1['stuff'].insert({:a => 2}, :safe => true) end end @db2.logout threaded_exec do assert_raise Mongo::OperationFailure do assert @db2['stuff'].insert({:a => 2}, :safe => true) end end end end