monga 0.0.2 → 0.0.3

data/spec/monga/em/cursor_spec.rb

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

data/spec/monga/em/database_spec.rb

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

@@ -0,0 +1,86 @@
+require 'spec_helper'
+
+describe Monga::Clients::ReplicaSetClient do
+  before do
+    EM.run 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: :em, timeout: 1
+      @collection = @client["dbTest"]["myCollection"]
+      EM.stop
+    end
+  end
+
+  it "should fail on disconnect and reconnect when primary is up again" do
+    EM.run do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter") do |err, resp|
+        @replset.primary.stop
+        @collection.safe_insert(name: "Peter") do |err, resp|
+          err.class.must_equal Monga::Exceptions::Disconnected
+          @collection.safe_insert(name: "Peter") do |err, resp|
+            err.class.must_equal Monga::Exceptions::Disconnected
+            @collection.safe_insert(name: "Peter") do |err, resp|
+              err.class.must_equal Monga::Exceptions::Disconnected
+              @replset.primary.start
+              @collection.safe_insert(name: "Madonna") do |err, resp|
+                err.must_equal nil
+                @collection.safe_insert(name: "Madonna") do |err, resp|
+                  err.must_equal nil
+                  @collection.safe_insert(name: "Madonna") do |err|
+                    err.must_equal nil
+                    EM.stop
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  it "should work even if secondaries down" do
+    EM.run do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter") do |err|
+        err.must_equal nil
+        @collection.safe_insert(name: "Peter") do |err|
+          err.must_equal nil
+          @replset.secondaries.each(&:stop)
+          @collection.safe_insert(name: "Peter") do |err|
+            err.must_equal nil
+            @collection.safe_insert(name: "Peter") do |err|
+              err.must_equal nil
+
+              EM.stop
+            end
+          end
+        end
+      end
+    end
+  end
+
+  it "should find new primary if it is down" do
+    EM.run do
+      @replset.start_all
+      @collection.safe_insert(name: "Peter") do |err|
+        err.must_equal nil
+        @replset.primary.stop
+        @collection.safe_insert(name: "Peter") do |err|
+          err.class.must_equal Monga::Exceptions::Disconnected
+          @collection.safe_insert(name: "Peter") do |err|
+            err.class.must_equal Monga::Exceptions::Disconnected
+            @collection.safe_insert(name: "Peter") do |err|
+              err.class.must_equal Monga::Exceptions::Disconnected
+              @replset.vote
+              @collection.safe_insert(name: "Madonna") do |err|
+                err.must_equal nil
+                EM.stop
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/spec/monga/em/single_instance_client_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+describe Monga::Clients::SingleInstanceClient do
+  before do
+    EM.synchrony do
+      @client = Monga::Client.new port: 29000, type: :em
+      @collection = @client["dbTest"]["myCollection"]
+      @instance = Fake::SingleInstance.new(29000)
+      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") do
+        @instance.stop
+        @collection.safe_insert(name: "Peter") do |err, resp|
+          err.class.must_equal Monga::Exceptions::Disconnected
+          @instance.start
+          @collection.safe_insert(name: "Madonna") do
+            EM.stop
+          end
+        end
+      end
+    end
+  end
+end