monga 0.0.2 → 0.0.3

data/spec/monga/sync/collection_spec.rb

@@ -0,0 +1,247 @@
+require 'spec_helper'
+
+describe Monga::Collection do
+  before do
+    EM.synchrony do
+      @client = Monga::Client.new(type: :sync, pool_size: 10)
+      @db = @client["dbTest"]
+      @collection = @db["testCollection"]
+      @collection.safe_remove
+      docs = []
+      10.times do |i|
+        docs << { artist: "Madonna", title: "Track #{i+1}" }
+        docs << { artist: "Radiohead", title: "Track #{i+1}" }
+      end
+      @collection.safe_insert(docs)
+      EM.stop
+    end
+  end
+
+  # QUERY
+
+  describe "query" do
+    it "should fetch all documents" do
+      EM.synchrony do
+        docs = @collection.find.all
+        docs.size.must_equal 20
+        EM.stop
+      end
+    end
+
+    it "should fetch all docs with skip and limit" do
+      EM.synchrony do
+        docs = @collection.find.skip(10).limit(4).all
+        docs.size.must_equal 4
+        EM.stop
+      end
+    end
+
+    it "should fetch first" do
+      EM.synchrony do
+        doc = @collection.first
+        doc.keys.must_equal ["_id", "artist", "title"]
+        EM.stop
+      end
+    end
+  end
+
+  # INSERT
+
+  describe "insert" do
+    before do
+      EM.synchrony do
+        @collection.safe_ensure_index({ "personal_id" => 1 }, { unique: true, sparse: true })
+        EM.stop
+      end
+    end
+
+    after do
+      EM.synchrony do
+        @collection.drop_index( personal_id: 1 )
+        EM.stop
+      end
+    end
+
+    it "should insert single doc" do
+      EM.synchrony do
+        doc = { name: "Peter", age: 18 }
+        @collection.safe_insert(doc)
+        resp = @collection.find(name: "Peter").all
+        resp.size.must_equal 1
+        resp.first["age"].must_equal 18
+        EM.stop
+      end
+    end
+
+    it "should insert batch of docs" do
+      EM.synchrony do
+        docs = [{ name: "Peter", age: 18 }, {name: "Jhon", age: 18}]
+        @collection.safe_insert(docs)
+        resp = @collection.find(age: 18).all
+        resp.size.must_equal 2
+        EM.stop
+      end
+    end
+
+    it "should fail on uniq index" do
+      EM.synchrony do
+        docs = [{ name: "Peter", age: 18, personal_id: 20 }, {name: "Jhon", age: 18, personal_id: 20}, {name: "Rebeca", age: 21, personal_id: 5}]
+        proc{ @collection.safe_insert(docs) }.must_raise Monga::Exceptions::QueryFailure
+        @collection.count.must_equal 21
+        EM.stop
+      end
+    end
+
+    it "should continue_on_error" do
+      EM.synchrony do
+        docs = [{ name: "Peter", age: 18, personal_id: 20 }, {name: "Jhon", age: 18, personal_id: 20}, {name: "Rebeca", age: 21, personal_id: 5}]
+        proc{ @collection.safe_insert(docs, continue_on_error: true) }.must_raise Monga::Exceptions::QueryFailure
+        @collection.count.must_equal 22
+        EM.stop
+      end
+    end
+  end
+
+  # UPDATE
+
+  describe "update" do
+    it "should make simple update (first matching)" do
+      EM.synchrony do
+        @collection.safe_update({ artist: "Madonna" }, { "$set" => { country: "USA" } })
+        @collection.count( query: { artist: "Madonna", country: "USA" }).must_equal 1
+        EM.stop
+      end
+    end
+
+    it "should create non existing item (upsert)" do
+      EM.synchrony do
+        @collection.safe_update({ artist: "Bjork" }, { "$set" => { country: "Iceland" } }, { upsert: true }) 
+        @collection.count(query: { artist: "Bjork" }).must_equal 1
+        EM.stop
+      end
+    end
+
+    it "should update all matching data (multi_update)" do
+      EM.synchrony do
+        @collection.safe_update({ artist: "Madonna" }, { "$set" => { country: "USA" } }, {multi_update: true})
+        docs = @collection.find(artist: "Madonna").all
+        docs.each{ |d| d["country"].must_equal "USA" }
+        EM.stop
+      end
+    end
+  end
+
+  # REMOVE
+
+  describe "remove" do
+    it "should delete all matching docs" do
+      EM.synchrony do
+        @collection.safe_delete(artist: "Madonna")
+        @collection.count(query: { artist: "Madonna" }).must_equal 0
+        EM.stop
+      end
+    end
+
+    it "should delete first matching doc (single_remove)" do
+      EM.synchrony do
+        @collection.safe_delete({ artist: "Madonna" }, single_remove: true)
+        @collection.count(query: { artist: "Madonna" }).must_equal 9
+        EM.stop
+      end
+    end
+  end
+
+  # COUNT
+
+  describe "count" do
+    it "should count all docs" do
+      EM.synchrony do
+        @collection.count.must_equal 20
+        EM.stop
+      end
+    end
+
+    it "should count all docs with query" do
+      EM.synchrony do
+        @collection.count(query: { artist: "Madonna" }).must_equal 10
+        EM.stop
+      end
+    end
+
+    it "should count all docs with limit" do
+      EM.synchrony do
+        @collection.count(query: { artist: "Madonna" }, limit: 5).must_equal 5
+        EM.stop
+      end
+    end
+
+    it "should count all docs with limit and skip" do
+      EM.synchrony do
+        @collection.count(query: { artist: "Madonna" }, limit: 5, skip: 6).must_equal 4
+        EM.stop
+      end
+    end
+  end
+
+  # ENSURE/DROP INDEX
+
+  describe "ensure_index" do
+    before do
+      EM.synchrony do
+        @collection.drop_indexes
+        EM.stop
+      end
+    end
+
+    it "should create index" do
+      EM.synchrony do
+        @collection.safe_ensure_index(title: 1)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"title" => 1}}.must_equal true
+        EM.stop
+      end
+    end
+
+    it "should create sparse index" do
+      EM.synchrony do
+        @collection.safe_ensure_index({ title: 1 }, sparse: true)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"title" => 1} && doc["sparse"] == true }.must_equal true
+        EM.stop
+      end
+    end
+
+    it "should create unique index" do
+      EM.synchrony do
+        @collection.safe_ensure_index({ some_field: 1 }, unique: true, sparse: true)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"some_field" => 1} && doc["unique"] == true }.must_equal true
+        EM.stop
+      end
+    end
+
+    it "should drop single index" do
+      EM.synchrony do
+        @collection.safe_ensure_index(title: 1)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"title" => 1}}.must_equal true
+        @collection.drop_index(title: 1)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"title" => 1}}.must_equal false
+        EM.stop
+      end
+    end
+
+    it "should drop all indexes (except primary on _id)" do
+      EM.synchrony do
+        @collection.safe_ensure_index(title: 1)
+        docs = @collection.get_indexes
+        docs.any?{ |doc| doc["key"] == {"title" => 1}}.must_equal true
+        @collection.drop_indexes
+        docs = @collection.get_indexes
+        docs.select{ |d| d["ns"] == "dbTest.testCollection" }.size.must_equal 1
+        EM.stop
+      end
+    end
+  end
+end

data/spec/monga/sync/cursor_spec.rb

@@ -0,0 +1,211 @@
+require 'spec_helper'
+
+describe Monga::Cursor do
+  before do
+    EM.synchrony do
+      @client = Monga::Client.new(type: :sync, pool_size: 10)
+      @db = @client["dbTest"]
+      @collection = @db["testCollection"]
+      @collection.safe_remove
+      docs = []
+      10.times do |i|
+        docs << { artist: "Madonna", title: "Track #{i+1}" }
+        docs << { artist: "Radiohead", title: "Track #{i+1}" }
+      end
+      @collection.safe_insert(docs)
+      EM.stop
+    end
+  end
+
+  # ALL
+
+  describe "all" do
+    it "should find all" do
+      EM.synchrony do
+        docs = @collection.find.all
+        docs.size.must_equal 20
+        EM.stop
+      end
+    end
+
+    it "should find all with query" do
+      EM.synchrony do
+        docs = @collection.find(artist: "Madonna").all
+        docs.size.must_equal 10
+        docs.each{ |d| d["artist"].must_equal "Madonna" }
+        EM.stop
+      end
+    end
+
+    it "should find all with limit" do
+      EM.synchrony do
+        docs = @collection.find.limit(5).all
+        docs.size.must_equal 5
+        EM.stop
+      end
+    end
+
+    it "should find all with batch size" do
+      EM.synchrony do
+        docs = @collection.find.batch_size(2).all
+        docs.size.must_equal 20
+        EM.stop
+      end
+    end
+
+    it "should find all with skip" do
+      EM.synchrony do
+        docs = @collection.find.skip(10).all
+        docs.size.must_equal 10
+        EM.stop
+      end
+    end
+  end
+
+  # FIRST
+
+  describe "first" do
+    it "should fetch first with sort" do
+      EM.synchrony do
+        doc = @collection.find.sort(title: 1).first
+        doc["title"].must_equal "Track 1"
+        EM.stop
+      end
+    end
+
+    it "should fetch first with sort and skip" do
+      EM.synchrony do
+        doc = @collection.find.sort(title: 1).skip(2).first
+        doc["title"].must_equal "Track 10"
+        EM.stop
+      end
+    end
+  end
+
+  # NEXT_BATCH
+
+  describe "next_batch" do
+    it "should fetch batches" do
+      EM.synchrony do
+        cursor = @collection.find.batch_size(2).limit(3)
+        batch, more = cursor.next_batch
+        batch.size.must_equal 2
+        more.must_equal true
+        batch, more = cursor.next_batch
+        batch.size.must_equal 1
+        more.must_equal false
+        EM.stop
+      end
+    end
+  end
+
+  # EACH_BATCH
+
+  describe "each_batch" do
+    it "should fetch 3 items by batches" do
+      EM.synchrony do
+        docs = []
+        @collection.find.batch_size(2).limit(3).each_batch do |batch|
+          docs += batch
+        end
+        docs.size.must_equal 3
+        EM.stop
+      end
+    end
+  end
+
+  # NEXT_DOC
+
+  describe "next_doc" do
+    it "should fetch doc by doc" do
+      EM.synchrony do
+        cursor = @collection.find.limit(3).batch_size(2)
+        doc, more = cursor.next_doc
+        more.must_equal true
+        doc, more = cursor.next_doc
+        doc, more = cursor.next_doc
+        more.must_equal false
+        EM.stop
+      end
+    end
+  end
+
+  # # EACH_DOC
+
+  describe "each_doc" do
+    it "should iterate over some docs" do
+      EM.synchrony do
+        docs = []
+        @collection.find.limit(100).skip(15).batch_size(3).each_doc do |doc|
+          docs << doc
+        end
+        docs.size.must_equal 5
+        EM.stop
+      end
+    end
+
+    it "should iterate over all docs" do
+      EM.synchrony do
+        docs = []
+        @collection.find.batch_size(3).each_doc do |doc|
+          docs << doc
+        end
+        docs.size.must_equal 20
+        EM.stop
+      end
+    end
+  end
+
+  # KILL CURSOR
+
+  describe "kill" do
+    it "should work with kill" do
+      EM.synchrony do
+        cursor = @collection.find
+        batch, more = cursor.next_batch
+        cursor.kill
+        proc{ cursor.next_batch }.must_raise Monga::Exceptions::ClosedCursor
+        EM.stop
+      end
+    end
+  end
+
+  # TAILABLE CURSOR
+
+  describe "tailable cursor" do
+    before do
+      EM.synchrony do
+        @db.create_collection("testCapped", capped: true, size: 4*1024)
+        @capped = @db["testCapped"]
+        @capped.safe_insert(title: "Test")
+        EM.stop
+      end
+    end
+
+    after do
+      EM.synchrony do
+        @db["testCapped"].drop
+        EM.stop
+      end
+    end
+
+    it "should be tailable" do
+      EM.synchrony do
+        tailable_cursor = @capped.find.flag(tailable_cursor: true)
+        docs = []
+        tailable_cursor.each_doc do |doc|
+          @capped.insert(title: "New!")
+          if doc
+            docs << doc
+            if docs.size == 2
+              docs.map{ |d| d["title"] }.must_equal ["Test", "New!"]
+              break
+            end
+          end
+        end
+        tailable_cursor.kill
+        EM.stop
+      end
+    end
+  end
+end

data/spec/monga/sync/database_spec.rb

@@ -0,0 +1,110 @@
+require 'spec_helper'
+
+describe Monga::Database do
+  before do
+    EM.synchrony do
+      @client = Monga::Client.new
+      @db = @client["dbTest"]
+      @collection = @db["testCollection"]
+      @collection.safe_remove
+      EM.stop
+    end
+  end
+
+  after do
+    EM.synchrony do
+      @collection.safe_remove
+      EM.stop
+    end
+  end
+
+  it "should create and drop collection" do
+    EM.synchrony do
+      @db.create_collection("cappedCollection")
+      @db.list_collections["retval"].must_include "cappedCollection"
+      @db.drop_collection("cappedCollection")
+      @db.list_collections["retval"].wont_include "cappedCollection"
+      EM.stop
+    end
+  end
+
+  it "should count in collection" do
+    EM.synchrony do
+      @collection.safe_insert([{ title: 1 }, { title: 2 }])
+      @db.count("testCollection").must_equal 2
+      EM.stop
+    end
+  end
+
+  it "should eval javascript" do
+    EM.synchrony do
+      @db.eval("1+1")["retval"].must_equal 2.0
+      EM.stop
+    end
+  end
+
+  # INDEXES
+
+  describe "indexes" do
+    before do
+      EM.synchrony do
+        @db.drop_indexes("testCollection", "*")
+        EM.stop
+      end
+    end
+
+    it "should drop index" do
+      EM.synchrony do
+        @collection.safe_ensure_index(title: 1)
+        @collection.get_indexes.select{ |i| i["ns"] == "dbTest.testCollection" }.size.must_equal 2
+        @db.drop_indexes("testCollection", title: 1)
+        @collection.get_indexes.select{ |i| i["ns"] == "dbTest.testCollection" }.size.must_equal 1
+        EM.stop
+      end
+    end
+  end
+
+  # GET LAST ERROR
+
+  describe "getLastError" do
+    before do
+      EM.synchrony do
+        @collection.drop_indexes
+        @collection.safe_ensure_index({ personal_id: 1 }, { unique: true, sparse: true })
+        EM.stop
+      end
+    end
+
+    it "should get last error" do
+      EM.synchrony do
+        req = @collection.insert(name: "Peter", personal_id: 10)
+        @db.get_last_error(req.connection)["ok"].must_equal 1.0
+        req = @collection.insert(name: "Peter", personal_id: 10)
+        @db.get_last_error(req.connection).class.must_equal Monga::Exceptions::QueryFailure
+        EM.stop
+      end
+    end
+
+    it "should getLastError with fsync" do
+      EM.synchrony do
+        req = @collection.insert(name: "Peter", personal_id: 10)
+        @db.get_last_error(req.connection, fsync: true)["ok"].must_equal 1.0
+        req = @collection.insert(name: "Peter", personal_id: 10)
+        @db.get_last_error(req.connection, fsync: true).class.must_equal Monga::Exceptions::QueryFailure
+        EM.stop
+      end
+    end
+  end
+
+  # AGGREGATION
+
+  describe "aggregation" do
+    it "should aggregate"
+  end
+
+  # MAP REDUCE
+
+  describe "map reduce" do
+    it "should run map reduce"
+  end
+end

data/spec/monga/sync/replica_set_client_spec.rb

@@ -0,0 +1,54 @@
+require 'spec_helper'
+
+describe Monga::Clients::ReplicaSetClient do
+  before do
+    EM.synchrony do
+      @replset = Fake::ReplicaSet.new([29000, 29100, 29200])
+      @client = Monga::Client.new servers: ['127.0.0.1:29000', '127.0.0.1:29100', '127.0.0.1:29200'], type: :sync, timeout: 1
+      @collection = @client["dbTest"]["myCollection"]
+      EM.stop
+    end
+  end
+
+  it "should fail on disconnect and reconnect when primary is up again" do
+    EM.synchrony do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter")
+      @replset.primary.stop
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      @replset.primary.start
+      @collection.safe_insert(name: "Madonna")
+      @collection.safe_insert(name: "Madonna")
+      @collection.safe_insert(name: "Madonna")
+      EM.stop
+    end
+  end
+
+  it "should work even if secondaries down" do
+    EM.synchrony do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter")
+      @collection.safe_insert(name: "Peter")
+      @replset.secondaries.each(&:stop)
+      @collection.safe_insert(name: "Peter")
+      @collection.safe_insert(name: "Peter")
+      EM.stop
+    end
+  end
+
+  it "should find new primary if it is down" do
+    EM.synchrony do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter")
+      @replset.primary.stop
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      @replset.vote
+      @collection.safe_insert(name: "Madonna")
+      EM.stop
+    end
+  end
+end

data/spec/monga/sync/single_instance_client_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe Monga::Clients::SingleInstanceClient do
+  before do
+    EM.synchrony do
+      @instance = Fake::SingleInstance.new(29000)
+      @client = Monga::Client.new port: 29000, type: :sync
+      @collection = @client["dbTest"]["myCollection"]
+      EM.stop
+    end
+  end
+
+  it "should fail on disconnect and reconnect when instance is up again" do
+    EM.synchrony do
+      @instance.start
+      @collection.safe_insert(name: "Peter")
+      @instance.stop
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      proc{ @collection.safe_insert(name: "Peter") }.must_raise Monga::Exceptions::Disconnected
+      @instance.start
+      @collection.safe_insert(name: "Madonna")
+      EM.stop
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1,29 +1,11 @@
 LIB_PATH = File.expand_path('../../lib/monga',  __FILE__)
 
 require LIB_PATH
-require 'helpers/truncate'
-require 'helpers/mongodb'
 require 'minitest/spec'
 require 'minitest/autorun'
 require 'minitest/reporters'
 MiniTest::Reporters.use! MiniTest::Reporters::SpecReporter.new
 
-EM.run do
-  CLIENT = Monga::Client.new(host: "localhost", port: 27017)
-  DB = CLIENT["dbTest"]
-  COLLECTION = DB["testCollection"]
-  EM.stop
-end
+Monga.logger.level = Logger::ERROR
 
-INSTANCE = Mongodb::Instance.new(dbpath: "/tmp/mongodb/instance/")
-
-REPL_SET_PORTS = [{ port: 29100 }, { port: 29200 }, { port: 29300 }]
-EM.run do
-  REPL_SET = Mongodb::ReplicaSet.new(REPL_SET_PORTS)
-  RS_CLIENT = Monga::ReplicaSetClient.new(servers: REPL_SET_PORTS)
-  RS_DB = RS_CLIENT["dbTest"]
-  RS_COLLECTION = RS_DB["testCollection"]
-  EM.stop
-end
-
-# And welcome to callback Hell
+require 'helpers/mongodb'