sequel 4.36.0 → 5.61.0

data/spec/extensions/association_pks_spec.rb

@@ -1,405 +0,0 @@
-require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
-
-describe "Sequel::Plugins::AssociationPks" do
-  before do
-    @db = Sequel.mock(:autoid=>1, :fetch=>proc do |sql|
-      case sql
-      when /SELECT \* FROM (?:artists|albums) WHERE \(id = (\d+)\) LIMIT 1/
-        {:id=>$1.to_i}
-      when "SELECT id FROM albums WHERE (albums.artist_id = 1)"
-        [{:id=>1}, {:id=>2}, {:id=>3}]
-      when /SELECT tag_id FROM albums_tags WHERE \(album_id = (\d)\)/
-        a = []
-        a << {:tag_id=>1} if $1 == '1'
-        a << {:tag_id=>2} if $1 != '3'
-        a << {:tag_id=>3} if $1 == '2'
-        a
-      when "SELECT first, last FROM vocalists WHERE (vocalists.album_id = 1)"
-        [{:first=>"F1", :last=>"L1"}, {:first=>"F2", :last=>"L2"}]
-      when /SELECT first, last FROM albums_vocalists WHERE \(album_id = (\d)\)/
-        a = []
-        a << {:first=>"F1", :last=>"L1"} if $1 == '1'
-        a << {:first=>"F2", :last=>"L2"} if $1 != '3'
-        a << {:first=>"F3", :last=>"L3"} if $1 == '2'
-        a
-      when "SELECT id FROM instruments WHERE ((instruments.first = 'F1') AND (instruments.last = 'L1'))"
-        [{:id=>1}, {:id=>2}]
-      when /SELECT instrument_id FROM vocalists_instruments WHERE \(\((?:first|last) = '?[FL1](\d)/
-        a = []
-        a << {:instrument_id=>1} if $1 == "1"
-        a << {:instrument_id=>2} if $1 != "3"
-        a << {:instrument_id=>3} if $1 == "2"
-        a
-      when "SELECT year, week FROM hits WHERE ((hits.first = 'F1') AND (hits.last = 'L1'))"
-        [{:year=>1997, :week=>1}, {:year=>1997, :week=>2}]
-      when /SELECT year, week FROM vocalists_hits WHERE \(\((?:first|last) = '?[FL1](\d)/
-        a = []
-        a << {:year=>1997, :week=>1} if $1 == "1"
-        a << {:year=>1997, :week=>2} if $1 != "3"
-        a << {:year=>1997, :week=>3} if $1 == "2"
-        a
-      end
-    end)
-    @Artist = Class.new(Sequel::Model(@db[:artists]))
-    @Artist.columns :id
-    @Album = Class.new(Sequel::Model(@db[:albums]))
-    @Album.columns :id, :artist_id
-    @Tag = Class.new(Sequel::Model(@db[:tags]))
-    @Tag.columns :id
-    @Vocalist = Class.new(Sequel::Model(@db[:vocalists]))
-    @Vocalist.columns :first, :last, :album_id
-    @Vocalist.set_primary_key [:first, :last]
-    @Instrument = Class.new(Sequel::Model(@db[:instruments]))
-    @Instrument.columns :id, :first, :last
-    @Hit = Class.new(Sequel::Model(@db[:hits]))
-    @Hit.columns :year, :week, :first, :last
-    @Hit.set_primary_key [:year, :week]
-    @Artist.plugin :association_pks
-    @Album.plugin :association_pks
-    @Vocalist.plugin :association_pks
-    @Artist.one_to_many :albums, :class=>@Album, :key=>:artist_id
-    @Album.many_to_many :tags, :class=>@Tag, :join_table=>:albums_tags, :left_key=>:album_id
-    @db.sqls
-  end
-
-  it "should return correct associated pks for one_to_many associations" do
-    @Artist.load(:id=>1).album_pks.must_equal [1,2,3]
-    @Artist.load(:id=>2).album_pks.must_equal []
-  end
-
-  it "should return correct associated pks for many_to_many associations" do
-    @Album.load(:id=>1).tag_pks.must_equal [1, 2]
-    @Album.load(:id=>2).tag_pks.must_equal [2, 3]
-    @Album.load(:id=>3).tag_pks.must_equal []
-  end
-
-  it "should set associated pks correctly for a one_to_many association" do
-    @Artist.load(:id=>1).album_pks = [1, 2]
-    @db.sqls.must_equal ["UPDATE albums SET artist_id = 1 WHERE (id IN (1, 2))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN (1, 2)))"]
-  end
-
-  it "should use associated class's primary key for a one_to_many association" do
-    @Album.set_primary_key :foo
-    @Artist.load(:id=>1).album_pks = [1, 2]
-    @db.sqls.must_equal ["UPDATE albums SET artist_id = 1 WHERE (foo IN (1, 2))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (foo NOT IN (1, 2)))"]
-  end
-
-  it "should set associated pks correctly for a many_to_many association" do
-    @Album.load(:id=>2).tag_pks = [1, 3]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM albums_tags WHERE ((album_id = 2) AND (tag_id NOT IN (1, 3)))"
-    sqls[1].must_equal 'SELECT tag_id FROM albums_tags WHERE (album_id = 2)'
-    sqls[2].must_equal 'BEGIN'
-    sqls[3].must_match(/INSERT INTO albums_tags \((album_id, tag_id|tag_id, album_id)\) VALUES \((2, 1|1, 2)\)/)
-    sqls[4].must_equal 'COMMIT'
-    sqls.length.must_equal 5
-  end
-
-  it "should return correct right-side associated cpks for one_to_many associations" do
-    @Album.one_to_many :vocalists, :class=>@Vocalist, :key=>:album_id
-    @Album.load(:id=>1).vocalist_pks.must_equal [["F1", "L1"], ["F2", "L2"]]
-    @Album.load(:id=>2).vocalist_pks.must_equal []
-  end
-
-  it "should return correct right-side associated cpks for many_to_many associations" do
-    @Album.many_to_many :vocalists, :class=>@Vocalist, :join_table=>:albums_vocalists, :left_key=>:album_id, :right_key=>[:first, :last]
-    @Album.load(:id=>1).vocalist_pks.must_equal [["F1", "L1"], ["F2", "L2"]]
-    @Album.load(:id=>2).vocalist_pks.must_equal [["F2", "L2"], ["F3", "L3"]]
-    @Album.load(:id=>3).vocalist_pks.must_equal []
-  end
-
-  it "should set associated right-side cpks correctly for a one_to_many association" do
-    @Album.one_to_many :vocalists, :class=>@Vocalist, :key=>:album_id
-    @Album.load(:id=>1).vocalist_pks = [["F1", "L1"], ["F2", "L2"]]
-    @db.sqls.must_equal ["UPDATE vocalists SET album_id = 1 WHERE ((first, last) IN (('F1', 'L1'), ('F2', 'L2')))",
-      "UPDATE vocalists SET album_id = NULL WHERE ((vocalists.album_id = 1) AND ((first, last) NOT IN (('F1', 'L1'), ('F2', 'L2'))))"]
-  end
-
-  it "should set associated right-side cpks correctly for a many_to_many association" do
-    @Album.many_to_many :vocalists, :class=>@Vocalist, :join_table=>:albums_vocalists, :left_key=>:album_id, :right_key=>[:first, :last]
-    @Album.load(:id=>2).vocalist_pks = [["F1", "L1"], ["F2", "L2"]]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM albums_vocalists WHERE ((album_id = 2) AND ((first, last) NOT IN (('F1', 'L1'), ('F2', 'L2'))))"
-    sqls[1].must_equal 'SELECT first, last FROM albums_vocalists WHERE (album_id = 2)'
-    match = sqls[3].match(/INSERT INTO albums_vocalists \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"'F1'", "last"=>"'L1'", "album_id"=>"2")
-    sqls.length.must_equal 5
-  end
-
-  it "should return correct associated pks for left-side cpks for one_to_many associations" do
-    @Vocalist.one_to_many :instruments, :class=>@Instrument, :key=>[:first, :last]
-    @Vocalist.load(:first=>'F1', :last=>'L1').instrument_pks.must_equal [1, 2]
-    @Vocalist.load(:first=>'F2', :last=>'L2').instrument_pks.must_equal []
-  end
-
-  it "should return correct associated pks for left-side cpks for many_to_many associations" do
-    @Vocalist.many_to_many :instruments, :class=>@Instrument, :join_table=>:vocalists_instruments, :left_key=>[:first, :last]
-    @Vocalist.load(:first=>'F1', :last=>'L1').instrument_pks.must_equal [1, 2]
-    @Vocalist.load(:first=>'F2', :last=>'L2').instrument_pks.must_equal [2, 3]
-    @Vocalist.load(:first=>'F3', :last=>'L3').instrument_pks.must_equal []
-  end
-
-  it "should set associated pks correctly for left-side cpks for a one_to_many association" do
-    @Vocalist.one_to_many :instruments, :class=>@Instrument, :key=>[:first, :last]
-    @Vocalist.load(:first=>'F1', :last=>'L1').instrument_pks = [1, 2]
-    sqls = @db.sqls
-    sqls[0].must_match(/UPDATE instruments SET (first = 'F1', last = 'L1'|last = 'L1', first = 'F1') WHERE \(id IN \(1, 2\)\)/)
-    sqls[1].must_match(/UPDATE instruments SET (first = NULL, last = NULL|last = NULL, first = NULL) WHERE \(\(instruments.first = 'F1'\) AND \(instruments.last = 'L1'\) AND \(id NOT IN \(1, 2\)\)\)/)
-    sqls.length.must_equal 2
-  end
-
-  it "should set associated pks correctly for left-side cpks for a many_to_many association" do
-    @Vocalist.many_to_many :instruments, :class=>@Instrument, :join_table=>:vocalists_instruments, :left_key=>[:first, :last]
-    @Vocalist.load(:first=>'F2', :last=>'L2').instrument_pks = [1, 2]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM vocalists_instruments WHERE ((first = 'F2') AND (last = 'L2') AND (instrument_id NOT IN (1, 2)))"
-    sqls[1].must_equal "SELECT instrument_id FROM vocalists_instruments WHERE ((first = 'F2') AND (last = 'L2'))"
-    match = sqls[3].match(/INSERT INTO vocalists_instruments \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"'F2'", "last"=>"'L2'", "instrument_id"=>"1")
-    sqls.length.must_equal 5
-  end
-
-  it "should return correct right-side associated cpks for left-side cpks for one_to_many associations" do
-    @Vocalist.one_to_many :hits, :class=>@Hit, :key=>[:first, :last]
-    @Vocalist.load(:first=>'F1', :last=>'L1').hit_pks.must_equal [[1997, 1], [1997, 2]]
-    @Vocalist.load(:first=>'F2', :last=>'L2').hit_pks.must_equal []
-  end
-
-  it "should return correct right-side associated cpks for left-side cpks for many_to_many associations" do
-    @Vocalist.many_to_many :hits, :class=>@Hit, :join_table=>:vocalists_hits, :left_key=>[:first, :last], :right_key=>[:year, :week]
-    @Vocalist.load(:first=>'F1', :last=>'L1').hit_pks.must_equal [[1997, 1], [1997, 2]]
-    @Vocalist.load(:first=>'F2', :last=>'L2').hit_pks.must_equal [[1997, 2], [1997, 3]]
-    @Vocalist.load(:first=>'F3', :last=>'L3').hit_pks.must_equal []
-  end
-
-  it "should set associated right-side cpks correctly for left-side cpks for a one_to_many association" do
-    @Vocalist.one_to_many :hits, :class=>@Hit, :key=>[:first, :last], :order=>:week
-    @Vocalist.load(:first=>'F1', :last=>'L1').hit_pks = [[1997, 1], [1997, 2]]
-    sqls = @db.sqls
-    sqls[0].must_match(/UPDATE hits SET (first = 'F1', last = 'L1'|last = 'L1', first = 'F1') WHERE \(\(year, week\) IN \(\(1997, 1\), \(1997, 2\)\)\)/)
-    sqls[1].must_match(/UPDATE hits SET (first = NULL, last = NULL|last = NULL, first = NULL) WHERE \(\(hits.first = 'F1'\) AND \(hits.last = 'L1'\) AND \(\(year, week\) NOT IN \(\(1997, 1\), \(1997, 2\)\)\)\)/)
-    sqls.length.must_equal 2
-  end
-
-  it "should set associated right-side cpks correctly for left-side cpks for a many_to_many association" do
-    @Vocalist.many_to_many :hits, :class=>@Hit, :join_table=>:vocalists_hits, :left_key=>[:first, :last], :right_key=>[:year, :week]
-    @Vocalist.load(:first=>'F2', :last=>'L2').hit_pks = [[1997, 1], [1997, 2]]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM vocalists_hits WHERE ((first = 'F2') AND (last = 'L2') AND ((year, week) NOT IN ((1997, 1), (1997, 2))))"
-    sqls[1].must_equal "SELECT year, week FROM vocalists_hits WHERE ((first = 'F2') AND (last = 'L2'))"
-    match = sqls[3].match(/INSERT INTO vocalists_hits \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"'F2'", "last"=>"'L2'", "year"=>"1997", "week"=>"1")
-    sqls.length.must_equal 5
-  end
-
-  it "should use transactions if the object is configured to use transactions" do
-    artist = @Artist.load(:id=>1)
-    artist.use_transactions = true
-    artist.album_pks = [1, 2]
-    @db.sqls.must_equal ["BEGIN",
-      "UPDATE albums SET artist_id = 1 WHERE (id IN (1, 2))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN (1, 2)))",
-      "COMMIT"]
-
-    album = @Album.load(:id=>2)
-    album.use_transactions = true
-    album.tag_pks = [1, 3]
-    sqls = @db.sqls
-    sqls[0].must_equal "BEGIN"
-    sqls[1].must_equal "DELETE FROM albums_tags WHERE ((album_id = 2) AND (tag_id NOT IN (1, 3)))"
-    sqls[2].must_equal 'SELECT tag_id FROM albums_tags WHERE (album_id = 2)'
-    sqls[3].must_match(/INSERT INTO albums_tags \((album_id, tag_id|tag_id, album_id)\) VALUES \((2, 1|1, 2)\)/)
-    sqls[4].must_equal "COMMIT"
-    sqls.length.must_equal 5
-  end
-
-  it "should automatically convert keys to numbers if the primary key is an integer for one_to_many associations" do
-    @Album.db_schema[:id][:type] = :integer
-    @Artist.load(:id=>1).album_pks = %w'1 2'
-    @db.sqls.must_equal ["UPDATE albums SET artist_id = 1 WHERE (id IN (1, 2))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN (1, 2)))"]
-  end
-
-  it "should not automatically convert keys if the primary key is not an integer for one_to_many associations" do
-    @Album.db_schema[:id][:type] = :string
-    @Artist.load(:id=>1).album_pks = %w'1 2'
-    @db.sqls.must_equal ["UPDATE albums SET artist_id = 1 WHERE (id IN ('1', '2'))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN ('1', '2')))"]
-  end
-
-  it "should automatically convert keys to numbers if the primary key is an integer for many_to_many associations" do
-    @Tag.db_schema[:id][:type] = :integer
-    @Album.load(:id=>2).tag_pks = %w'1 3'
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM albums_tags WHERE ((album_id = 2) AND (tag_id NOT IN (1, 3)))"
-    sqls[1].must_equal 'SELECT tag_id FROM albums_tags WHERE (album_id = 2)'
-    sqls[3].must_match(/INSERT INTO albums_tags \((album_id, tag_id|tag_id, album_id)\) VALUES \((2, 1|1, 2)\)/)
-    sqls.length.must_equal 5
-  end
-
-  it "should not automatically convert keys to numbers if the primary key is an integer for many_to_many associations" do
-    @Tag.db_schema[:id][:type] = :string
-    @Album.load(:id=>2).tag_pks = %w'1 3'
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM albums_tags WHERE ((album_id = 2) AND (tag_id NOT IN ('1', '3')))"
-    sqls[1].must_equal 'SELECT tag_id FROM albums_tags WHERE (album_id = 2)'
-    sqls[3].must_match(/INSERT INTO albums_tags \((album_id, tag_id|tag_id, album_id)\) VALUES \((2, '1'|'1', 2)\)/)
-    sqls[4].must_match(/INSERT INTO albums_tags \((album_id, tag_id|tag_id, album_id)\) VALUES \((2, '3'|'3', 2)\)/)
-    sqls.length.must_equal 6
-  end
-
-  it "should automatically convert keys to numbers for appropriate integer primary key for composite key associations" do
-    @Hit.db_schema[:year][:type] = :integer
-    @Hit.db_schema[:week][:type] = :integer
-    @Vocalist.many_to_many :hits, :class=>@Hit, :join_table=>:vocalists_hits, :left_key=>[:first, :last], :right_key=>[:year, :week]
-    @Vocalist.load(:first=>'F2', :last=>'L2').hit_pks = [['1997', '1'], ['1997', '2']]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM vocalists_hits WHERE ((first = 'F2') AND (last = 'L2') AND ((year, week) NOT IN ((1997, 1), (1997, 2))))"
-    sqls[1].must_equal "SELECT year, week FROM vocalists_hits WHERE ((first = 'F2') AND (last = 'L2'))"
-    match = sqls[3].match(/INSERT INTO vocalists_hits \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"'F2'", "last"=>"'L2'", "year"=>"1997", "week"=>"1")
-    sqls.length.must_equal 5
-
-    @Vocalist.db_schema[:first][:type] = :integer
-    @Vocalist.db_schema[:last][:type] = :integer
-    @Album.one_to_many :vocalists, :class=>@Vocalist, :key=>:album_id
-    @Album.load(:id=>1).vocalist_pks = [["11", "11"], ["12", "12"]]
-    @db.sqls.must_equal ["UPDATE vocalists SET album_id = 1 WHERE ((first, last) IN ((11, 11), (12, 12)))",
-      "UPDATE vocalists SET album_id = NULL WHERE ((vocalists.album_id = 1) AND ((first, last) NOT IN ((11, 11), (12, 12))))"]
-
-    @Album.many_to_many :vocalists, :class=>@Vocalist, :join_table=>:albums_vocalists, :left_key=>:album_id, :right_key=>[:first, :last]
-    @Album.load(:id=>2).vocalist_pks = [["11", "11"], ["12", "12"]]
-    sqls = @db.sqls
-    sqls[0].must_equal "DELETE FROM albums_vocalists WHERE ((album_id = 2) AND ((first, last) NOT IN ((11, 11), (12, 12))))"
-    sqls[1].must_equal 'SELECT first, last FROM albums_vocalists WHERE (album_id = 2)'
-    match = sqls[3].match(/INSERT INTO albums_vocalists \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"11", "last"=>"11", "album_id"=>"2")
-    match = sqls[4].match(/INSERT INTO albums_vocalists \((.*)\) VALUES \((.*)\)/)
-    Hash[match[1].split(', ').zip(match[2].split(', '))].must_equal("first"=>"12", "last"=>"12", "album_id"=>"2")
-    sqls.length.must_equal 6
-  end
-
-  it "should handle delaying setting of association pks until after saving for new objects, if :delay plugin option is used" do
-    @Artist.one_to_many :albums, :clone=>:albums, :delay_pks=>true
-    @Album.many_to_many :tags, :clone=>:tags, :delay_pks=>true
-    @Album.db_schema[:id][:type] = :integer
-
-    ar = @Artist.new
-    ar.album_pks.must_equal []
-    ar.album_pks = ["1","2","3"]
-    ar.album_pks.must_equal [1,2,3]
-    @db.sqls.must_equal []
-
-    ar.save
-    @db.sqls.must_equal [
-      "INSERT INTO artists DEFAULT VALUES",
-      "UPDATE albums SET artist_id = 1 WHERE (id IN (1, 2, 3))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN (1, 2, 3)))",
-      "SELECT * FROM artists WHERE (id = 1) LIMIT 1",
-    ]
-
-    al = @Album.new
-    al.tag_pks.must_equal []
-    al.tag_pks = [1,2]
-    al.tag_pks.must_equal [1, 2]
-    @db.sqls.must_equal []
-
-    al.save
-    @db.sqls.must_equal [
-      "INSERT INTO albums DEFAULT VALUES",
-      "DELETE FROM albums_tags WHERE ((album_id = 2) AND (tag_id NOT IN (1, 2)))",
-      "SELECT tag_id FROM albums_tags WHERE (album_id = 2)",
-      "BEGIN",
-      "INSERT INTO albums_tags (album_id, tag_id) VALUES (2, 1)",
-      "COMMIT",
-      "SELECT * FROM albums WHERE (id = 2) LIMIT 1"
-    ]
-  end
-
-  it "should handle delaying setting of association pks until after saving for existing objects, if :delay=>:all plugin option is used" do
-    @Artist.one_to_many :albums, :clone=>:albums, :delay_pks=>:always
-    @Album.many_to_many :tags, :clone=>:tags, :delay_pks=>:always
-    @Album.db_schema[:id][:type] = :integer
-
-    ar = @Artist.load(:id=>1)
-    ar.album_pks.must_equal [1,2,3]
-    @db.sqls
-    ar.album_pks = ["2","4"]
-    ar.album_pks.must_equal [2,4]
-    @db.sqls.must_equal []
-
-    ar.save_changes
-    @db.sqls.must_equal [
-      "UPDATE albums SET artist_id = 1 WHERE (id IN (2, 4))",
-      "UPDATE albums SET artist_id = NULL WHERE ((albums.artist_id = 1) AND (id NOT IN (2, 4)))"
-    ]
-
-    ar.album_pks = []
-    @db.sqls.must_equal []
-
-    ar.save_changes
-    @db.sqls.must_equal [
-      "UPDATE albums SET artist_id = NULL WHERE (artist_id = 1)"
-    ]
-
-    al = @Album.load(:id=>1)
-    al.tag_pks.must_equal [1,2]
-    @db.sqls
-    al.tag_pks = [2,3]
-    al.tag_pks.must_equal [2,3]
-    @db.sqls.must_equal []
-
-    al.save_changes
-    @db.sqls.must_equal [
-      "DELETE FROM albums_tags WHERE ((album_id = 1) AND (tag_id NOT IN (2, 3)))",
-      "SELECT tag_id FROM albums_tags WHERE (album_id = 1)",
-      "BEGIN",
-      "INSERT INTO albums_tags (album_id, tag_id) VALUES (1, 3)",
-      "COMMIT",
-    ]
-
-    al.tag_pks = []
-    @db.sqls.must_equal []
-
-    al.save_changes
-    @db.sqls.must_equal [
-      "DELETE FROM albums_tags WHERE (album_id = 1)",
-    ]
-  end
-
-  it "should clear delayed associated pks if refreshing, if :delay plugin option is used" do
-    @Artist.one_to_many :albums, :clone=>:albums, :delay_pks=>:always
-    @Album.many_to_many :tags, :clone=>:tags, :delay_pks=>:always
-
-    ar = @Artist.load(:id=>1)
-    ar.album_pks.must_equal [1,2,3]
-    ar.album_pks = [2,4]
-    ar.album_pks.must_equal [2,4]
-    ar.refresh
-    ar.album_pks.must_equal [1,2,3]
-  end
-
-  it "should remove all values if nil given to setter and :association_pks_nil=>:remove" do
-    @Artist.one_to_many :albums, :clone=>:albums, :association_pks_nil=>:remove
-
-    ar = @Artist.load(:id=>1)
-    ar.album_pks = nil
-    @db.sqls.must_equal ["UPDATE albums SET artist_id = NULL WHERE (artist_id = 1)"]
-  end
-
-  it "should take no action if nil given to setter and :association_pks_nil=>:ignore" do
-    @Artist.one_to_many :albums, :clone=>:albums, :association_pks_nil=>:ignore
-
-    ar = @Artist.load(:id=>1)
-    ar = @Artist.new
-    ar.album_pks = nil
-    @db.sqls.must_equal []
-  end
-
-  it "should raise error if nil given to setter by default" do
-    ar = @Artist.load(:id=>1)
-    proc{ar.album_pks = nil}.must_raise Sequel::Error
-  end
-end

data/spec/extensions/association_proxies_spec.rb

@@ -1,86 +0,0 @@
-require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
-
-describe "Sequel::Plugins::AssociationProxies" do
-  before do
-    class ::Tag < Sequel::Model
-    end
-    class ::Item < Sequel::Model
-      plugin :association_proxies
-      many_to_many :tags, :extend=>Module.new{def size; count end}
-    end
-    @i = Item.load(:id=>1)
-    @t = @i.tags
-    Item.db.reset
-  end 
-  after do
-    Object.send(:remove_const, :Tag)
-    Object.send(:remove_const, :Item)
-  end
-
-  it "should send method calls to the associated object array if sent an array method" do
-    @i.associations.has_key?(:tags).must_equal false
-    @t.select{|x| false}.must_equal []
-    @i.associations.has_key?(:tags).must_equal true
-  end
-  
-  it "should send method calls to the association dataset if sent a non-array method" do
-    @i.associations.has_key?(:tags).must_equal false
-    @t.filter(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-    @i.associations.has_key?(:tags).must_equal false
-  end
-  
-  it "should accept block to plugin to specify which methods to proxy to dataset" do
-    Item.plugin :association_proxies do |opts|
-      opts[:method] == :where || opts[:arguments].length == 2 || opts[:block]
-    end
-    @i.associations.has_key?(:tags).must_equal false
-    @t.where(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-    @t.filter('a = ?', 1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-    @t.filter{{:a=>1}}.sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-
-    @i.associations.has_key?(:tags).must_equal false
-    Item.plugin :association_proxies do |opts|
-      proxy_arg = opts[:proxy_argument]
-      proxy_block = opts[:proxy_block]
-      cached = opts[:instance].associations[opts[:reflection][:name]]
-      is_size = opts[:method] == :size
-      is_size && !cached && !proxy_arg && !proxy_block
-    end
-    @t.size.must_equal 1
-    Item.db.sqls.must_equal ["SELECT count(*) AS count FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1) LIMIT 1"]
-    @i.tags{|ds| ds}.size.must_equal 1
-    Item.db.sqls.must_equal ["SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1)"]
-    @i.tags(true).size.must_equal 1
-    Item.db.sqls.must_equal ["SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1)"]
-    @t.size.must_equal 1
-    Item.db.sqls.must_equal []
-  end
-  
-  it "should reload the cached association if sent an array method and the reload flag was given" do
-    @t.select{|x| false}.must_equal []
-    Item.db.sqls.length.must_equal 1
-    @t.select{|x| false}.must_equal []
-    Item.db.sqls.length.must_equal 0
-    @i.tags(true).select{|x| false}.must_equal []
-    Item.db.sqls.length.must_equal 1
-    @t.filter(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-    Item.db.sqls.length.must_equal 0
-  end
-  
-  it "should not return a proxy object for associations that do not return an array" do
-    Item.many_to_one :tag
-    proc{@i.tag.filter(:a=>1)}.must_raise(NoMethodError)
-    
-    Tag.one_to_one :item
-    proc{Tag.load(:id=>1, :item_id=>2).item.filter(:a=>1)}.must_raise(NoMethodError)
-  end
-
-  it "should work correctly in subclasses" do
-    i = Class.new(Item).load(:id=>1)
-    i.associations.has_key?(:tags).must_equal false
-    i.tags.select{|x| false}.must_equal []
-    i.associations.has_key?(:tags).must_equal true
-    i.tags.filter(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
-  end
-  
-end