mongo-find_replace 0.18.3

data/test/test_ordered_hash.rb

@@ -0,0 +1,168 @@
+require 'test/test_helper'
+
+class OrderedHashTest < Test::Unit::TestCase
+
+  def setup
+    @oh = OrderedHash.new
+    @oh['c'] = 1
+    @oh['a'] = 2
+    @oh['z'] = 3
+    @ordered_keys = %w(c a z)
+  end
+
+  def test_initialize
+    a = OrderedHash.new
+    a['x'] = 1
+    a['y'] = 2
+
+    b = OrderedHash['x' => 1, 'y' => 2]
+    assert_equal a, b
+  end
+
+  def test_hash_code
+    o = OrderedHash.new
+    o['number'] = 50
+    assert o.hash
+  end
+
+  def test_empty
+    assert_equal [], OrderedHash.new.keys
+  end
+
+  def test_uniq
+    list = []
+    doc  = OrderedHash.new
+    doc['_id']  = 'ab12'
+    doc['name'] = 'test'
+
+    same_doc = OrderedHash.new
+    same_doc['_id']  = 'ab12'
+    same_doc['name'] = 'test'
+    list << doc
+    list << same_doc
+
+    assert_equal 2, list.size
+    assert_equal 1, list.uniq.size
+  end
+
+  def test_equality
+    a = OrderedHash.new
+    a['x'] = 1
+    a['y'] = 2
+
+    b = OrderedHash.new
+    b['y'] = 2
+    b['x'] = 1
+
+    c = OrderedHash.new
+    c['x'] = 1
+    c['y'] = 2
+
+    d = OrderedHash.new
+    d['x'] = 2
+    d['y'] = 3
+
+    e = OrderedHash.new
+    e['z'] = 1
+    e['y'] = 2
+
+    assert_equal a, c
+    assert_not_equal a, b
+    assert_not_equal a, d
+    assert_not_equal a, e
+  end
+
+  def test_order_preserved
+    assert_equal @ordered_keys, @oh.keys
+  end
+
+  def test_order_preserved_after_replace
+    @oh['a'] = 42
+    assert_equal @ordered_keys, @oh.keys
+    @oh['c'] = 'foobar'
+    assert_equal @ordered_keys, @oh.keys
+    @oh['z'] = /huh?/
+    assert_equal @ordered_keys, @oh.keys
+  end
+
+  def test_each
+    keys = []
+    @oh.each { |k, v| keys << k }
+    assert_equal keys, @oh.keys
+
+    @oh['z'] = 42
+    assert_equal keys, @oh.keys
+
+    assert_equal @oh, @oh.each {|k,v|}
+  end
+
+  def test_values
+    assert_equal [1, 2, 3], @oh.values
+  end
+
+  def test_merge
+    other = OrderedHash.new
+    other['f'] = 'foo'
+    noob = @oh.merge(other)
+    assert_equal @ordered_keys + ['f'], noob.keys
+    assert_equal [1, 2, 3, 'foo'], noob.values
+  end
+
+  def test_merge_bang
+    other = OrderedHash.new
+    other['f'] = 'foo'
+    @oh.merge!(other)
+    assert_equal @ordered_keys + ['f'], @oh.keys
+    assert_equal [1, 2, 3, 'foo'], @oh.values
+  end
+
+  def test_merge_bang_with_overlap
+    other = OrderedHash.new
+    other['a'] = 'apple'
+    other['c'] = 'crab'
+    other['f'] = 'foo'
+    @oh.merge!(other)
+    assert_equal @ordered_keys + ['f'], @oh.keys
+    assert_equal ['crab', 'apple', 3, 'foo'], @oh.values
+  end
+
+  def test_merge_bang_with_hash_with_overlap
+    other = Hash.new
+    other['a'] = 'apple'
+    other['c'] = 'crab'
+    other['f'] = 'foo'
+    @oh.merge!(other)
+    assert_equal @ordered_keys + ['f'], @oh.keys
+    assert_equal ['crab', 'apple', 3, 'foo'], @oh.values
+  end
+
+  def test_update
+    other = OrderedHash.new
+    other['f'] = 'foo'
+    noob = @oh.update(other)
+    assert_equal @ordered_keys + ['f'], noob.keys
+    assert_equal [1, 2, 3, 'foo'], noob.values
+  end
+
+  def test_inspect_retains_order
+    assert_equal '{"c"=>1, "a"=>2, "z"=>3}', @oh.inspect
+  end
+
+  def test_clear
+    @oh.clear
+    assert @oh.keys.empty?
+  end
+
+  def test_delete
+    assert @oh.keys.include?('z')
+    @oh.delete('z')
+    assert !@oh.keys.include?('z')
+  end
+
+  def test_delete_if
+    assert @oh.keys.include?('z')
+    @oh.delete_if { |k,v| k == 'z' }
+    assert !@oh.keys.include?('z')
+  end
+
+end

data/test/test_round_trip.rb

@@ -0,0 +1,114 @@
+HERE = File.dirname(__FILE__)
+$LOAD_PATH[0,0] = File.join(HERE, '..', 'lib')
+require 'test/test_helper'
+require 'mongo/util/xml_to_ruby'
+
+# For each xml/bson file in the data subdirectory, we turn the XML into an
+# OrderedHash and then test both Ruby-to-BSON and BSON-to-Ruby translations.
+#
+# There is a whole other project that includes similar tests
+# (http://github.com/mongodb/mongo-qa). If the directory ../../mongo-qa
+# exists, (that is, the top-level dir of mongo-qa is next to the top-level dir
+# of this project), then we find the BSON test files there and use those, too.
+class RoundTripTest < Test::Unit::TestCase
+
+  include Mongo
+
+  @@ruby = nil
+
+  def setup
+    unless @@ruby
+      names = Dir[File.join(HERE, 'data', '*.xml')].collect {|f| File.basename(f).sub(/\.xml$/, '') }
+      @@ruby = {}
+      names.each { |name|
+        File.open(File.join(HERE, 'data', "#{name}.xml")) { |f|
+          @@ruby[name] = XMLToRuby.new.xml_to_ruby(f)
+        }
+      }
+    end
+  end
+
+  def test_dummy
+    assert true
+  end
+
+  def self.create_test_for_round_trip_files_in_dir(dir)
+    names = Dir[File.join(dir, '*.xson')].collect {|f| File.basename(f).sub(/\.xson$/, '') }
+    names.each { |name|
+      eval <<EOS
+def test_#{name}_#{dir.gsub(/[^a-zA-Z0-9_]/, '_')}
+  one_round_trip("#{dir}", "#{name}")
+end
+EOS
+    }
+  end
+
+  # Dynamically generate one test for each test file. This way, if one test
+  # fails the others will still run.
+  create_test_for_round_trip_files_in_dir(File.join(HERE, 'data'))
+  mongo_qa_dir = File.join(HERE, '../..', 'mongo-qa/modules/bson_tests/tests')
+  if File.exist?(mongo_qa_dir)
+    %w(basic_types complex single_types).each { |subdir_name|
+      create_test_for_round_trip_files_in_dir(File.join(mongo_qa_dir, subdir_name))
+    }
+  end
+
+  # Round-trip comparisons of Ruby-to-BSON and back.
+  # * Take the objects that were read from XML
+  # * Turn them into BSON bytes
+  # * Compare that with the BSON files we have
+  # * Turn those BSON bytes back in to Ruby objects
+  # * Turn them back into BSON bytes
+  # * Compare that with the BSON files we have (or the bytes that were already
+  #   generated)
+  def one_round_trip(dir, name)
+    obj = File.open(File.join(dir, "#{name}.xson")) { |f|
+      begin
+        XMLToRuby.new.xml_to_ruby(f)
+      rescue => ex # unsupported type
+        return
+      end
+    }
+
+    File.open(File.join(dir, "#{name}.bson"), 'rb') { |f|
+      # Read the BSON from the file
+      bson = f.read
+
+      # Turn the Ruby object into BSON bytes and compare with the BSON bytes
+      # from the file.
+      bson_from_ruby = BSON.serialize(obj)
+
+      begin
+        assert_equal bson.length, bson_from_ruby.to_s.length
+        assert_equal bson, bson_from_ruby.to_s
+      rescue => ex
+#         File.open(File.join(dir, "#{name}_out_a.bson"), 'wb') { |f| # DEBUG
+#           bson_from_ruby.each { |b| f.putc(b) }
+#         }
+        raise ex
+      end
+
+      # Turn those BSON bytes back into a Ruby object.
+      #
+      # We're passing a nil db to the contructor here, but that's OK because
+      # the BSON DBRef bytes don't contain the db object in any case, and we
+      # don't care what the database is.
+      obj_from_bson = BSON.deserialize(bson_from_ruby)
+      assert_kind_of OrderedHash, obj_from_bson
+
+      # Turn that Ruby object into BSON and compare it to the original BSON
+      # bytes.
+      bson_from_ruby = BSON.serialize(obj_from_bson)
+      begin
+        assert_equal bson.length, bson_from_ruby.to_s.length
+        assert_equal bson, bson_from_ruby.to_s
+      rescue => ex
+#         File.open(File.join(dir, "#{name}_out_b.bson"), 'wb') { |f| # DEBUG
+#           bson_from_ruby.each { |b| f.putc(b) }
+#         }
+        raise ex
+      end
+    }
+  end
+
+end

data/test/test_slave_connection.rb

@@ -0,0 +1,36 @@
+require 'test/test_helper'
+
+# NOTE: these tests are run only if we can connect to a single MongoDB in slave mode.
+class SlaveConnectionTest < Test::Unit::TestCase
+  include Mongo
+
+  def self.connect_to_slave
+    @@host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+    @@port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+    conn = Connection.new(@@host, @@port, :slave_ok => true)
+    cmd = conn['admin'].command(:ismaster => 1)
+    cmd['ok'] == 1 && cmd['ismaster'] != 1
+  end
+
+  if self.connect_to_slave
+    puts "Connected to slave; running slave tests."
+
+    def test_connect_to_slave
+      assert_raise Mongo::ConfigurationError do 
+        @db = Connection.new(@@host, @@port, :slave_ok => false).db('ruby-mongo-demo')
+      end
+    end
+
+    def test_slave_ok_sent_to_queries
+      @db   = Connection.new(@@host, @@port, :slave_ok => true).db('ruby-mongo-demo')
+      assert_equal true, @db.slave_ok?
+    end
+  else
+    puts "Not connected to slave; skipping slave connection tests."
+
+    def test_slave_ok_false_on_queries
+      @conn = Connection.new(@@host, @@port)
+      assert !@conn.slave_ok?
+    end
+  end
+end

data/test/test_threading.rb

@@ -0,0 +1,87 @@
+require 'test/test_helper'
+
+class TestThreading < Test::Unit::TestCase
+
+  include Mongo
+
+  @@db = Connection.new('localhost', 27017, :pool_size => 1, :timeout => 30).db('ruby-mongo-test')
+  @@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", 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

data/test/threading/test_threading_large_pool.rb

@@ -0,0 +1,90 @@
+require 'test/test_helper'
+
+# Essentialy the same as test_threading.rb but with an expanded pool for
+# testing multiple connections.
+class TestThreadingLargePool < Test::Unit::TestCase
+
+  include Mongo
+
+  @@db = Connection.new('localhost', 27017, :pool_size => 50, :timeout => 60).db('ruby-mongo-test')
+  @@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", true)
+  end
+
+  def test_safe_update
+    set_up_safe_data
+    threads = []
+    300.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
+
+    300.times do |i|
+      threads[i].join
+    end
+  end
+
+  def test_safe_insert
+    set_up_safe_data
+    threads = []
+    300.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
+
+    300.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