Contents

require './test/test_helper'

class TestThreading < Test::Unit::TestCase

  include Mongo

  @@db = standard_connection(:pool_size => 1, :timeout => 30).db(MONGO_TEST_DB)
  @@coll = @@db.collection('thread-test-collection')

  def set_up_safe_data
    @@db.drop_collection('duplicate')
    @@db.drop_collection('unique')
    @duplicate = @@db.collection('duplicate')
    @unique    = @@db.collection('unique')

    @duplicate.insert("test" => "insert")
    @duplicate.insert("test" => "update")
    @unique.insert("test" => "insert")
    @unique.insert("test" => "update")
    @unique.create_index("test", :unique => true)
  end

  def test_safe_update
    set_up_safe_data
    threads = []
    100.times do |i|
      threads[i] = Thread.new do
        if i % 2 == 0
          assert_raise Mongo::OperationFailure do
            @unique.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true)
          end
        else
          @duplicate.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true)
        end
      end
    end

    100.times do |i|
      threads[i].join
    end
  end

  def test_safe_insert
    set_up_safe_data
    threads = []
    100.times do |i|
      threads[i] = Thread.new do
        if i % 2 == 0
          assert_raise Mongo::OperationFailure do
            @unique.insert({"test" => "insert"}, :safe => true)
          end
        else
          @duplicate.insert({"test" => "insert"}, :safe => true)
        end
      end
    end

    100.times do |i|
      threads[i].join
    end
  end

  def test_threading
    @@coll.drop
    @@coll = @@db.collection('thread-test-collection')

    1000.times do |i|
      @@coll.insert("x" => i)
    end

    threads = []

    10.times do |i|
      threads[i] = Thread.new do
        sum = 0
        @@coll.find().each do |document|
          sum += document["x"]
        end
        assert_equal 499500, sum
      end
    end

    10.times do |i|
      threads[i].join
    end
  end
end

Version data entries

20 entries across 20 versions & 4 rubygems

Version	Path
vanity-1.7.1	vendor/ruby/1.9.1/gems/mongo-1.3.1/test/threading_test.rb
mongo-1.4.0	test/threading_test.rb
jonbell-mongo-1.3.1.2	test/threading_test.rb
mongo-lyon-1.2.4	test/threading_test.rb
mongo-1.3.1	test/threading_test.rb
mongo-1.3.0	test/threading_test.rb
mongo-1.3.0.rc0	test/threading_test.rb
mongo-1.2.4	test/threading_test.rb
mongo-1.2.3	test/threading_test.rb
mongo-1.2.2	test/threading_test.rb
mongo-1.2.1	test/threading_test.rb
mongo-1.2.0	test/threading_test.rb
mongo-1.2.rc2	test/threading_test.rb
mongo-1.2.rc1	test/threading_test.rb
mongo-1.2.rc0	test/threading_test.rb
mongo-1.1.5	test/threading_test.rb
mongo-1.1.4	test/threading_test.rb
mongo-1.1.3	test/threading_test.rb
mongo-1.1.2	test/threading_test.rb
mongo-1.1.1	test/threading_test.rb