test/functional/cli.rb in mosql-0.1.2 vs test/functional/cli.rb in mosql-0.2.0

@@ -9,10 +9,18 @@
     :meta:
       :table: sqltable
     :columns:
       - _id: TEXT
       - var: INTEGER
+  renameid:
+    :meta:
+      :table: sqltable2
+    :columns:
+      - id:
+        :source: _id
+        :type: TEXT
+      - goats: INTEGER
 EOF
 
   def fake_cli
     # This is a hack. We should refactor cli.rb to be more testable.
     MoSQL::CLI.any_instance.expects(:setup_signal_handlers)
@@ -27,10 +35,11 @@
   before do
     @map = MoSQL::Schema.new(YAML.load(TEST_MAP))
     @adapter = MoSQL::SQLAdapter.new(@map, sql_test_uri)
 
     @sequel.drop_table?(:sqltable)
+    @sequel.drop_table?(:sqltable2)
     @map.create_schema(@sequel)
 
     @cli = fake_cli
   end
 
@@ -70,7 +79,75 @@
                      'op' => 'u',
                      'o2' => { '_id' => o['_id'] },
                      'o'  => { '$set' => { 'var' => 100 } },
                    })
     assert_equal(100, sequel[:sqltable].where(:_id => o['_id'].to_s).select.first[:var])
+  end
+
+  it 'handle "u" ops with $set and a renamed _id' do
+    o = { '_id' => BSON::ObjectId.new, 'goats' => 96 }
+    @adapter.upsert_ns('mosql_test.renameid', o)
+
+    # $set's are currently a bit of a hack where we read the object
+    # from the db, so make sure the new object exists in mongo
+    connect_mongo['mosql_test']['renameid'].insert(o.merge('goats' => 0),
+                                                   :w => 1)
+
+    @cli.handle_op({ 'ns' => 'mosql_test.renameid',
+                     'op' => 'u',
+                     'o2' => { '_id' => o['_id'] },
+                     'o'  => { '$set' => { 'goats' => 0 } },
+                   })
+    assert_equal(0, sequel[:sqltable2].where(:id => o['_id'].to_s).select.first[:goats])
+  end
+
+  it 'handles "d" ops with a renamed id' do
+    o = { '_id' => BSON::ObjectId.new, 'goats' => 1 }
+    @adapter.upsert_ns('mosql_test.renameid', o)
+
+    @cli.handle_op({ 'ns' => 'mosql_test.renameid',
+                     'op' => 'd',
+                     'o' => { '_id' => o['_id'] },
+                   })
+    assert_equal(0, sequel[:sqltable2].where(:id => o['_id'].to_s).count)
+  end
+
+  describe '.bulk_upsert' do
+    it 'inserts multiple rows' do
+      objs = [
+              { '_id' => BSON::ObjectId.new, 'var' => 0 },
+              { '_id' => BSON::ObjectId.new, 'var' => 1 },
+              { '_id' => BSON::ObjectId.new, 'var' => 3 },
+             ].map { |o| @map.transform('mosql_test.collection', o) }
+
+      @cli.bulk_upsert(sequel[:sqltable], 'mosql_test.collection',
+                       objs)
+
+      assert(sequel[:sqltable].where(:_id => objs[0].first, :var => 0).count)
+      assert(sequel[:sqltable].where(:_id => objs[1].first, :var => 1).count)
+      assert(sequel[:sqltable].where(:_id => objs[2].first, :var => 3).count)
+    end
+
+    it 'upserts' do
+      _id = BSON::ObjectId.new
+      objs = [
+              { '_id' => _id, 'var' => 0 },
+              { '_id' => BSON::ObjectId.new, 'var' => 1 },
+              { '_id' => BSON::ObjectId.new, 'var' => 3 },
+             ].map { |o| @map.transform('mosql_test.collection', o) }
+
+      @cli.bulk_upsert(sequel[:sqltable], 'mosql_test.collection',
+                       objs)
+
+      newobjs = [
+                 { '_id' => _id, 'var' => 117 },
+                 { '_id' => BSON::ObjectId.new, 'var' => 32 },
+                ].map { |o| @map.transform('mosql_test.collection', o) }
+      @cli.bulk_upsert(sequel[:sqltable], 'mosql_test.collection',
+                       newobjs)
+
+
+      assert(sequel[:sqltable].where(:_id => newobjs[0].first, :var => 117).count)
+      assert(sequel[:sqltable].where(:_id => newobjs[1].first, :var => 32).count)
+    end
   end
 end