dm-core 0.9.2

data/spec/integration/sti_spec.rb

@@ -0,0 +1,185 @@
+require 'pathname'
+require Pathname(__FILE__).dirname.expand_path.parent + 'spec_helper'
+
+if HAS_SQLITE3
+  describe DataMapper::AutoMigrations, '.auto_migrate! on STI models with sqlite3' do
+    before :all do
+      @adapter = repository(:sqlite3).adapter
+
+      @property_class = Struct.new(:name, :type, :nullable, :default, :serial)
+
+      class Book
+        include DataMapper::Resource
+
+        property :id,       Serial
+        property :title,    String,     :nullable => false
+        property :isbn,     Integer,    :nullable => false
+        property :class_type, Discriminator
+      end
+
+      class Propaganda < Book
+        property :marxist,  Boolean,    :nullable => false, :default => false
+      end
+
+      class Fiction < Book
+        property :series,   String
+      end
+
+      class ShortStory < Fiction
+        property :moral,    String
+      end
+
+      class ScienceFiction < Fiction
+        property :aliens, Boolean
+      end
+
+      class SpaceWestern < ScienceFiction
+        property :cowboys, Boolean
+      end
+    end
+
+    describe "with a parent class" do
+      before :all do
+        Book.auto_migrate!(:sqlite3).should be_true
+
+        @table_set = @adapter.query('PRAGMA table_info("books")').inject({}) do |ts,column|
+          default = if 'NULL' == column.dflt_value || column.dflt_value.nil?
+            nil
+          else
+            /^(['"]?)(.*)\1$/.match(column.dflt_value)[2]
+          end
+
+          property = @property_class.new(
+            column.name,
+            column.type.upcase,
+            column.notnull == 0,
+            default,
+            column.pk == 1  # in SQLite3 the serial key is also primary
+          )
+
+          ts.update(property.name => property)
+        end
+
+        @index_list = @adapter.query('PRAGMA index_list("books")')
+      end
+
+      it "should create the child class property columns" do
+        @table_set.keys.should include("series", "marxist")
+      end
+
+      it "should create all property columns of the child classes in the inheritance tree" do
+        @table_set.keys.should include("moral")
+      end
+    end
+
+    describe "with a child class" do
+      before :all do
+        Propaganda.auto_migrate!(:sqlite3).should be_true
+
+        @table_set = @adapter.query('PRAGMA table_info("books")').inject({}) do |ts,column|
+          default = if 'NULL' == column.dflt_value || column.dflt_value.nil?
+            nil
+          else
+            /^(['"]?)(.*)\1$/.match(column.dflt_value)[2]
+          end
+
+          property = @property_class.new(
+            column.name,
+            column.type.upcase,
+            column.notnull == 0,
+            default,
+            column.pk == 1  # in SQLite3 the serial key is also primary
+          )
+
+          ts.update(property.name => property)
+        end
+
+        @index_list = @adapter.query('PRAGMA index_list("books")')
+      end
+
+      it "should create the parent class' property columns" do
+        @table_set.keys.should include("id", "title", "isbn")
+      end
+    end
+
+    describe "with a child class with it's own child class" do
+      before :all do
+        Fiction.auto_migrate!(:sqlite3).should be_true
+
+        @table_set = @adapter.query('PRAGMA table_info("books")').inject({}) do |ts,column|
+          default = if 'NULL' == column.dflt_value || column.dflt_value.nil?
+            nil
+          else
+            /^(['"]?)(.*)\1$/.match(column.dflt_value)[2]
+          end
+
+          property = @property_class.new(
+            column.name,
+            column.type.upcase,
+            column.notnull == 0,
+            default,
+            column.pk == 1  # in SQLite3 the serial key is also primary
+          )
+
+          ts.update(property.name => property)
+        end
+
+        @index_list = @adapter.query('PRAGMA index_list("books")')
+      end
+
+      it "should create the parent class' property columns" do
+        @table_set.keys.should include("id", "title", "isbn")
+      end
+
+      it "should create the child class' property columns" do
+        @table_set.keys.should include("moral")
+      end
+    end
+
+    describe "with a nephew class" do
+      before :all do
+        ShortStory.auto_migrate!(:sqlite3).should be_true
+
+        @table_set = @adapter.query('PRAGMA table_info("books")').inject({}) do |ts,column|
+          default = if 'NULL' == column.dflt_value || column.dflt_value.nil?
+            nil
+          else
+            /^(['"]?)(.*)\1$/.match(column.dflt_value)[2]
+          end
+
+          property = @property_class.new(
+            column.name,
+            column.type.upcase,
+            column.notnull == 0,
+            default,
+            column.pk == 1  # in SQLite3 the serial key is also primary
+          )
+
+          ts.update(property.name => property)
+        end
+        @index_list = @adapter.query('PRAGMA index_list("books")')
+      end
+
+
+      it "should create the grandparent class' property columns" do
+        @table_set.keys.should include("id", "title", "isbn")
+      end
+
+      it "should create the uncle class' property columns" do
+        @table_set.keys.should include("marxist")
+      end
+    end
+
+    describe "with a great-grandchild class" do
+      it "should inherit its parent's properties" do
+        SpaceWestern.properties[:aliens].should_not be_nil
+      end
+      it "should inherit its grandparent's properties" do
+        SpaceWestern.properties[:series].should_not be_nil
+      end
+      it "should inherit its great-granparent's properties" do
+        SpaceWestern.properties[:title].should_not be_nil
+      end
+    end
+  end
+end

data/spec/integration/transaction_spec.rb

@@ -0,0 +1,75 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+# transaction capable adapters
+ADAPTERS = []
+ADAPTERS << :postgres if HAS_POSTGRES
+ADAPTERS << :mysql    if HAS_MYSQL
+ADAPTERS << :sqlite3  if HAS_SQLITE3
+
+if ADAPTERS.any?
+  class Sputnik
+    include DataMapper::Resource
+
+    property :id, Serial
+    property :name, DM::Text
+  end
+
+  describe DataMapper::Transaction do
+    before :all do
+      @repositories = []
+
+      ADAPTERS.each do |name|
+        @repositories << repository(name)
+      end
+    end
+
+    before :each do
+      ADAPTERS.each do |name|
+        Sputnik.auto_migrate!(name)
+      end
+    end
+
+    it "should commit changes to all involved adapters on a two phase commit" do
+      DataMapper::Transaction.new(*@repositories) do
+        ADAPTERS.each do |name|
+          repository(name) { Sputnik.create!(:name => 'hepp') }
+        end
+      end
+
+      ADAPTERS.each do |name|
+        repository(name) { Sputnik.all.size.should == 1 }
+      end
+    end
+
+    it "should not commit any changes if the block raises an exception" do
+      lambda do
+        DataMapper::Transaction.new(*@repositories) do
+          ADAPTERS.each do |name|
+            repository(name) { Sputnik.create!(:name => 'hepp') }
+          end
+          raise "plur"
+        end
+      end.should raise_error(Exception, /plur/)
+
+      ADAPTERS.each do |name|
+        repository(name) { Sputnik.all.size.should == 0 }
+      end
+    end
+
+    it "should not commit any changes if any of the adapters doesnt prepare properly" do
+      lambda do
+        DataMapper::Transaction.new(*@repositories) do |transaction|
+          ADAPTERS.each do |name|
+            repository(name) { Sputnik.create!(:name => 'hepp') }
+          end
+
+          transaction.primitive_for(@repositories.last.adapter).should_receive(:prepare).and_throw(Exception.new("I am the famous test exception"))
+        end
+      end.should raise_error(Exception, /I am the famous test exception/)
+
+      ADAPTERS.each do |name|
+        repository(name) { Sputnik.all.size.should == 0 }
+      end
+    end
+  end
+end

data/spec/integration/type_spec.rb

@@ -0,0 +1,149 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+gem 'fastercsv', '>=1.2.3'
+require 'fastercsv'
+
+if ADAPTER
+  module TypeTests
+    class Impostor < DataMapper::Type
+      primitive String
+    end
+
+    class Coconut
+      include DataMapper::Resource
+
+      storage_names[ADAPTER] = 'coconuts'
+
+      def self.default_repository_name
+        ADAPTER
+      end
+
+      property :id, Serial
+      property :faked, Impostor
+      property :active, Boolean
+      property :note, Text
+    end
+  end
+
+  class Lemon
+    include DataMapper::Resource
+
+    def self.default_repository_name
+      ADAPTER
+    end
+
+    property :id, Serial
+    property :color, String
+    property :deleted_at, DataMapper::Types::ParanoidDateTime
+  end
+
+  class Lime
+    include DataMapper::Resource
+
+    def self.default_repository_name
+      ADAPTER
+    end
+
+    property :id, Serial
+    property :color, String
+    property :deleted_at, DataMapper::Types::ParanoidBoolean
+  end
+
+  describe DataMapper::Type, "with #{ADAPTER}" do
+    before do
+      TypeTests::Coconut.auto_migrate!(ADAPTER)
+
+      @document = <<-EOS.margin
+        NAME, RATING, CONVENIENCE
+        Freebird's, 3, 3
+        Whataburger, 1, 5
+        Jimmy John's, 3, 4
+        Mignon, 5, 2
+        Fuzi Yao's, 5, 1
+        Blue Goose, 5, 1
+      EOS
+
+      @stuff = YAML::dump({ 'Happy Cow!' => true, 'Sad Cow!' => false })
+
+      @active = true
+      @note = "This is a note on our ol' guy bob"
+    end
+
+    it "should instantiate an object with custom types" do
+      coconut = TypeTests::Coconut.new(:faked => 'bob', :active => @active, :note => @note)
+      coconut.faked.should == 'bob'
+      coconut.active.should be_a_kind_of(TrueClass)
+      coconut.note.should be_a_kind_of(String)
+    end
+
+    it "should CRUD an object with custom types" do
+      repository(ADAPTER) do
+        coconut = TypeTests::Coconut.new(:faked => 'bob', :active => @active, :note => @note)
+        coconut.save.should be_true
+        coconut.id.should_not be_nil
+
+        fred = TypeTests::Coconut.get!(coconut.id)
+        fred.faked.should == 'bob'
+        fred.active.should be_a_kind_of(TrueClass)
+        fred.note.should be_a_kind_of(String)
+
+        note = "Seems like bob is just mockin' around"
+        fred.note = note
+
+        fred.save.should be_true
+
+        active = false
+        fred.active = active
+
+        fred.save.should be_true
+
+        # Can't call coconut.reload since coconut.collection isn't setup.
+        mac = TypeTests::Coconut.get!(fred.id)
+        mac.active.should == active
+        mac.note.should == note
+      end
+    end
+
+    it "should respect paranoia with a datetime" do
+      Lemon.auto_migrate!(ADAPTER)
+
+      lemon = nil
+
+      repository(ADAPTER) do |repository|
+        lemon = Lemon.new
+        lemon.color = 'green'
+
+        lemon.save
+        lemon.destroy
+
+        lemon.deleted_at.should be_kind_of(DateTime)
+      end
+
+      repository(ADAPTER) do |repository|
+        Lemon.all.should be_empty
+        Lemon.get(lemon.id).should be_nil
+      end
+    end
+
+    it "should respect paranoia with a boolean" do
+      Lime.auto_migrate!(ADAPTER)
+
+      lime = nil
+
+      repository(ADAPTER) do |repository|
+        lime = Lime.new
+        lime.color = 'green'
+
+        lime.save
+        lime.destroy
+
+        lime.deleted_at.should be_kind_of(TrueClass)
+      end
+
+      repository(ADAPTER) do |repository|
+        Lime.all.should be_empty
+        Lime.get(lime.id).should be_nil
+      end
+    end
+  end
+end

data/spec/lib/mock_adapter.rb

@@ -0,0 +1,27 @@
+module DataMapper
+  module Adapters
+    class MockAdapter < DataMapper::Adapters::DataObjectsAdapter
+
+      def create(resources)
+        1
+      end
+
+      def exists?(storage_name)
+        true
+      end
+
+    end
+  end
+end
+
+module DataObjects
+  module Mock
+
+    def self.logger
+    end
+
+    def self.logger=(value)
+    end
+
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,112 @@
+require 'rubygems'
+gem 'rspec', '>=1.1.3'
+require 'spec'
+require 'pathname'
+
+require Pathname(__FILE__).dirname.expand_path.parent + 'lib/dm-core'
+require DataMapper.root / 'spec' / 'lib' / 'mock_adapter'
+
+# setup mock adapters
+[ :default, :mock, :legacy, :west_coast, :east_coast ].each do |repository_name|
+  DataMapper.setup(repository_name, "mock://localhost/#{repository_name}")
+end
+
+def setup_adapter(name, default_uri)
+  begin
+    DataMapper.setup(name, ENV["#{name.to_s.upcase}_SPEC_URI"] || default_uri)
+    Object.const_set('ADAPTER', ENV['ADAPTER'].to_sym) if name.to_s == ENV['ADAPTER']
+    true
+  rescue Exception => e
+    if name.to_s == ENV['ADAPTER']
+      Object.const_set('ADAPTER', nil)
+      warn "Could not load #{name} adapter: #{e}"
+    end
+    false
+  end
+end
+
+ENV['ADAPTER'] ||= 'sqlite3'
+
+HAS_SQLITE3  = setup_adapter(:sqlite3,  'sqlite3::memory:')
+HAS_MYSQL    = setup_adapter(:mysql,    'mysql://localhost/dm_core_test')
+HAS_POSTGRES = setup_adapter(:postgres, 'postgres://postgres@localhost/dm_core_test')
+
+DataMapper::Logger.new(nil, :debug)
+
+class Article
+  include DataMapper::Resource
+
+  property :id,         Serial
+  property :blog_id,    Integer
+  property :created_at, DateTime
+  property :author,     String
+  property :title,      String
+end
+
+class Comment
+  include DataMapper::Resource
+end
+
+class NormalClass
+  # should not include DataMapper::Resource
+end
+
+# ==========================
+# Used for Association specs
+class Vehicle
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+
+  class << self
+    attr_accessor :mock_relationship
+  end
+end
+
+class Manufacturer
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+
+  class << self
+    attr_accessor :mock_relationship
+  end
+end
+
+class Supplier
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+end
+
+class Class
+  def publicize_methods
+    klass = class << self; self; end
+
+    saved_private_class_methods      = klass.private_instance_methods
+    saved_protected_class_methods    = klass.protected_instance_methods
+    saved_private_instance_methods   = self.private_instance_methods
+    saved_protected_instance_methods = self.protected_instance_methods
+
+    self.class_eval do
+      klass.send(:public, *saved_private_class_methods)
+      klass.send(:public, *saved_protected_class_methods)
+      public(*saved_private_instance_methods)
+      public(*saved_protected_instance_methods)
+    end
+
+    begin
+      yield
+    ensure
+      self.class_eval do
+        klass.send(:private, *saved_private_class_methods)
+        klass.send(:protected, *saved_protected_class_methods)
+        private(*saved_private_instance_methods)
+        protected(*saved_protected_instance_methods)
+      end
+    end
+  end
+end