sequel 5.29.0 → 5.34.0

data/spec/extensions/touch_spec.rb

@@ -1,226 +0,0 @@
-require_relative "spec_helper"
-
-describe "Touch plugin" do
-  before do
-    @c = Class.new(Sequel::Model)
-    p = proc{def touch_instance_value; touch_association_value; end}
-    @Artist = Class.new(@c, &p).set_dataset(:artists)
-    @Album = Class.new(@c, &p).set_dataset(:albums)
-
-    @Artist.columns :id, :updated_at, :modified_on
-    @Artist.one_to_many :albums, :class=>@Album, :key=>:artist_id
-
-    @Album.columns :id, :updated_at, :modified_on, :artist_id, :original_album_id
-    @Album.one_to_many :followup_albums, :class=>@Album, :key=>:original_album_id
-    @Album.many_to_one :artist, :class=>@Artist
-
-    @a = @Artist.load(:id=>1)
-    DB.reset
-  end
-
-  it "should default to using Time.now when setting the column values for model instances" do
-    c = Class.new(Sequel::Model).set_dataset(:a)
-    c.plugin :touch
-    c.columns :id, :updated_at
-    c.load(:id=>1).touch
-    DB.sqls.first.must_match(/UPDATE a SET updated_at = '[-0-9 :.]+' WHERE \(id = 1\)/)
-  end
-
-  it "should work with current_datetime_timestamp extension" do
-    c = Class.new(Sequel::Model).set_dataset(:a)
-    c.dataset = c.dataset.extension(:current_datetime_timestamp)
-    c.plugin :touch
-    c.columns :id, :updated_at
-    c.load(:id=>1).touch
-    DB.sqls.must_equal ["UPDATE a SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should allow #touch instance method for updating the updated_at column" do
-    @Artist.plugin :touch
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should have #touch take an argument for the column to touch" do
-    @Artist.plugin :touch
-    @a.touch(:modified_on)
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should be able to specify the default column to touch in the plugin call using the :column option" do
-    @Artist.plugin :touch, :column=>:modified_on
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should be able to specify the default column to touch using the touch_column model accessor" do
-    @Artist.plugin :touch
-    @Artist.touch_column = :modified_on
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should be able to specify the associations to touch in the plugin call using the :associations option" do
-    @Artist.plugin :touch, :associations=>:albums
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-  end
-
-  it "should clear associations after touching them  :associations option" do
-    @Artist.plugin :touch, :associations=>:albums
-    @a.associations[:albums] = [@Album.call(:id=>1)]
-    @a.touch
-    @a.associations[:albums].must_be_nil
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-  end
-
-  it "should be able to give an array to the :associations option specifying multiple associations" do
-    @Album.plugin :touch, :associations=>[:artist, :followup_albums]
-    @Album.load(:id=>4, :artist_id=>1).touch
-    sqls = DB.sqls
-    sqls.shift.must_equal "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 4)"
-    sqls.sort.must_equal ["UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (albums.original_album_id = 4)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 1)"]
-  end
-
-  it "should be able to give a hash to the :associations option specifying the column to use for each association" do
-    @Artist.plugin :touch, :associations=>{:albums=>:modified_on}
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-  end
-
-  it "should default to using the touch_column as the default touch column for associations" do
-    @Artist.plugin :touch, :column=>:modified_on, :associations=>:albums
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-  end
-
-  it "should allow the mixed use of symbols and hashes inside an array for the :associations option" do
-    @Album.plugin :touch, :associations=>[:artist, {:followup_albums=>:modified_on}]
-    @Album.load(:id=>4, :artist_id=>1).touch
-    sqls = DB.sqls
-    sqls.shift.must_equal "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 4)"
-    sqls.sort.must_equal ["UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.original_album_id = 4)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 1)"]
-  end
-
-  it "should be able to specify the associations to touch via a touch_associations_method" do
-    @Album.plugin :touch
-    @Album.touch_associations(:artist, {:followup_albums=>:modified_on})
-    @Album.load(:id=>4, :artist_id=>1).touch
-    sqls = DB.sqls
-    sqls.shift.must_equal "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 4)"
-    sqls.sort.must_equal ["UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.original_album_id = 4)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 1)"]
-  end
-
-  it "should touch associated objects when destroying an object" do
-    @Album.plugin :touch
-    @Album.touch_associations(:artist, {:followup_albums=>:modified_on})
-    @Album.load(:id=>4, :artist_id=>1).destroy
-    sqls = DB.sqls
-    sqls.shift.must_equal "DELETE FROM albums WHERE id = 4"
-    sqls.sort.must_equal ["UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.original_album_id = 4)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 1)"]
-  end
-
-  it "should be able to touch many_to_one associations" do
-    @Album.plugin :touch, :associations=>:artist
-    @Album.load(:id=>3, :artist_id=>4).touch
-    DB.sqls.must_equal ["UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 3)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 4)"]
-  end
-
-  it "should be able to touch many_to_one associations" do
-    @Album.plugin :touch, :associations=>:artist
-    @Album.plugin :skip_create_refresh
-    @Album.create(:artist_id=>4)
-    DB.sqls.must_equal ["INSERT INTO albums (artist_id) VALUES (4)",
-      "UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (artists.id = 4)"]
-  end
-
-  it "should be able to touch one_to_one associations" do
-    @Artist.one_to_one :album, :class=>@Album, :key=>:artist_id
-    @Artist.plugin :touch, :associations=>:album
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-  end
-
-  it "should be able to touch many_to_many associations" do
-    @Artist.many_to_many :albums, :class=>@Album, :left_key=>:artist_id, :join_table=>:aa
-    @Artist.plugin :touch, :associations=>:albums
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "SELECT albums.* FROM albums INNER JOIN aa ON (aa.album_id = albums.id) WHERE (aa.artist_id = 1)",
-      "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should be able to touch many_through_many associations" do
-    @Artist.plugin :many_through_many
-    @Artist.many_through_many :albums, [[:aa, :artist_id, :album_id]], :class=>@Album
-    @Artist.plugin :touch, :associations=>:albums
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "SELECT albums.* FROM albums INNER JOIN aa ON (aa.album_id = albums.id) WHERE (aa.artist_id = 1)",
-      "UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)"]
-  end
-
-  it "should handle touching many_to_one associations with no associated object" do
-    @Album.plugin :touch, :associations=>:artist
-    @Album.load(:id=>3, :artist_id=>nil).touch
-    DB.sqls.must_equal ["UPDATE albums SET updated_at = CURRENT_TIMESTAMP WHERE (id = 3)"]
-  end
-
-  it "should not update a column that doesn't exist" do
-    @Album.plugin :touch, :column=>:x
-    a = @Album.load(:id=>1)
-    a.touch
-    DB.sqls.must_equal []
-    a.artist_id = 1
-    a.touch
-    DB.sqls.must_equal ['UPDATE albums SET artist_id = 1 WHERE (id = 1)']
-  end
-
-  it "should raise an error if given a column argument in touch that doesn't exist" do
-    @Artist.plugin :touch
-    proc{@a.touch(:x)}.must_raise(Sequel::MassAssignmentRestriction)
-  end
-
-  it "should raise an Error when a nonexistent association is given" do
-    @Artist.plugin :touch
-    proc{@Artist.plugin :touch, :associations=>:blah}.must_raise(Sequel::Error)
-  end
-
-  it "should work correctly in subclasses" do
-    @Artist.plugin :touch
-    c1 = Class.new(@Artist)
-    c1.load(:id=>4).touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 4)"]
-
-    c1.touch_column = :modified_on
-    c1.touch_associations :albums
-    c1.load(:id=>1).touch
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 1)",
-      "UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.artist_id = 1)"]
-
-    @a.touch
-    DB.sqls.must_equal ["UPDATE artists SET updated_at = CURRENT_TIMESTAMP WHERE (id = 1)"]
-
-    @Artist.plugin :touch, :column=>:modified_on, :associations=>:albums
-    c2 = Class.new(@Artist)
-    c2.load(:id=>4).touch
-    DB.sqls.must_equal ["UPDATE artists SET modified_on = CURRENT_TIMESTAMP WHERE (id = 4)",
-      "UPDATE albums SET modified_on = CURRENT_TIMESTAMP WHERE (albums.artist_id = 4)"]
-  end
-
-  it "should freeze touched associations when freezing model class" do
-    @Artist.plugin :touch, :associations=>:albums
-    @Artist.freeze
-    @Artist.touched_associations.frozen?.must_equal true
-  end
-end

data/spec/extensions/tree_spec.rb

@@ -1,334 +0,0 @@
-require_relative "spec_helper"
-
-describe Sequel::Model, "tree plugin" do
-  def klass(opts={})
-    @db = DB
-    c = Class.new(Sequel::Model(@db[:nodes]))
-    c.class_eval do
-      def self.name; 'Node'; end
-      columns :id, :name, :parent_id, :i, :pi
-      plugin :tree, opts
-    end
-    c
-  end
-
-  before do
-    @c = klass
-    @ds = @c.dataset
-    @o = @c.load(:id=>2, :parent_id=>1, :name=>'AA', :i=>3, :pi=>4)
-    @db.reset
-  end
-
-  it "should define the correct associations" do
-    @c.associations.sort_by{|x| x.to_s}.must_equal [:children, :parent]
-  end
-  
-  it "should define the correct reciprocals" do
-    @c.associations.sort_by{|x| x.to_s}.map{|x| @c.association_reflection(x).reciprocal}.must_equal [:parent, :children]
-  end
-  
-  it "should define the correct associations when giving options" do
-    klass(:children=>{:name=>:cs}, :parent=>{:name=>:p}).associations.sort_by{|x| x.to_s}.must_equal [:cs, :p]
-  end
-
-  it "should use the correct SQL for lazy associations" do
-    @o.parent_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.id = 1) LIMIT 1'
-    @o.children_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.parent_id = 2)'
-  end
-  
-  it "should use the correct SQL for lazy associations when giving options" do
-    o = klass(:primary_key=>:i, :key=>:pi, :order=>:name, :children=>{:name=>:cs}, :parent=>{:name=>:p}).load(:id=>2, :parent_id=>1, :name=>'AA', :i=>3, :pi=>4)
-    o.p_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.i = 4) ORDER BY name LIMIT 1'
-    o.cs_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.pi = 3) ORDER BY name'
-  end
-
-  it "should have parent_column give the symbol of the parent column" do
-    @c.parent_column.must_equal :parent_id
-    klass(:key=>:p_id).parent_column.must_equal :p_id
-  end
-
-  it "should have tree_order give the order of the association" do
-    @c.tree_order.must_be_nil
-    klass(:order=>:name).tree_order.must_equal :name
-    klass(:order=>[:parent_id, :name]).tree_order.must_equal [:parent_id, :name]
-  end
-
-  it "should work correctly in subclasses" do
-    o = Class.new(klass(:primary_key=>:i, :key=>:pi, :order=>:name, :children=>{:name=>:cs}, :parent=>{:name=>:p})).load(:id=>2, :parent_id=>1, :name=>'AA', :i=>3, :pi=>4)
-    o.p_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.i = 4) ORDER BY name LIMIT 1'
-    o.cs_dataset.sql.must_equal 'SELECT * FROM nodes WHERE (nodes.pi = 3) ORDER BY name'
-  end
-
-  it "should have roots return an array of the tree's roots" do
-    @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :name=>'r'}])
-    @c.roots.must_equal [@c.load(:id=>1, :parent_id=>nil, :name=>'r')]
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE (parent_id IS NULL)"]
-    @c.exclude(id: 2).roots.must_equal [@c.load(:id=>1, :parent_id=>nil, :name=>'r')]
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE ((id != 2) AND (parent_id IS NULL))"]
-  end
-
-  it "should have roots_dataset be a dataset representing the tree's roots" do
-    @c.roots_dataset.sql.must_equal "SELECT * FROM nodes WHERE (parent_id IS NULL)"
-    @c.exclude(id: 2).roots_dataset.sql.must_equal "SELECT * FROM nodes WHERE ((id != 2) AND (parent_id IS NULL))"
-  end
-
-  it "should have ancestors return the ancestors of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :parent_id=>5, :name=>'r'}], [{:id=>5, :parent_id=>nil, :name=>'r2'}]])
-    @o.ancestors.must_equal [@c.load(:id=>1, :parent_id=>5, :name=>'r'), @c.load(:id=>5, :parent_id=>nil, :name=>'r2')]
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE id = 1",
-      "SELECT * FROM nodes WHERE id = 5"]
-  end
-
-  it "should have descendants return the descendants of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>3, :parent_id=>2, :name=>'r'}, {:id=>4, :parent_id=>2, :name=>'r2'}], [{:id=>5, :parent_id=>4, :name=>'r3'}], []])
-    @o.descendants.must_equal [@c.load(:id=>3, :parent_id=>2, :name=>'r'), @c.load(:id=>4, :parent_id=>2, :name=>'r2'), @c.load(:id=>5, :parent_id=>4, :name=>'r3')] 
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE (nodes.parent_id = 2)",
-      "SELECT * FROM nodes WHERE (nodes.parent_id = 3)",
-      "SELECT * FROM nodes WHERE (nodes.parent_id = 5)",
-      "SELECT * FROM nodes WHERE (nodes.parent_id = 4)"]
-  end
-
-  it "should have root return the root of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :parent_id=>5, :name=>'r'}], [{:id=>5, :parent_id=>nil, :name=>'r2'}]])
-    @o.root.must_equal @c.load(:id=>5, :parent_id=>nil, :name=>'r2')
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE id = 1",
-      "SELECT * FROM nodes WHERE id = 5"]
-  end
-
-  it "should have root? return true for a root node and false for a child node" do
-    @c.load(:parent_id => nil).root?.must_equal true
-    @c.load(:parent_id => 1).root?.must_equal false
-  end
-
-  it "should have root? return false for an new node" do
-    @c.new.root?.must_equal false
-  end
-
-  it "should have self_and_siblings return the children of the current node's parent" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :parent_id=>3, :name=>'r'}], [{:id=>7, :parent_id=>1, :name=>'r2'}, @o.values.dup]])
-    @o.self_and_siblings.must_equal [@c.load(:id=>7, :parent_id=>1, :name=>'r2'), @o] 
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE id = 1",
-      "SELECT * FROM nodes WHERE (nodes.parent_id = 1)"]
-  end
-
-  it "should have self_and_siblings return the roots if the current object is a root" do
-    h = {:id=>2, :parent_id=>nil, :name=>'AA'}
-    @c.dataset = @c.dataset.with_fetch(h)
-    @c.load(h).self_and_siblings.must_equal [@c.load(h)]
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE (parent_id IS NULL)"]
-  end
-
-  it "should have siblings return the children of the current node's parent, except for the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :parent_id=>3, :name=>'r'}], [{:id=>7, :parent_id=>1, :name=>'r2'}, @o.values.dup]])
-    @o.siblings.must_equal [@c.load(:id=>7, :parent_id=>1, :name=>'r2')] 
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE id = 1",
-      "SELECT * FROM nodes WHERE (nodes.parent_id = 1)"]
-  end
-
-  it "should have methods work correctly with custom association names" do
-    o = klass(:primary_key=>:i, :key=>:pi, :order=>:name, :children=>{:name=>:cs}, :parent=>{:name=>:p}).load(:id=>2, :parent_id=>1, :name=>'AA', :i=>3, :pi=>4)
-    
-    o.model.parent_association_name.must_equal :p
-    o.model.children_association_name.must_equal :cs
-    o.model.dataset = o.model.dataset.with_fetch(lambda do |sql|
-      case sql
-      when "SELECT * FROM nodes WHERE (nodes.i = 4) ORDER BY name LIMIT 1"
-        {:id=>7, :parent_id=>8, :name=>'r2', :i=>4, :pi=>5}
-      when "SELECT * FROM nodes WHERE (nodes.i = 5) ORDER BY name LIMIT 1"
-        {:id=>10, :parent_id=>11, :name=>'r3', :i=>5, :pi=>nil}
-      when 'SELECT * FROM nodes WHERE (nodes.pi = 3) ORDER BY name'
-        {:id=>12, :parent_id=>13, :name=>'r4', :i=>7, :pi=>3}
-      when 'SELECT * FROM nodes WHERE (nodes.pi = 7) ORDER BY name'
-        {:id=>14, :parent_id=>15, :name=>'r5', :i=>8, :pi=>7}
-      when 'SELECT * FROM nodes WHERE (nodes.pi = 8) ORDER BY name'
-        []
-      when 'SELECT * FROM nodes WHERE (nodes.pi = 4) ORDER BY name'
-        [{:id=>2, :parent_id=>1, :name=>'AA', :i=>3, :pi=>4}, {:id=>20, :parent_id=>21, :name=>'r6', :i=>9, :pi=>4}]
-      else
-        raise sql
-      end
-    end)
-    o.db.sqls.must_equal []
-
-    o.ancestors.must_equal [o.model.load(:id=>7, :parent_id=>8, :name=>'r2', :i=>4, :pi=>5),
-      o.model.load(:id=>10, :parent_id=>11, :name=>'r3', :i=>5, :pi=>nil)]
-    o.db.sqls.must_equal ["SELECT * FROM nodes WHERE (nodes.i = 4) ORDER BY name LIMIT 1",
-      "SELECT * FROM nodes WHERE (nodes.i = 5) ORDER BY name LIMIT 1"]
-
-    o.descendants.must_equal [o.model.load(:id=>12, :parent_id=>13, :name=>'r4', :i=>7, :pi=>3),
-      o.model.load(:id=>14, :parent_id=>15, :name=>'r5', :i=>8, :pi=>7)]
-    o.db.sqls.must_equal ["SELECT * FROM nodes WHERE (nodes.pi = 3) ORDER BY name",
-      "SELECT * FROM nodes WHERE (nodes.pi = 7) ORDER BY name",
-      "SELECT * FROM nodes WHERE (nodes.pi = 8) ORDER BY name"]
-
-    o.siblings.must_equal [o.model.load(:id=>20, :parent_id=>21, :name=>'r6', :i=>9, :pi=>4)] 
-    o.db.sqls.must_equal ["SELECT * FROM nodes WHERE (nodes.pi = 4) ORDER BY name"]
-  end
-
-  describe ":single_root option" do
-    before do
-      @c = klass(:single_root => true)
-    end
-
-    it "should have root class method return the root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :name=>'r'}])
-      @c.root.must_equal @c.load(:id=>1, :parent_id=>nil, :name=>'r')
-    end
-
-    it "prevents creating a second root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :name=>'r'}])
-      lambda { @c.create }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-    end
-
-    it "errors when promoting an existing record to a second root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :name=>'r'}])
-      n = @c.load(:id => 2, :parent_id => 1)
-      lambda { n.update(:parent_id => nil) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-    end
-
-    it "allows updating existing root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :name=>'r'}])
-       @c.root.update(:name => 'fdsa') 
-    end
-  end
-end
-
-describe Sequel::Model, "tree plugin with composite keys" do
-  def klass(opts={})
-    @db = DB
-    c = Class.new(Sequel::Model(@db[:nodes]))
-    c.class_eval do
-      def self.name; 'Node'; end
-      columns :id, :id2, :name, :parent_id, :parent_id2, :i, :pi
-      set_primary_key [:id, :id2]
-      plugin :tree, opts.merge(:key=>[:parent_id, :parent_id2])
-      def self.set_dataset(ds)
-        super
-        set_primary_key [:id, :id2]
-      end
-    end
-    c
-  end
-
-  before do
-    @c = klass
-    @ds = @c.dataset
-    @o = @c.load(:id=>2, :id2=>5, :parent_id=>1, :parent_id2=>6, :name=>'AA', :i=>3, :pi=>4)
-    @db.reset
-  end
-
-  
-  it "should use the correct SQL for lazy associations" do
-    @o.parent_dataset.sql.must_equal 'SELECT * FROM nodes WHERE ((nodes.id = 1) AND (nodes.id2 = 6)) LIMIT 1'
-    @o.children_dataset.sql.must_equal 'SELECT * FROM nodes WHERE ((nodes.parent_id = 2) AND (nodes.parent_id2 = 5))'
-  end
-  
-  it "should have parent_column give an array of symbols of the parent column" do
-    @c.parent_column.must_equal [:parent_id, :parent_id2]
-  end
-
-  it "should have roots return an array of the tree's roots" do
-    @c.dataset = @c.dataset.with_fetch([{:id=>1, :parent_id=>nil, :parent_id2=>nil, :name=>'r'}])
-    @c.roots.must_equal [@c.load(:id=>1, :parent_id=>nil, :parent_id2=>nil, :name=>'r')]
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE ((parent_id IS NULL) OR (parent_id2 IS NULL))"]
-  end
-
-  it "should have roots_dataset be a dataset representing the tree's roots" do
-    @c.roots_dataset.sql.must_equal "SELECT * FROM nodes WHERE ((parent_id IS NULL) OR (parent_id2 IS NULL))"
-  end
-
-  it "should have ancestors return the ancestors of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :id2=>6, :parent_id=>5, :parent_id2=>7, :name=>'r'}], [{:id=>5, :id2=>7, :parent_id=>nil, :parent_id2=>nil, :name=>'r2'}]])
-    @o.ancestors.must_equal [@c.load(:id=>1, :id2=>6, :parent_id=>5, :parent_id2=>7, :name=>'r'), @c.load(:id=>5, :id2=>7, :parent_id=>nil, :parent_id2=>nil, :name=>'r2')]
-    sqls = @db.sqls
-    sqls.length.must_equal 2
-    ["SELECT * FROM nodes WHERE ((id = 1) AND (id2 = 6)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 6) AND (id = 1)) LIMIT 1"].must_include(sqls[0])
-    ["SELECT * FROM nodes WHERE ((id = 5) AND (id2 = 7)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 7) AND (id = 5)) LIMIT 1"].must_include(sqls[1])
-  end
-
-  it "should have descendants return the descendants of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>3, :id2=>7, :parent_id=>2, :parent_id2=>5, :name=>'r'}, {:id=>4, :id2=>8, :parent_id=>2, :parent_id2=>5, :name=>'r2'}], [{:id=>5, :id2=>9, :parent_id=>4, :parent_id2=>8, :name=>'r3'}], []])
-    @o.descendants.must_equal [@c.load(:id=>3, :id2=>7, :parent_id=>2, :parent_id2=>5, :name=>'r'), @c.load(:id=>4, :id2=>8, :parent_id=>2, :parent_id2=>5, :name=>'r2'), @c.load(:id=>5, :id2=>9, :parent_id=>4, :parent_id2=>8, :name=>'r3')] 
-    @db.sqls.must_equal ["SELECT * FROM nodes WHERE ((nodes.parent_id = 2) AND (nodes.parent_id2 = 5))",
-      "SELECT * FROM nodes WHERE ((nodes.parent_id = 3) AND (nodes.parent_id2 = 7))",
-      "SELECT * FROM nodes WHERE ((nodes.parent_id = 5) AND (nodes.parent_id2 = 9))",
-      "SELECT * FROM nodes WHERE ((nodes.parent_id = 4) AND (nodes.parent_id2 = 8))"]
-  end
-
-  it "should have root return the root of the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :id2=>6, :parent_id=>5, :parent_id2=>7, :name=>'r'}], [{:id=>5, :id2=>7, :parent_id=>nil, :parent_id2=>nil, :name=>'r2'}]])
-    @o.root.must_equal @c.load(:id=>5, :id2=>7, :parent_id=>nil, :parent_id2=>nil, :name=>'r2')
-    sqls = @db.sqls
-    sqls.length.must_equal 2
-    ["SELECT * FROM nodes WHERE ((id = 1) AND (id2 = 6)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 6) AND (id = 1)) LIMIT 1"].must_include(sqls[0])
-    ["SELECT * FROM nodes WHERE ((id = 5) AND (id2 = 7)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 7) AND (id = 5)) LIMIT 1"].must_include(sqls[1])
-  end
-
-  it "should have root? return true for a root node and false for a child node" do
-    @c.load(:parent_id => nil, :parent_id2=>nil).root?.must_equal true
-    @c.load(:parent_id => 1, :parent_id2=>nil).root?.must_equal true
-    @c.load(:parent_id => nil, :parent_id2=>2).root?.must_equal true
-    @c.load(:parent_id => 1, :parent_id2=>2).root?.must_equal false
-  end
-
-  it "should have root? return false for an new node" do
-    @c.new.root?.must_equal false
-  end
-
-  it "should have self_and_siblings return the children of the current node's parent" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :id2=>6, :parent_id=>3, :parent_id2=>7, :name=>'r'}], [{:id=>7, :id2=>9, :parent_id=>1, :parent_id2=>6, :name=>'r2'}, @o.values.dup]])
-    @o.self_and_siblings.must_equal [@c.load(:id=>7, :id2=>9, :parent_id=>1, :parent_id2=>6, :name=>'r2'), @o] 
-    sqls = @db.sqls
-    sqls.length.must_equal 2
-    ["SELECT * FROM nodes WHERE ((id = 1) AND (id2 = 6)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 6) AND (id = 1)) LIMIT 1"].must_include(sqls[0])
-    sqls[1].must_equal "SELECT * FROM nodes WHERE ((nodes.parent_id = 1) AND (nodes.parent_id2 = 6))"
-  end
-
-  it "should have siblings return the children of the current node's parent, except for the current node" do
-    @c.dataset = @c.dataset.with_fetch([[{:id=>1, :id2=>6, :parent_id=>3, :parent_id2=>7, :name=>'r'}], [{:id=>7, :id2=>9, :parent_id=>1, :parent_id2=>6, :name=>'r2'}, @o.values.dup]])
-    @o.siblings.must_equal [@c.load(:id=>7, :id2=>9, :parent_id=>1, :parent_id2=>6, :name=>'r2')] 
-    sqls = @db.sqls
-    sqls.length.must_equal 2
-    ["SELECT * FROM nodes WHERE ((id = 1) AND (id2 = 6)) LIMIT 1", "SELECT * FROM nodes WHERE ((id2 = 6) AND (id = 1)) LIMIT 1"].must_include(sqls[0])
-    sqls[1].must_equal "SELECT * FROM nodes WHERE ((nodes.parent_id = 1) AND (nodes.parent_id2 = 6))"
-  end
-
-  describe ":single_root option" do
-    before do
-      @c = klass(:single_root => true)
-    end
-
-    it "prevents creating a second root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>nil, :name=>'r'}])
-      lambda { @c.create }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>1, :parent_id2=>nil, :name=>'r'}])
-      lambda { @c.create(:parent_id2=>1) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>2, :name=>'r'}])
-      lambda { @c.create(:parent_id=>2) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-    end
-
-    it "errors when promoting an existing record to a second root" do
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>nil, :name=>'r'}])
-      lambda { @c.load(:id => 2, :id2=>7, :parent_id => 1, :parent_id2=>2).update(:parent_id => nil, :parent_id2=>nil) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>1, :parent_id2=>nil, :name=>'r'}])
-      lambda { @c.load(:id => 2, :id2=>7, :parent_id => 1, :parent_id2=>2).update(:parent_id => nil) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-      @c.dataset = @c.dataset.with_fetch([{:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>2, :name=>'r'}])
-      lambda { @c.load(:id => 2, :id2=>7, :parent_id => 1, :parent_id2=>2).update(:parent_id2 => nil) }.must_raise(Sequel::Plugins::Tree::TreeMultipleRootError)
-    end
-
-    it "allows updating existing root" do
-      @c.dataset = @c.dataset.with_fetch(:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>nil, :name=>'r')
-      @c.root.update(:name => 'fdsa') 
-      @c.dataset = @c.dataset.with_fetch(:id=>1, :id2=>6, :parent_id=>1, :parent_id2=>nil, :name=>'r')
-      @c.root.update(:name => 'fdsa') 
-      @c.dataset = @c.dataset.with_fetch(:id=>1, :id2=>6, :parent_id=>nil, :parent_id2=>2, :name=>'r')
-      @c.root.update(:name => 'fdsa') 
-    end
-
-    it "freezes tree_order if it is an array" do
-      @c.tree_order = [:id]
-      @c.freeze
-      @c.tree_order.frozen?.must_equal true
-    end
-  end
-end