mongo_db 0.1.9 → 0.1.10

data/spec/model/assignment_spec.rb

@@ -0,0 +1,79 @@
+require 'model/spec_helper'
+
+describe 'Model callbacks' do
+  with_mongo
+
+  after{remove_constants :User, :Writer}
+
+  it "should update attributes" do
+    class User
+      inherit Mongo::Model
+
+      attr_accessor :name, :active, :age, :banned
+    end
+
+    u = User.new
+    u.set name: 'Alex', active: '1', age: '31', banned: '0'
+    [u.name, u.active, u.age, u.banned].should == ['Alex', '1', '31', '0']
+  end
+
+  it "should update only specified attributes" do
+    class User
+      inherit Mongo::Model
+
+      attr_accessor :name, :active, :age, :banned
+
+      assignment do
+        name     String,  true
+        active   Boolean, true
+        age      Integer, true
+        banned   Boolean
+      end
+    end
+
+    u = User.new
+    u.set name: 'Alex', active: '1', age: '31', password: 'fake'
+    [u.name, u.active, u.age, u.banned].should == ['Alex', true, 31, nil]
+
+    # should allow to forcefully cast and update any attribute
+    u.set! banned: '0'
+    u.banned.should == false
+  end
+
+  it "should inherit assignment rules" do
+    class User
+      inherit Mongo::Model
+
+      attr_accessor :age
+
+      assignment do
+        age Integer,  true
+      end
+    end
+
+    class Writer < User
+      attr_accessor :posts
+
+      assignment do
+        posts Integer, true
+      end
+    end
+
+    u = Writer.new
+    u.set age: '20', posts: '12'
+    [u.age, u.posts].should == [20, 12]
+  end
+
+  it 'casting smoke test' do
+    [
+      Boolean, '1',          true,
+      Date,    '2011-08-23', Date.parse('2011-08-23'),
+      Float,   '1.2',        1.2,
+      Integer, '10',         10,
+      String,  'Hi',         'Hi',
+      Time,    '2011-08-23', Date.parse('2011-08-23').to_time
+    ].each_slice 3 do |type, raw, expected|
+      type.cast(raw).should == expected
+    end
+  end
+end

data/spec/model/callbacks_spec.rb

@@ -0,0 +1,47 @@
+require 'model/spec_helper'
+
+describe 'Model callbacks' do
+  with_mongo
+
+  after(:all){remove_constants :TheModel, :Player}
+
+  it "callback integration" do
+    class TheModel
+      inherit Mongo::Model
+    end
+
+    model = TheModel.new
+
+    model.should_receive(:run_before_callbacks).with(:save, {method: :save})
+    model._run_callbacks :before, :save
+  end
+
+  it "integration smoke test" do
+    class Player
+      inherit Mongo::Model
+
+      before_validate :before_validate_check
+      after_save :after_save_check
+
+      attr_accessor :missions
+
+      class Mission
+        inherit Mongo::Model
+
+        before_validate :before_validate_check
+        after_save :after_save_check
+      end
+    end
+
+    mission = Player::Mission.new
+    player = Player.new
+    player.missions = [mission]
+
+    player.should_receive(:before_validate_check).once.ordered.and_return(nil)
+    mission.should_receive(:before_validate_check).once.ordered.and_return(nil)
+    player.should_receive(:after_save_check).once.ordered.and_return(nil)
+    mission.should_receive(:after_save_check).once.ordered.and_return(nil)
+
+    db.units.save(player).should be_true
+  end
+end

data/spec/model/{model_crud.rb → crud_spec.rb}

@@ -1,101 +1,111 @@
-require 'spec_helper'
+require 'model/spec_helper'
 require 'object/crud_shared'
 
 describe "Model CRUD" do
-  with_mongo
+  with_mongo_model
 
   describe 'simple' do
-    before do
-      class Person
+    before :all do
+      class Unit
         inherit Mongo::Model
-        collection{db.units}
+        collection :units
 
-        def initialize name = nil, info = nil; @name, @info = name, info end
         attr_accessor :name, :info
         def == o; [self.class, name, info] == [o.class, o.respond_to(:name), o.respond_to(:info)] end
       end
+    end
+    after(:all){remove_constants :Unit}
 
-      @zeratul = Person.new 'Zeratul', 'Dark Templar'
+    before do
+      @zeratul = Unit.new.set name: 'Zeratul', info: 'Dark Templar'
     end
-    after{remove_constants :Person}
 
     it_should_behave_like "object CRUD"
 
     it 'model crud' do
       # read
-      Person.count.should == 0
-      Person.all.should == []
-      Person.first.should == nil
+      Unit.count.should == 0
+      Unit.all.should == []
+      Unit.first.should == nil
 
       # create
       @zeratul.save.should be_true
       @zeratul._id.should_not be_nil
 
       # read
-      Person.count.should == 1
-      Person.all.should == [@zeratul]
-      Person.first.should == @zeratul
-      Person.first.object_id.should_not == @zeratul.object_id
+      Unit.count.should == 1
+      Unit.all.should == [@zeratul]
+      Unit.first.should == @zeratul
+      Unit.first.object_id.should_not == @zeratul.object_id
 
       # update
       @zeratul.info = 'Killer of Cerebrates'
       @zeratul.save.should be_true
-      Person.count.should == 1
-      Person.first(name: 'Zeratul').info.should == 'Killer of Cerebrates'
+      Unit.count.should == 1
+      Unit.first(name: 'Zeratul').info.should == 'Killer of Cerebrates'
 
       # destroy
       @zeratul.destroy.should be_true
-      Person.count.should == 0
+      Unit.count.should == 0
     end
 
     it 'should be able to save to another collection' do
       # create
-      @zeratul.save(collection: db.protosses).should be_true
+      @zeratul.save(collection: db.heroes).should be_true
       @zeratul._id.should_not be_nil
 
       # read
-      Person.count.should == 0
-      db.protosses.count.should == 1
-      db.protosses.should == @zeratul
-      db.protosses.object_id.should_not == @zeratul.object_id
+      Unit.count.should == 0
+      db.heroes.count.should == 1
+      db.heroes.first.should == @zeratul
+      db.heroes.first.object_id.should_not == @zeratul.object_id
 
       # update
       @zeratul.info = 'Killer of Cerebrates'
-      @zeratul.save(collection: db.protosses).should be_true
-      Person.count.should == 0
-      db.protosses.count.should == 1
-      db.protosses.first(name: 'Zeratul').info.should == 'Killer of Cerebrates'
+      @zeratul.save(collection: db.heroes).should be_true
+      Unit.count.should == 0
+      db.heroes.count.should == 1
+      db.heroes.first(name: 'Zeratul').info.should == 'Killer of Cerebrates'
 
       # destroy
-      @zeratul.destroy(collection: db.protosses).should be_true
-      db.protosses.count.should == 0
+      @zeratul.destroy(collection: db.heroes).should be_true
+      db.heroes.count.should == 0
     end
+
+    it 'create'
+
+    it 'destroy_all'
   end
 
   describe 'embedded' do
-    before do
+    before :all do
       class Player
         inherit Mongo::Model
+        collection :players
+
         attr_accessor :missions
         def == o; [self.class, self.missions] == [o.class, o.respond_to(:missions)] end
 
         class Mission
           inherit Mongo::Model
-          def initialize name = nil, stats = nil; @name, @stats = name, stats end
+
           attr_accessor :name, :stats
           def == o; [self.class, self.name, self.stats] == [o.class, o.respond_to(:name), o.respond_to(:stats)] end
         end
       end
+    end
+    after(:all){remove_constants :Player}
 
+    before do
+      @mission_class = Player::Mission
       @player = Player.new
       @player.missions = [
-        Player::Mission.new('Wasteland',         {buildings: 5, units: 10}),
-        Player::Mission.new('Backwater Station', {buildings: 8, units: 25}),
+        Player::Mission.new.set(name: 'Wasteland',         stats: {buildings: 5, units: 10}),
+        Player::Mission.new.set(name: 'Backwater Station', stats: {buildings: 8, units: 25}),
       ]
     end
-    after{remove_constants :Player}
 
-    it_should_behave_like 'shared object CRUD'
+    it_should_behave_like 'embedded object CRUD'
 
     it 'crud' do
       # create
@@ -109,7 +119,7 @@ describe "Model CRUD" do
 
       # update
       @player.missions.first.stats[:units] = 9
-      @player.missions << Player::Mission.new('Desperate Alliance', {buildings: 11, units: 40}),
+      @player.missions << Player::Mission.new.set(name: 'Desperate Alliance', stats: {buildings: 11, units: 40})
       @player.save.should be_true
       Player.count.should == 1
       Player.first.should == @player

data/spec/model/db_spec.rb

@@ -0,0 +1,63 @@
+require 'model/spec_helper'
+
+describe 'Collection & Database' do
+  with_mongo
+
+  before :all do
+    class TheModel
+      inherit Mongo::Model
+    end
+  end
+  after(:all){remove_constants :TheModel}
+
+  after do
+    TheModel.db = nil
+    TheModel.collection = nil
+    Mongo::Model.connection, Mongo::Model.db = nil, nil
+  end
+
+  it "global setting" do
+    Mongo::Model.connection = db.connection
+    Mongo::Model.db = db
+
+    Mongo::Model.connection.should == db.connection
+    Mongo::Model.db.should == db
+  end
+
+  it "should allow set database per model" do
+    Mongo::Model.connection = db.connection
+    Mongo::Model.db = db
+
+    TheModel.db.should == db
+
+    TheModel.db :test
+    TheModel.db.name.should == 'test'
+
+    TheModel.db = nil
+    TheModel.db db
+    TheModel.db.should == db
+
+    TheModel.db = nil
+    TheModel.db{db}
+    TheModel.db.should == db
+  end
+
+  it "should allow set collection per model" do
+    Mongo::Model.db = db
+
+    # TheModel.default_collection_name.should == :the_model
+    # TheModel.collection.name.should == 'the_model'
+
+    TheModel.collection :units
+    TheModel.collection.name.should == 'units'
+
+    TheModel.collection = nil
+    units = db.units
+    TheModel.collection units
+    TheModel.collection.should == units
+
+    TheModel.collection = nil
+    TheModel.collection{units}
+    TheModel.collection.should == units
+  end
+end

data/spec/model/misc_spec.rb

@@ -0,0 +1,32 @@
+require 'model/spec_helper'
+
+describe 'Model Miscellaneous' do
+  with_mongo_model
+
+  after{remove_constants :Unit}
+
+  it "timestamps" do
+    class Unit3
+      inherit Mongo::Model
+      collection :units
+
+      attr_accessor :name
+
+      timestamps!
+    end
+
+    unit = Unit3.new.set name: 'Zeratul'
+    unit.save!
+
+    unit = Unit3.first
+    unit.created_at.should_not be_nil
+    unit.updated_at.should_not be_nil
+    created_at,updated_at = unit.created_at, unit.updated_at
+
+    unit.save!
+    unit.created_at.should == created_at
+    unit.updated_at.should >  updated_at
+  end
+
+
+end

data/spec/model/query_spec.rb

@@ -0,0 +1,47 @@
+require 'model/spec_helper'
+require 'object/crud_shared'
+
+describe "Model Query" do
+  with_mongo_model
+
+  before :all do
+    class Unit
+      inherit Mongo::Model
+      collection :units
+
+      attr_accessor :name
+    end
+  end
+  after(:all){remove_constants :Unit}
+
+  before{@zeratul = Unit.new.set name: 'Zeratul'}
+
+  it 'exist?' do
+    Unit.should_not exist(name: 'Zeratul')
+    @zeratul.save!
+    Unit.should exist(name: 'Zeratul')
+  end
+
+  it 'first, first!' do
+    Unit.first.should be_nil
+    -> {Unit.first!}.should raise_error(Mongo::NotFound)
+    @zeratul.save
+    Unit.first.should_not be_nil
+    Unit.first!.should_not be_nil
+  end
+
+  it 'all, each' do
+    list = []; Unit.each{|o| list << o}
+    list.size.should == 0
+
+    @zeratul.save
+    list = []; Unit.each{|o| list << o}
+    list.size.should == 1
+  end
+
+  it 'dynamic finders integration' do
+    Unit.first_by_name('Zeratul').should be_nil
+    Unit.new.set(name: 'Zeratul').save!
+    Unit.first_by_name('Zeratul').name.should == 'Zeratul'
+  end
+end

data/spec/model/scope_spec.rb

@@ -0,0 +1,149 @@
+require 'model/spec_helper'
+
+describe "Scope" do
+  with_mongo_model
+
+  before do
+    class Unit
+      inherit Mongo::Model
+      collection :units
+
+      attr_accessor :name, :status, :race
+    end
+  end
+
+  after{remove_constants :Unit, :Protoss}
+
+  describe 'current scope' do
+    it "should affect finders" do
+      Unit.new.set(name: 'Zeratul',  status: 'alive').save!
+      Unit.new.set(name: 'Jim',      status: 'alive').save!
+      Unit.new.set(name: 'Tassadar', status: 'dead').save!
+
+      Unit.count.should == 3
+      Unit.all.size.should == 3
+      Unit.first(name: 'Tassadar').should_not be_nil
+      Unit.first!(name: 'Tassadar').should_not be_nil
+
+      Unit.stub!(:current_scope).and_return(status: 'alive')
+
+      Unit.count.should == 2
+      Unit.all.size.should == 2
+      Unit.first(name: 'Tassadar').should be_nil
+      -> {Unit.first!(name: 'Tassadar')}.should raise_error(Mongo::NotFound)
+
+      # should be merged with finders
+      Unit.count(status: 'dead').should == 1
+      Unit.all(status: 'dead').size.should == 1
+      Unit.first(name: 'Tassadar', status: 'dead').should_not be_nil
+      Unit.first!(name: 'Tassadar', status: 'dead').should_not be_nil
+    end
+  end
+
+  describe 'default scope' do
+    it "should not affect objects without default_scope" do
+      Unit.current_scope.should == {}
+    end
+
+    it "definition" do
+      Unit.default_scope status: 'alive'
+      Unit.current_scope.should == {status: 'alive'}
+
+      Unit.default_scope do
+        {status: 'alive'}
+      end
+      Unit.current_scope.should == {status: 'alive'}
+    end
+
+    it "should be inherited" do
+      Unit.default_scope status: 'alive'
+
+      class Protoss < Unit; end
+      Protoss.current_scope.should == {status: 'alive'}
+
+      Protoss.default_scope status: 'dead'
+      Unit.current_scope.should == {status: 'alive'}
+      Protoss.current_scope.should == {status: 'dead'}
+    end
+  end
+
+  describe 'scope' do
+    it "definition" do
+      Unit.scope :alive, status: 'alive'
+      Unit.alive.current_scope.should == {status: 'alive'}
+
+      Unit.scope :alive do
+        {status: 'alive'}
+      end
+      Unit.alive.current_scope.should == {status: 'alive'}
+    end
+
+    it 'scope should affect current scope' do
+      Unit.scope :alive, status: 'alive'
+
+      Unit.current_scope.should == {}
+
+      Unit.alive.current_scope.should == {status: 'alive'}
+      Unit.alive.should == Unit
+    end
+
+    it 'should be merged with default scope' do
+      Unit.default_scope race: 'Protoss'
+      Unit.scope :alive, status: 'alive'
+      Unit.alive.current_scope.should == {race: 'Protoss', status: 'alive'}
+    end
+
+    it 'should allow to chain scopes' do
+      Unit.scope :alive, status: 'alive'
+      Unit.scope :protosses, race: 'Protoss'
+      Unit.alive.protosses.current_scope.should == {race: 'Protoss', status: 'alive'}
+    end
+  end
+
+  describe 'with_scope' do
+    it "shouldn't allow to nest exclusive scope" do
+      -> {
+        Unit.with_exclusive_scope do
+          Unit.with_exclusive_scope{}
+        end
+      }.should raise_error(/exclusive scope already applied/)
+
+      -> {
+        Unit.with_exclusive_scope do
+          Unit.with_scope{}
+        end
+      }.should raise_error(/exclusive scope already applied/)
+    end
+
+    it "with_exclusive_scope should clear other scopes" do
+      Unit.default_scope status: 'alive'
+
+      Unit.with_scope race: 'Protoss' do
+        Unit.current_scope.should == {status: 'alive', race: 'Protoss'}
+
+        Unit.with_exclusive_scope do
+          Unit.current_scope.should == {}
+        end
+
+        Unit.with_exclusive_scope race: 'Terran' do
+          Unit.current_scope.should == {race: 'Terran'}
+        end
+      end
+    end
+
+    it "usage" do
+      Unit.with_scope status: 'alive' do
+        Unit.current_scope.should == {status: 'alive'}
+      end
+    end
+
+    it "should merge scope" do
+      Unit.default_scope status: 'alive'
+      Unit.with_scope race: 'Protoss' do
+        Unit.with_scope name: 'Zeratul' do
+          Unit.current_scope.should == {name: 'Zeratul', race: 'Protoss', status: 'alive'}
+        end
+      end
+    end
+  end
+end

data/spec/model/spec_helper.rb

@@ -0,0 +1,4 @@
+require 'mongo_db/model'
+
+require 'object/spec_helper'
+require 'mongo_db/model/spec'

data/spec/model/validation_spec.rb

@@ -0,0 +1,37 @@
+require 'model/spec_helper'
+
+describe "Validations" do
+  with_mongo_model
+
+  before do
+    class Unit
+      inherit Mongo::Model
+      collection :units
+
+      attr_accessor :errors
+
+      attr_accessor :name
+    end
+  end
+  after{remove_constants :Unit}
+
+  it "should not save model with errors" do
+    unit = Unit.new.set name: 'Zeratul'
+    unit.save.should be_true
+
+    unit.errors = []
+    unit.save.should be_true
+
+    unit.errors = ['hairy error']
+    unit.save.should be_false
+
+    unit.errors = {name: 'hairy error'}
+    unit.save.should be_false
+  end
+
+  it "should check :errors only and ignore valid? method" do
+    unit = Unit.new.set name: 'Zeratul'
+    unit.should_not_receive(:valid?)
+    unit.save.should be_true
+  end
+end

data/spec/object/callbacks_spec.rb

@@ -3,7 +3,7 @@ require 'object/spec_helper'
 describe 'Object callbacks' do
   with_mongo
 
-  before do
+  before :all do
     class Player
       include RSpec::CallbackHelper
       attr_accessor :missions
@@ -12,12 +12,14 @@ describe 'Object callbacks' do
         include RSpec::CallbackHelper
       end
     end
+  end
+  after(:all){remove_constants :Player}
 
+  before do
     @mission = Player::Mission.new
     @player = Player.new
     @player.missions = [@mission]
   end
-  after{remove_constants :Player}
 
   it 'create' do
     %w(before_validate before_save before_create after_create after_save after_validate).each do |name|
@@ -61,8 +63,8 @@ describe 'Object callbacks' do
   end
 
   it 'should be able interrupt CRUD' do
-    @mission.stub! :_run_callbacks do |name|
-      false if name == :before_save
+    @mission.stub! :_run_callbacks do |type, method_name|
+      false if type == :before and method_name == :save
     end
     db.players.save(@player).should be_false
     db.players.count.should == 0

data/spec/object/crud_shared.rb

@@ -40,7 +40,7 @@ shared_examples_for 'embedded object CRUD' do
 
     # update
     @player.missions.first.stats[:units] = 9
-    @player.missions << Player::Mission.new('Desperate Alliance', {buildings: 11, units: 40})
+    @player.missions << @mission_class.new('Desperate Alliance', {buildings: 11, units: 40})
     db.players.save @player
     db.players.count.should == 1
     db.players.first.should == @player