datamapper 0.1.1 → 0.2.0

data/spec/magic_columns_spec.rb

@@ -0,0 +1,9 @@
+describe "Magic Columns" do
+  
+  it "should update updated_at on save" do
+    zoo = Zoo.new(:name => 'Mary')
+    zoo.save
+    zoo.updated_at.should be_a_kind_of(Time)
+  end
+  
+end

data/spec/mock_adapter.rb

@@ -0,0 +1,20 @@
+require 'data_mapper/adapters/data_object_adapter'
+
+class MockAdapter < DataMapper::Adapters::DataObjectAdapter
+  COLUMN_QUOTING_CHARACTER = "`"
+  TABLE_QUOTING_CHARACTER = "`"
+  
+  def delete(instance_or_klass, options = nil)
+  end
+  
+  def save(session, instance)
+  end
+  
+  def load(session, klass, options)
+  end
+  
+  def table_exists?(name)
+    true
+  end
+  
+end

data/spec/models/animal.rb

@@ -1,6 +1,6 @@
 class Animal < DataMapper::Base
   property :name, :string
-  property :notes, :text, :lazy => true
+  property :notes, :text
   
   has_one :favourite_fruit, :class => 'Fruit', :foreign_key => 'devourer_id'
   has_and_belongs_to_many :exhibits

data/spec/models/animals_exhibit.rb

@@ -0,0 +1,6 @@
+# This is just here to get around the fact I use a Class to load
+# fixtures right now.
+class AnimalsExhibit < DataMapper::Base
+  property :animal_id, :integer, :key => true, :serial => false
+  property :exhibit_id, :integer
+end

data/spec/models/exhibit.rb

@@ -1,6 +1,8 @@
 class Exhibit < DataMapper::Base
   property :name, :string
   
+  validates_presence_of :name
+  
   belongs_to :zoo
   has_and_belongs_to_many :animals
 end

data/spec/models/person.rb

@@ -3,5 +3,30 @@ class Person < DataMapper::Base
   property :age, :integer
   property :occupation, :string
   property :type, :class
-  property :notes, :text, :lazy => true
+  property :notes, :text
+  property :date_of_birth, :datetime
+  
+  embed :address do
+    property :street, :string
+    property :city, :string
+    property :state, :string, :size => 2
+    property :zip_code, :string, :size => 10
+    
+    def city_state_zip_code
+      "#{city}, #{state} #{zip_code}"
+    end
+    
+  end
+  
+  class Location < DataMapper::EmbeddedValue
+    property :city, :string
+    property :state, :string, :size => 2
+
+    def to_s
+      "#{city}, #{state}"
+    end
+  end
+  
+  embed Location
+  
 end

data/spec/models/project.rb

@@ -0,0 +1,19 @@
+class Project < DataMapper::Base
+  property :title, :string
+  property :description, :string
+
+  has_many :sections
+  
+  before_create :create_main_section
+  
+  def tickets
+    return [] if sections.empty?
+    sections.map { |section| section.tickets }
+  end
+  
+  private
+  
+  def create_main_section
+    sections << Section.new(:title => "Main") if sections.empty?
+  end
+end

data/spec/models/sales_person.rb

@@ -1,4 +1,5 @@
 require File.dirname(__FILE__) + '/person'
 
 class SalesPerson < Person
+  property :commission, :integer
 end

data/spec/models/section.rb

@@ -0,0 +1,6 @@
+class Section < DataMapper::Base
+  property :title, :string
+  property :created_at, :datetime
+  
+  belongs_to :project
+end

data/spec/models/zoo.rb

@@ -1,7 +1,9 @@
 class Zoo < DataMapper::Base
   property :name, :string
   property :notes, :text
+  property :updated_at, :datetime
   
   has_many :exhibits
-  advanced_has_many :exhibits2, :class_name => 'Exhibit', :foreign_key => 'cow_id'
+  
+  validates_presence_of :name
 end

data/spec/query_spec.rb

@@ -0,0 +1,9 @@
+describe('A query') do
+  
+  it 'should return a simple Array of primitives' do
+    database.query("SELECT name FROM zoos").all? do |entry|
+      entry.kind_of?(String)
+    end.should == true
+  end
+  
+end

data/spec/save_command_spec.rb

@@ -1,13 +1,77 @@
-describe DataMapper::Adapters::Sql::Commands::SaveCommand do
+describe "Save Commands" do
   
   it "should create a new row" do
+    total = Zoo.all.length
     Zoo.create({ :name => 'bob' })
+    zoo = Zoo[:name => 'bob']
+    zoo.name.should == 'bob'
+    Zoo.all.length.should == total+1
   end
   
   it "should update an existing row" do
     dallas = Zoo[:name => 'Dallas']
     dallas.name = 'bob'
     dallas.save
+    dallas.name = 'Dallas'
+    dallas.save
+  end
+  
+  it "should be able to read other attributes after a record is created" do
+    zoo = Zoo.create({ :name => 'bob' })
+    zoo.notes.should == nil
+  end
+  
+  it "should stamp association on save" do
+    database do
+      dallas = Zoo[:name => 'Dallas']
+      dallas.exhibits << Exhibit.new(:name => 'Flying Monkeys')
+      dallas.save
+      Exhibit[:name => 'Flying Monkeys'].zoo.should == dallas
+    end
+  end
+  
+  it "should be invalid if invalid associations are loaded" do
+    miami = Zoo[:name => 'Miami']
+    fish_fancy = Exhibit.new
+    miami.exhibits << fish_fancy
+    miami.should_not be_valid
+    fish_fancy.name = 'Fish Fancy'
+    fish_fancy.should be_valid
+    miami.should be_valid
+  end
+  
+  it "should retrieve it's id on creation if the key is auto-incrementing so it can be successively updated" do
+    database do # Use the same Session so the reference-equality checks will pass.
+      mary = Animal::create(:name => 'Mary')
+      mary.name = 'Jane'
+      # Without retrieving the id during creation, the following #save call would fail,
+      # because we wouldn't know what id to update.
+      mary.save.should == true
+      jane = Animal.first(:name => 'Jane')
+      mary.should == jane
+    end
+  end
+  
+  it "should not be dirty if there are no attributes to update" do
+    bob = Animal.new
+    bob.should_not be_dirty
+    bob.name = 'bob'
+    bob.dirty_attributes.should == { :name => 'bob' }
+    bob.should be_dirty
+  end
+  
+  it "should not persist invalid objects" do
+    zoo = Zoo.create(:notes => "I'm invalid!")
+    zoo.should_not be_valid
+    zoo.should be_new_record
+  end
+  
+  it "should create a Project with a \"Main\" Section" do
+    database do
+      project = Project::create(:title => 'Test')
+      project.sections.first.should == Section.first
+      project.sections.first.should be_a_kind_of(Section)
+    end
   end
   
 end

data/spec/schema_spec.rb

@@ -0,0 +1,89 @@
+if ENV['ADAPTER'] == 'postgresql' && false
+  
+  describe DataMapper::Adapters::PostgresqlAdapter::Mappings::Table do
+    
+    before(:all) do
+      class Cage < DataMapper::Base
+        set_table_name "cages"
+        property :name, :string
+      end
+  
+      class CageInSchema < DataMapper::Base
+        set_table_name "my_schema.cages"
+        property :name, :string
+      end
+    end
+    
+    it "should return a quoted table name for a simple table" do
+      table_sql = database.adapter.table(Cage).to_sql
+      table_sql.should == "\"cages\""
+    end
+  
+    it "should return a quoted schema and table name for a table which specifies a schema" do
+      table_sql = database.adapter.table(CageInSchema).to_sql
+      table_sql.should == "\"my_schema\".\"cages\""
+    end
+
+    it "should search only the specified schema if qualified" do
+      database.save(Cage)
+      database.adapter.table(CageInSchema).exists?.should == false
+      database.save(CageInSchema)
+      database.adapter.table(CageInSchema).exists?.should == true
+    end
+    
+    after do
+      database.adapter.execute("DROP SCHEMA my_schema CASCADE") rescue nil
+    end
+    
+  end
+  
+  describe DataMapper::Adapters::PostgresqlAdapter::Commands::SaveCommand do
+    
+    before(:all) do
+      class Cage < DataMapper::Base
+        set_table_name "cages"
+        property :name, :string
+      end
+  
+      class CageInSchema < DataMapper::Base
+        set_table_name "my_schema.cages"
+        property :name, :string
+      end
+    end
+    
+    def table_mapping_for(klass)
+      session = database
+      DataMapper::Adapters::PostgresqlAdapter::Commands::SaveCommand.new(session.adapter, session, klass)
+    end
+  
+    it "should create a schema if it doesn't already exist" do
+      create_sql  = table_mapping_for(CageInSchema).to_create_table_sql
+      create_sql.should == <<-EOS.compress_lines
+      CREATE SCHEMA "my_schema"; CREATE TABLE "my_schema"."cages" ("id" serial primary key, "name" varchar)
+      EOS
+    end
+
+    it "shouldn't create a schema if it exists" do
+      database.save(CageInSchema)
+      create_sql  = table_mapping_for(CageInSchema).to_create_table_sql
+      create_sql.should == <<-EOS.compress_lines
+      CREATE TABLE "my_schema"."cages" ("id" serial primary key, "name" varchar)
+      EOS
+    end
+    
+    it "basic crud should work in schemas" do
+      database.save(CageInSchema)
+      CageInSchema.find(:all).size.should == 0
+      CageInSchema.create({ :name => 'bob' })
+      CageInSchema.find(:all).size.should == 1
+      cage = CageInSchema.first(:name => 'bob')
+      cage.name.should == 'bob'
+      cage.destroy!.should == true
+    end
+    
+    after do
+      database.adapter.execute("DROP SCHEMA my_schema CASCADE") rescue nil
+    end
+    
+  end
+end

data/spec/single_table_inheritance_spec.rb

@@ -0,0 +1,27 @@
+describe 'Single Table Inheritance' do
+  
+  before(:all) do
+    fixtures(:people)
+  end
+  
+  it "should save and load the correct Type" do
+    database do
+      ted = SalesPerson.new(:name => 'Ted')
+      ted.save
+    
+      clone = Person.first(:name => 'Ted')
+      ted.should == clone
+      
+      ted.should be_a_kind_of(SalesPerson)
+    end
+  end
+  
+  it "secondary database should inherit the same attributes" do
+    
+    database(:mock) do |db|
+      db.table(SalesPerson)[:name].should_not be_nil
+    end
+    
+  end
+  
+end

data/spec/spec_helper.rb

@@ -1,66 +1,20 @@
-require File.dirname(__FILE__) + '/../lib/data_mapper'
-require 'yaml'
-require 'pp'
-
-log_path = File.dirname(__FILE__) + '/../spec.log'
-
-require 'fileutils'
-FileUtils::rm log_path if File.exists?(log_path)
-
-case ENV['ADAPTER']
-when 'sqlite3' then
-  DataMapper::Database.setup do
-    adapter 'sqlite3'
-    database 'data_mapper_1.db'
-    log_stream 'spec.log'
-    log_level Logger::DEBUG
-  end
-when 'postgresql' then
-  DataMapper::Database.setup do
-    adapter  'postgresql'
-    database 'data_mapper_1'
-    username 'postgres'
-    log_stream 'spec.log'
-    log_level Logger::DEBUG
-  end
-else
-  DataMapper::Database.setup do
-    adapter 'mysql'
-    database 'data_mapper_1'
-    username 'root'
-    log_stream 'spec.log'
-    log_level Logger::DEBUG
-  end
-end
-
-Dir[File.dirname(__FILE__) + '/models/*.rb'].each do |path|
-  load path
-end
-
-database do |db|
-  db.schema.each do |table|
-    db.create_table(table.klass)
-  end
-end
-
-at_exit do
-  database do |db|
-    db.schema.each do |table|
-      db.drop_table(table.klass)
-    end
-  end
-end if ENV['DROP'] == '1'
+ENV['LOG_NAME'] = 'spec'
+require File.dirname(__FILE__) + '/../environment'
 
 # Define a fixtures helper method to load up our test data.
 def fixtures(name)
   entry = YAML::load_file(File.dirname(__FILE__) + "/fixtures/#{name}.yaml")
   klass = Kernel::const_get(Inflector.classify(Inflector.singularize(name)))
   
-  database.schema[klass].create!
-  klass.truncate!
+  database.log.debug { "AUTOMIGRATE: #{klass}" }
+  klass.auto_migrate!
   
   (entry.kind_of?(Array) ? entry : [entry]).each do |hash|
-    klass::create(hash)
+    if hash['type']
+      Object::const_get(hash['type'])::create(hash)
+    else
+      klass::create(hash)
+    end
   end
 end
 

data/spec/{symbolic_operators.rb → symbolic_operators_spec.rb}

@@ -1,21 +1,25 @@
-context 'Symbolic Operators' do
+describe Symbol::Operator do
   
-  specify('should use greater_than_or_equal_to to limit results') do
+  before(:all) do
+    fixtures(:people)
+  end
+  
+  it 'should use greater_than_or_equal_to to limit results' do
     Person.all(:age.gte => 28).size.should == 3
   end
   
-  specify('use an Array for in-clauses') do
+  it 'use an Array for in-clauses' do
     family = Person.all(:id => [1, 2, 4])
     family[0].name.should == 'Sam'
     family[1].name.should == 'Amy'
     family[2].name.should == 'Josh'
   end
   
-  specify('use "not" for not-equal operations') do
+  it 'use "not" for not-equal operations' do
     Person.all(:name.not => 'Bob').size.should == 4
   end
   
-  specify('age should not be nil') do
+  it 'age should not be nil' do
     Person.all(:age.not => nil).size.should == 5
   end
 end