monga 0.0.2 → 0.0.3

data/spec/monga/block/cursor_spec.rb

@@ -0,0 +1,160 @@
+require 'spec_helper'
+
+describe Monga::Cursor do
+  before do
+    @client = Monga::Client.new(type: :block, 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)
+  end
+
+  # ALL
+
+  describe "all" do
+    it "should find all" do
+      docs = @collection.find.all
+      docs.size.must_equal 20
+    end
+
+    it "should find all with query" do
+      docs = @collection.find(artist: "Madonna").all
+      docs.size.must_equal 10
+      docs.each{ |d| d["artist"].must_equal "Madonna" }
+    end
+
+    it "should find all with limit" do
+      docs = @collection.find.limit(5).all
+      docs.size.must_equal 5
+    end
+
+    it "should find all with batch size" do
+      docs = @collection.find.batch_size(2).all
+      docs.size.must_equal 20
+    end
+
+    it "should find all with skip" do
+      docs = @collection.find.skip(10).all
+      docs.size.must_equal 10
+    end
+  end
+
+  # FIRST
+
+  describe "first" do
+    it "should fetch first with sort" do
+      doc = @collection.find.sort(title: 1).first
+      doc["title"].must_equal "Track 1"
+    end
+
+    it "should fetch first with sort and skip" do
+      doc = @collection.find.sort(title: 1).skip(2).first
+      doc["title"].must_equal "Track 10"
+    end
+  end
+
+  # NEXT_BATCH
+
+  describe "next_batch" do
+    it "should fetch batches" 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
+    end
+  end
+
+  # EACH_BATCH
+
+  describe "each_batch" do
+    it "should fetch 3 items by batches" do
+      docs = []
+      @collection.find.batch_size(2).limit(3).each_batch do |batch|
+        docs += batch
+      end
+      docs.size.must_equal 3
+    end
+  end
+
+  # NEXT_DOC
+
+  describe "next_doc" do
+    it "should fetch doc by doc" 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
+    end
+  end
+
+  # # EACH_DOC
+
+  describe "each_doc" do
+    it "should iterate over some docs" do
+      docs = []
+      @collection.find.limit(100).skip(15).batch_size(3).each_doc do |doc|
+        docs << doc
+      end
+      docs.size.must_equal 5
+    end
+
+    it "should iterate over all docs" do
+      docs = []
+      @collection.find.batch_size(3).each_doc do |doc|
+        docs << doc
+      end
+      docs.size.must_equal 20
+    end
+  end
+
+  # KILL CURSOR
+
+  describe "kill" do
+    it "should work with kill" do
+      cursor = @collection.find
+      batch, more = cursor.next_batch
+      cursor.kill
+      proc{ cursor.next_batch }.must_raise Monga::Exceptions::ClosedCursor
+    end
+  end
+
+  # TAILABLE CURSOR
+
+  describe "tailable cursor" do
+    before do
+      @db.create_collection("testCapped", capped: true, size: 4*1024)
+      @capped =  @db["testCapped"]
+      @capped.safe_insert(title: "Test")
+    end
+
+    after do
+      @db["testCapped"].drop
+    end
+
+    it "should be tailable" 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
+    end
+  end
+end

data/spec/monga/block/database_spec.rb

@@ -0,0 +1,80 @@
+require 'spec_helper'
+
+describe Monga::Database do
+  before do
+    @client = Monga::Client.new
+    @db = @client["dbTest"]
+    @collection = @db["testCollection"]
+    @collection.safe_remove
+  end
+
+  after do
+    @collection.safe_remove
+  end
+
+  it "should create and drop collection" do
+    @db.create_collection("cappedCollection")
+    @db.list_collections["retval"].must_include "cappedCollection"
+    @db.drop_collection("cappedCollection")
+    @db.list_collections["retval"].wont_include "cappedCollection"
+  end
+
+  it "should count in collection" do
+    @collection.safe_insert([{ title: 1 }, { title: 2 }])
+    @db.count("testCollection").must_equal 2
+  end
+
+  it "should eval javascript" do
+    @db.eval("1+1")["retval"].must_equal 2.0
+  end
+
+  # INDEXES
+
+  describe "indexes" do
+    before do
+      @db.drop_indexes("testCollection", "*")
+    end
+
+    it "should drop index" 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
+    end
+  end
+
+  # GET LAST ERROR
+
+  describe "getLastError" do
+    before do
+      @collection.drop_indexes
+      @collection.safe_ensure_index({ personal_id: 1 }, { unique: true, sparse: true })
+    end
+
+    it "should get last error" 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
+    end
+
+    it "should getLastError with fsync" 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
+    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/block/single_instance_client_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+
+describe Monga::Clients::SingleInstanceClient do
+  before do
+    EM.synchrony do
+      @client = Monga::Client.new port: 29000, type: :block
+      @collection = @client["dbTest"]["myCollection"]
+      @instance = Fake::SingleInstance.new(29000)
+      EM.stop
+    end
+    @t = Thread.new do
+      EM.run do
+        @instance.start
+      end
+    end
+  end
+
+  after do
+    EM.stop
+    @t.join
+  end
+
+  it "should fail on disconnect and reconnect when instance is up again" do
+    @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")
+  end
+end

data/spec/monga/em/collection_spec.rb

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