datamapper 0.2.5 → 0.3.0

data/spec/models/fruit.rb

@@ -1,6 +1,6 @@
-class Fruit #< DataMapper::Base
+class Fruit #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
-  
+    
   set_table_name 'fruit'
   property :name, :string, :column => 'fruit_name'
   

data/spec/models/job.rb

@@ -0,0 +1,8 @@
+class Job #< DataMapper::Base # please do not remove this
+  include DataMapper::Persistence
+
+  property :name, :string
+  property :hours, :days, :integer, :default => 0
+
+  has_and_belongs_to_many :candidates, :join_table => "applications_candidates"
+end

data/spec/models/person.rb

@@ -1,9 +1,8 @@
-class Person #< DataMapper::Base
+class Person #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
-  property :name, :string
+  property [:name, :occupation], :string
   property :age, :integer
-  property :occupation, :string
   property :type, :class
   property :notes, :text
   property :date_of_birth, :date

data/spec/models/post.rb

@@ -1,4 +1,4 @@
-class Post #< DataMapper::Base
+class Post #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
   property :title, :string

data/spec/models/project.rb

@@ -1,8 +1,9 @@
-class Project #< DataMapper::Base
+class Project #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
   property :title, :string
   property :description, :string
+  property :deleted_at, :datetime
 
   has_many :sections
   
@@ -13,9 +14,28 @@ class Project #< DataMapper::Base
     sections.map { |section| section.tickets }
   end
   
+  
+  def set_us_up_the_bomb=(val)
+    @set_us_up_the_bomb = !val.blank?
+  end
+  
+  def set_up_for_bomb?
+    @set_us_up_the_bomb
+  end
+  
+  def wery_sneaky?
+    @be_wery_sneaky
+  end
+  
+  
   private
   
   def create_main_section
     sections << Section.find_or_create(:title => "Main") if sections.empty?
   end
+  
+  def be_wery_sneaky=(val)
+    @be_wery_sneaky = val
+  end
+
 end

data/spec/models/section.rb

@@ -1,4 +1,4 @@
-class Section #< DataMapper::Base
+class Section #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
   property :title, :string

data/spec/models/serializer.rb

@@ -1,4 +1,4 @@
-class Serializer #< DataMapper::Base
+class Serializer #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
   property :content, :object, :lazy => false

data/spec/models/task.rb

@@ -0,0 +1,9 @@
+class Task #< DataMapper::Base # please do not remove this
+  include DataMapper::Persistence
+
+  property :name, :string, :default => "No Name", :index => :unique
+  property :notes, :string
+  property :completed, :boolean, :default => false
+
+  has_and_belongs_to_many :tasks
+end

data/spec/models/tomato.rb

@@ -0,0 +1,27 @@
+class Tomato
+  include DataMapper::Persistence
+  
+  ATTRIBUTES << :bruised
+  
+  def initialize(details = nil)
+    super
+    
+    @name = 'Ugly'
+    @init_run = true
+    @bruised = true
+  end
+  
+  def initialized?
+    @init_run
+  end
+  
+  property :name, :string
+  
+  def heal!
+    @bruised = false
+  end
+  
+  def bruised?
+    @bruised
+  end
+end

data/spec/models/user.rb

@@ -1,6 +1,12 @@
-class User #< DataMapper::Base
-  include DataMapper::Persistence
+# This uses the inheritance method of declaring
+# a model to boost spec coverage and ensure that the to
+# ways of declaring a model are compatible within the
+# same project and/or object graphs.
+class User < DataMapper::Base
   
   property :name, :string
+  property :email, :string, :format => :email_address
   has_many :comments, :class => 'Comment', :foreign_key => 'user_id'
+  
+  after_create { false }
 end

data/spec/models/zoo.rb

@@ -1,16 +1,11 @@
-class Zoo #< DataMapper::Base
+class Zoo #< DataMapper::Base # please do not remove this
   include DataMapper::Persistence
   
-  property :name, :string
+  property :name, :string, :nullable => false, :default => "Zoo"
   property :notes, :text
   property :updated_at, :datetime
   
   has_many :exhibits
-  begin
-  validates_presence_of :name
-  rescue ArgumentError => e
-    throw e unless e.message =~ /specify a unique key/
-  end
   
   def name=(val)
     @name = (val == "Colorado Springs") ? "Cheyenne Mountain" : val

data/spec/paranoia_spec.rb

@@ -3,7 +3,7 @@ require File.dirname(__FILE__) + "/spec_helper"
 describe 'Paranoia' do
   
   before(:all) do
-    class Scared
+    class Scared #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
       
       property :name, :string

data/spec/{base_spec.rb → persistence_spec.rb}

@@ -1,6 +1,146 @@
 require File.dirname(__FILE__) + "/spec_helper"
 
-describe DataMapper::Persistence do
+describe "DataMapper::Persistence::ConvenienceMethods::ClassMethods" do
+  
+  before(:all) do
+    fixtures(:animals)
+    fixtures(:exhibits)
+  end
+  
+  describe 'A record' do
+    it 'should save and return true if validations pass' do
+      count = Exhibit.count
+      bugs_and_more_bugs = Exhibit.new(:name => 'Bugs And More Bugs')
+      bugs_and_more_bugs.save.should be_true
+      Exhibit.count.should == count + 1
+    end
+
+    it 'should return false on save if validations fail' do
+      count = Exhibit.count
+      bugs_and_more_bugs = Exhibit.new
+      bugs_and_more_bugs.save.should be_false
+      Exhibit.count.should == count
+    end
+
+    it 'should reload its attributes' do
+      frog = Animal.first(:name => 'Frog')
+      frog.name = 'MegaFrog'
+      frog.name.should == 'MegaFrog'
+      frog.reload!
+      frog.name.should == 'Frog'
+    end
+    
+    it "should prepare it's associations for reload" do
+      chippy = Animal.first(:name => 'Cup')
+      amazonia = Exhibit.first(:name => 'Amazonia')
+      amazonia.animals << chippy
+      amazonia.animals.should include(chippy)
+      amazonia.reload!
+      amazonia.animals.should_not include(chippy)
+    end
+
+    it 'should be destroyed!' do
+      capybara = Animal.create(:name => 'Capybara')
+      count = Animal.count
+      capybara.destroy!
+      Animal.first(:name => 'Capybara').should be_nil
+      Animal.count.should == count - 1
+    end
+  end
+
+  it 'should return the first match using find_or_create' do
+    count = Animal.count
+    frog = Animal.find_or_create(:name => 'Frog')
+    frog.name.should == 'Frog'
+    Animal.count.should == count
+  end
+
+  it 'should create a record if a match is not found with find_or_create' do
+    count = Animal.count
+    capybara = Animal.find_or_create(:name => 'Capybara')
+    capybara.name.should == 'Capybara'
+    Animal.count.should == count + 1
+  end
+
+  it 'should return all records' do
+    all_animals = Animal.all
+    all_animals.length.should == Animal.count
+    all_animals.each do |animal|
+      animal.class.should == Animal
+    end
+  end
+
+  it 'should return the first record' do
+    Animal.first.should == Animal.find(:first)
+  end
+
+  it 'should return a count of the records' do
+    Animal.count.should == Animal.find_by_sql("SELECT COUNT(*) FROM animals")[0]
+  end
+
+  it 'should delete all records' do
+    Animal.delete_all
+    Animal.count.should == 0
+
+    fixtures(:animals)
+  end
+
+  #it 'should be truncated' do
+  #  Animal.truncate!
+  #  Animal.count.should == 0
+  #
+  #  fixtures(:animals)
+  #end
+
+  it 'should find a matching record given an id' do
+    Animal.find(1).name.should == 'Frog'
+  end
+
+  it 'should find all records' do
+    Animal.find(:all).length.should == Animal.count
+  end
+
+  it 'should find all matching records given some condition' do
+    Animal.find(:all, :conditions => ["name = ?", "Frog"])[0].name.should == 'Frog'
+  end
+
+  it 'should find the first matching record' do
+    Animal.find(:first).name.should == 'Frog'
+  end
+
+  it 'should find the first matching record given some condition' do
+    Animal.find(:first, :conditions => ["name = ?", "Frog"]).name.should == 'Frog'
+  end
+
+  it 'should select records using the supplied sql fragment' do
+    Animal.find_by_sql("SELECT name FROM animals WHERE name='Frog'")[0].should == 'Frog'
+  end
+
+  it 'should retrieve the indexed element' do
+    Animal[1].id.should == 1
+  end
+
+  it 'should create a new record' do
+    count = Animal.count
+    capybara = Animal.create(:name => 'Capybara')
+    capybara.name.should == 'Capybara'
+    Animal.count.should == count + 1
+  end
+end
+
+
+# Can't describe DataMapper::Persistence because
+# rspec will include it for some crazy reason!
+describe "DataMapper::Persistence" do
+  
+  it "should raise ObjectNotFoundError for missing objects with the indexer finder" do
+    # pending "http://wm.lighthouseapp.com/projects/4819-datamapper/tickets/111-should-raise-objectnotfounderror-on-the-indexer-finder"
+    lambda { Zoo[900] }.should raise_error(DataMapper::ObjectNotFoundError)
+  end
+  
+  it "should raise IncompleteModelDefinitionError for a model with no properties" do
+    lambda { class IncompleteZoo; include DataMapper::Persistence; end; IncompleteZoo.new }.should raise_error(DataMapper::Persistence::IncompleteModelDefinitionError)
+  end
   
   it "attributes method should load all lazy-loaded values" do
     Animal.first(:name => 'Cup').attributes[:notes].should == 'I am a Cup!'
@@ -69,13 +209,14 @@ describe DataMapper::Persistence do
     p1.should == p2
   end
 
-  it "should not be equal if attributes have changed" do
-    p1 = Person.create(:name => 'Sam')
-    p2 = Person[p1.id]
-    p2.name = "Paul"
-
-    p1.should_not == p2
-  end
+  # This is unnecessary. Use #dirty? to check for changes.
+  # it "should not be equal if attributes have changed" do
+  #   p1 = Person.create(:name => 'Sam')
+  #   p2 = Person[p1.id]
+  #   p2.name = "Paul"
+  # 
+  #   p1.should_not == p2
+  # end
 
 end
 
@@ -165,7 +306,7 @@ end
 describe 'Properties' do
 
   it 'should default to public method visibility' do
-    class SoftwareEngineer
+    class SoftwareEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
       
       set_table_name 'people'
@@ -177,7 +318,7 @@ describe 'Properties' do
   end
 
   it 'should respect protected property options' do
-    class SanitationEngineer
+    class SanitationEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -190,7 +331,7 @@ describe 'Properties' do
   end
 
   it 'should respect private property options' do
-    class ElectricalEngineer
+    class ElectricalEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -203,7 +344,7 @@ describe 'Properties' do
   end
 
   it 'should set both reader and writer visibiliy when accessor option is passed' do
-    class TrainEngineer
+    class TrainEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       property :name, :string, :accessor => :private
@@ -214,7 +355,7 @@ describe 'Properties' do
   end
 
   it 'should only be listed in attributes if they have public getters' do
-    class SalesEngineer
+    class SalesEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -228,7 +369,7 @@ describe 'Properties' do
   end
 
   it 'should not allow mass assignment if private or protected' do
-    class ChemicalEngineer
+    class ChemicalEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -243,7 +384,7 @@ describe 'Properties' do
   end
 
   it 'should allow :protected to be passed as an alias for a public reader, protected writer' do
-    class CivilEngineer
+    class CivilEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -255,7 +396,7 @@ describe 'Properties' do
   end
 
   it 'should allow :private to be passed as an alias for a public reader, private writer' do
-    class AudioEngineer
+    class AudioEngineer #< DataMapper::Base # please do not remove this
       include DataMapper::Persistence
 
       set_table_name 'people'
@@ -268,7 +409,7 @@ describe 'Properties' do
   
   it 'should raise an error when invalid options are passsed' do
     lambda do
-      class JumpyCow
+      class JumpyCow #< DataMapper::Base # please do not remove this
         include DataMapper::Persistence
 
         set_table_name 'animals'
@@ -279,7 +420,7 @@ describe 'Properties' do
 
   it 'should raise an error when the first argument to index isnt an array' do
     lambda do
-      class JumpyCow
+      class JumpyCow #< DataMapper::Base # please do not remove this
         include DataMapper::Persistence
 
         set_table_name 'animals'
@@ -287,4 +428,52 @@ describe 'Properties' do
       end
     end.should raise_error(ArgumentError)
   end
+  
+  it "should return true on saving a new record" do
+    bob = User.new(:name => 'bob', :email => 'bob@example.com')
+    bob.save!.should == true
+    bob.destroy!
+  end
+  
+  it "should assign to public setters" do
+    x = Project.new
+    x.attributes = { :set_us_up_the_bomb => true }
+    x.should be_set_up_for_bomb
+  end
+  
+  it "should protect private setters" do
+    x = Project.new
+    x.attributes = { :be_wery_sneaky => true }
+    x.should_not be_wery_sneaky
+  end
+  
+  it "should not call initialize on materialization" do
+    # pending "http://wm.lighthouseapp.com/projects/4819-datamapper/tickets/113-use-allocate-to-materialize-objects"
+    x = Tomato.new
+    x.should be_initialized
+    x.save!
+    x.name.should eql('Ugly')
+    x.should be_bruised
+    
+    x.name = 'Bob'
+    x.save!
+    
+    x2 = Tomato.first(:name => 'Bob')
+    x2.should_not be_bruised
+    x2.heal!
+    x2.should == x
+    x2.name.should eql('Bob')
+    x2.should_not be_initialized
+    
+    x3 = Tomato.get(x.key)
+    x3.should_not be_bruised
+    x3.heal!
+    x3.should == x
+    x3.name.should eql('Bob')
+    x3.should_not be_initialized
+  end
+  
+  it "should report persistence" do
+    Tomato.should be_persistent
+  end
 end