colincasey-sequel 2.10.0 → 2.10.1

data/spec/sequel_core/expression_filters_spec.rb

@@ -0,0 +1,363 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+context "Blockless Ruby Filters" do
+  before do
+    db = Sequel::Database.new
+    db.quote_identifiers = false
+    @d = db[:items]
+    def @d.l(*args, &block)
+      literal(filter_expr(*args, &block))
+    end
+    def @d.lit(*args)
+      literal(*args)
+    end
+  end
+  
+  it "should support boolean columns directly" do
+    @d.l(:x).should == 'x'
+  end
+  
+  it "should support NOT via Symbol#~" do
+    @d.l(~:x).should == 'NOT x'
+  end
+  
+  it "should support qualified columns" do
+    @d.l(:x__y).should == 'x.y'
+    @d.l(~:x__y).should == 'NOT x.y'
+  end
+
+  it "should support NOT with SQL functions" do
+    @d.l(~:is_blah.sql_function).should == 'NOT is_blah()'
+    @d.l(~:is_blah.sql_function(:x)).should == 'NOT is_blah(x)'
+    @d.l(~:is_blah.sql_function(:x__y)).should == 'NOT is_blah(x.y)'
+    @d.l(~:is_blah.sql_function(:x, :x__y)).should == 'NOT is_blah(x, x.y)'
+  end
+
+  it "should handle multiple ~" do
+    @d.l(~~:x).should == 'x'
+    @d.l(~~~:x).should == 'NOT x'
+    @d.l(~~(:x & :y)).should == '(x AND y)'
+    @d.l(~~(:x | :y)).should == '(x OR y)'
+  end
+
+  it "should support = via Hash" do
+    @d.l(:x => 100).should == '(x = 100)'
+    @d.l(:x => 'a').should == '(x = \'a\')'
+    @d.l(:x => true).should == '(x = \'t\')'
+    @d.l(:x => false).should == '(x = \'f\')'
+    @d.l(:x => nil).should == '(x IS NULL)'
+    @d.l(:x => [1,2,3]).should == '(x IN (1, 2, 3))'
+  end
+  
+  it "should support != via Hash#~" do
+    @d.l(~{:x => 100}).should == '(x != 100)'
+    @d.l(~{:x => 'a'}).should == '(x != \'a\')'
+    @d.l(~{:x => true}).should == '(x != \'t\')'
+    @d.l(~{:x => false}).should == '(x != \'f\')'
+    @d.l(~{:x => nil}).should == '(x IS NOT NULL)'
+  end
+  
+  it "should support ~ via Hash and Regexp (if supported by database)" do
+    @d.l(:x => /blah/).should == '(x ~ \'blah\')'
+  end
+  
+  it "should support !~ via Hash#~ and Regexp" do
+    @d.l(~{:x => /blah/}).should == '(x !~ \'blah\')'
+  end
+  
+  it "should support LIKE via Symbol#like" do
+    @d.l(:x.like('a')).should == '(x LIKE \'a\')'
+    @d.l(:x.like(/a/)).should == '(x ~ \'a\')'
+    @d.l(:x.like('a', 'b')).should == '((x LIKE \'a\') OR (x LIKE \'b\'))'
+    @d.l(:x.like(/a/, /b/i)).should == '((x ~ \'a\') OR (x ~* \'b\'))'
+    @d.l(:x.like('a', /b/)).should == '((x LIKE \'a\') OR (x ~ \'b\'))'
+  end
+
+  it "should support NOT LIKE via Symbol#like and Symbol#~" do
+    @d.l(~:x.like('a')).should == '(x NOT LIKE \'a\')'
+    @d.l(~:x.like(/a/)).should == '(x !~ \'a\')'
+    @d.l(~:x.like('a', 'b')).should == '((x NOT LIKE \'a\') AND (x NOT LIKE \'b\'))'
+    @d.l(~:x.like(/a/, /b/i)).should == '((x !~ \'a\') AND (x !~* \'b\'))'
+    @d.l(~:x.like('a', /b/)).should == '((x NOT LIKE \'a\') AND (x !~ \'b\'))'
+  end
+
+  it "should support ILIKE via Symbol#ilike" do
+    @d.l(:x.ilike('a')).should == '(x ILIKE \'a\')'
+    @d.l(:x.ilike(/a/)).should == '(x ~* \'a\')'
+    @d.l(:x.ilike('a', 'b')).should == '((x ILIKE \'a\') OR (x ILIKE \'b\'))'
+    @d.l(:x.ilike(/a/, /b/i)).should == '((x ~* \'a\') OR (x ~* \'b\'))'
+    @d.l(:x.ilike('a', /b/)).should == '((x ILIKE \'a\') OR (x ~* \'b\'))'
+  end
+
+  it "should support NOT ILIKE via Symbol#ilike and Symbol#~" do
+    @d.l(~:x.ilike('a')).should == '(x NOT ILIKE \'a\')'
+    @d.l(~:x.ilike(/a/)).should == '(x !~* \'a\')'
+    @d.l(~:x.ilike('a', 'b')).should == '((x NOT ILIKE \'a\') AND (x NOT ILIKE \'b\'))'
+    @d.l(~:x.ilike(/a/, /b/i)).should == '((x !~* \'a\') AND (x !~* \'b\'))'
+    @d.l(~:x.ilike('a', /b/)).should == '((x NOT ILIKE \'a\') AND (x !~* \'b\'))'
+  end
+
+  it "should support negating ranges via Hash#~ and Range" do
+    @d.l(~{:x => 1..5}).should == '((x < 1) OR (x > 5))'
+    @d.l(~{:x => 1...5}).should == '((x < 1) OR (x >= 5))'
+  end
+  
+  it "should support negating NOT IN via Hash#~ and Dataset or Array" do
+    @d.l(~{:x => @d.select(:i)}).should == '(x NOT IN (SELECT i FROM items))'
+    @d.l(~{:x => [1,2,3]}).should == '(x NOT IN (1, 2, 3))'
+  end
+
+  it "should support + - * / via Symbol#+,-,*,/" do
+    @d.l(:x + 1 > 100).should == '((x + 1) > 100)'
+    @d.l((:x * :y) < 100.01).should == '((x * y) < 100.01)'
+    @d.l((:x - :y/2) >= 100000000000000000000000000000000000).should == '((x - (y / 2)) >= 100000000000000000000000000000000000)'
+    @d.l((((:x - :y)/(:x + :y))*:z) <= 100).should == '((((x - y) / (x + y)) * z) <= 100)'
+    @d.l(~((((:x - :y)/(:x + :y))*:z) <= 100)).should == '((((x - y) / (x + y)) * z) > 100)'
+  end
+  
+  it "should not allow negation of string expressions" do
+    proc{~:x.sql_string}.should raise_error
+    proc{~([:x, :y].sql_string_join)}.should raise_error
+  end
+
+  it "should not allow mathematical or string operations on true, false, or nil" do
+    proc{:x + 1}.should_not raise_error
+    proc{:x - true}.should raise_error(Sequel::Error)
+    proc{:x / false}.should raise_error(Sequel::Error)
+    proc{:x * nil}.should raise_error(Sequel::Error)
+    proc{[:x, nil].sql_string_join}.should raise_error(Sequel::Error)
+  end
+
+  it "should not allow mathematical or string operations on boolean complex expressions" do
+    proc{:x + (:y + 1)}.should_not raise_error
+    proc{:x - (~:y)}.should raise_error(Sequel::Error)
+    proc{:x / (:y & :z)}.should raise_error(Sequel::Error)
+    proc{:x * (:y | :z)}.should raise_error(Sequel::Error)
+    proc{:x + :y.like('a')}.should raise_error(Sequel::Error)
+    proc{:x - :y.like(/a/)}.should raise_error(Sequel::Error)
+    proc{:x * :y.like(/a/i)}.should raise_error(Sequel::Error)
+    proc{:x + ~:y.like('a')}.should raise_error(Sequel::Error)
+    proc{:x - ~:y.like(/a/)}.should raise_error(Sequel::Error)
+    proc{:x * ~:y.like(/a/i)}.should raise_error(Sequel::Error)
+    proc{[:x, ~:y.like(/a/i)].sql_string_join}.should raise_error(Sequel::Error)
+  end
+
+  it "should support AND conditions via &" do
+    @d.l(:x & :y).should == '(x AND y)'
+    @d.l(:x.sql_boolean & :y).should == '(x AND y)'
+    @d.l(:x & :y & :z).should == '((x AND y) AND z)'
+    @d.l(:x & {:y => :z}).should == '(x AND (y = z))'
+    @d.l({:y => :z} & :x).should == '((y = z) AND x)'
+    @d.l({:x => :a} & {:y => :z}).should == '((x = a) AND (y = z))'
+    @d.l((:x + 200 < 0) & (:y - 200 < 0)).should == '(((x + 200) < 0) AND ((y - 200) < 0))'
+    @d.l(:x & ~:y).should == '(x AND NOT y)'
+    @d.l(~:x & :y).should == '(NOT x AND y)'
+    @d.l(~:x & ~:y).should == '(NOT x AND NOT y)'
+  end
+  
+  it "should support OR conditions via |" do
+    @d.l(:x | :y).should == '(x OR y)'
+    @d.l(:x.sql_boolean | :y).should == '(x OR y)'
+    @d.l(:x | :y | :z).should == '((x OR y) OR z)'
+    @d.l(:x | {:y => :z}).should == '(x OR (y = z))'
+    @d.l({:y => :z} | :x).should == '((y = z) OR x)'
+    @d.l({:x => :a} | {:y => :z}).should == '((x = a) OR (y = z))'
+    @d.l((:x.sql_number > 200) | (:y.sql_number < 200)).should == '((x > 200) OR (y < 200))'
+  end
+  
+  it "should support & | combinations" do
+    @d.l((:x | :y) & :z).should == '((x OR y) AND z)'
+    @d.l(:x | (:y & :z)).should == '(x OR (y AND z))'
+    @d.l((:x & :w) | (:y & :z)).should == '((x AND w) OR (y AND z))'
+  end
+  
+  it "should support & | with ~" do
+    @d.l(~((:x | :y) & :z)).should == '((NOT x AND NOT y) OR NOT z)'
+    @d.l(~(:x | (:y & :z))).should == '(NOT x AND (NOT y OR NOT z))'
+    @d.l(~((:x & :w) | (:y & :z))).should == '((NOT x OR NOT w) AND (NOT y OR NOT z))'
+    @d.l(~((:x.sql_number > 200) | (:y & :z))).should == '((x <= 200) AND (NOT y OR NOT z))'
+  end
+  
+  it "should support LiteralString" do
+    @d.l('x'.lit).should == '(x)'
+    @d.l(~'x'.lit).should == 'NOT x'
+    @d.l(~~'x'.lit).should == 'x'
+    @d.l(~(('x'.lit | :y) & :z)).should == '((NOT x AND NOT y) OR NOT z)'
+    @d.l(~(:x | 'y'.lit)).should == '(NOT x AND NOT y)'
+    @d.l(~('x'.lit & 'y'.lit)).should == '(NOT x OR NOT y)'
+    @d.l({'y'.lit => 'z'.lit} & 'x'.lit).should == '((y = z) AND x)'
+    @d.l(('x'.lit > 200) & ('y'.lit < 200)).should == '((x > 200) AND (y < 200))'
+    @d.l(~('x'.lit + 1 > 100)).should == '((x + 1) <= 100)'
+    @d.l('x'.lit.like(/a/)).should == '(x ~ \'a\')'
+    @d.l('x'.lit + 1 > 100).should == '((x + 1) > 100)'
+    @d.l(('x'.lit * :y) < 100.01).should == '((x * y) < 100.01)'
+    @d.l(('x'.lit - :y/2) >= 100000000000000000000000000000000000).should == '((x - (y / 2)) >= 100000000000000000000000000000000000)'
+    @d.l(('z'.lit * (('x'.lit / :y)/(:x + :y))) <= 100).should == '((z * ((x / y) / (x + y))) <= 100)'
+    @d.l(~(((('x'.lit - :y)/(:x + :y))*:z) <= 100)).should == '((((x - y) / (x + y)) * z) > 100)'
+  end
+
+  it "should support hashes by ANDing the conditions" do
+    @d.l(:x => 100, :y => 'a')[1...-1].split(' AND ').sort.should == ['(x = 100)', '(y = \'a\')']
+    @d.l(:x => true, :y => false)[1...-1].split(' AND ').sort.should == ['(x = \'t\')', '(y = \'f\')']
+    @d.l(:x => nil, :y => [1,2,3])[1...-1].split(' AND ').sort.should == ['(x IS NULL)', '(y IN (1, 2, 3))']
+  end
+  
+  it "should support sql_negate on hashes" do
+    @d.l({:x => 100, :y => 'a'}.sql_negate)[1...-1].split(' AND ').sort.should == ['(x != 100)', '(y != \'a\')']
+    @d.l({:x => true, :y => false}.sql_negate)[1...-1].split(' AND ').sort.should == ['(x != \'t\')', '(y != \'f\')']
+    @d.l({:x => nil, :y => [1,2,3]}.sql_negate)[1...-1].split(' AND ').sort.should == ['(x IS NOT NULL)', '(y NOT IN (1, 2, 3))']
+  end
+  
+  it "should support ~ on hashes" do
+    @d.l(~{:x => 100, :y => 'a'})[1...-1].split(' OR ').sort.should == ['(x != 100)', '(y != \'a\')']
+    @d.l(~{:x => true, :y => false})[1...-1].split(' OR ').sort.should == ['(x != \'t\')', '(y != \'f\')']
+    @d.l(~{:x => nil, :y => [1,2,3]})[1...-1].split(' OR ').sort.should == ['(x IS NOT NULL)', '(y NOT IN (1, 2, 3))']
+  end
+  
+  it "should support sql_or on hashes" do
+    @d.l({:x => 100, :y => 'a'}.sql_or)[1...-1].split(' OR ').sort.should == ['(x = 100)', '(y = \'a\')']
+    @d.l({:x => true, :y => false}.sql_or)[1...-1].split(' OR ').sort.should == ['(x = \'t\')', '(y = \'f\')']
+    @d.l({:x => nil, :y => [1,2,3]}.sql_or)[1...-1].split(' OR ').sort.should == ['(x IS NULL)', '(y IN (1, 2, 3))']
+  end
+  
+  it "should support arrays with all two pairs the same as hashes" do
+    @d.l([[:x, 100],[:y, 'a']]).should == '((x = 100) AND (y = \'a\'))'
+    @d.l([[:x, true], [:y, false]]).should == '((x = \'t\') AND (y = \'f\'))'
+    @d.l([[:x, nil], [:y, [1,2,3]]]).should == '((x IS NULL) AND (y IN (1, 2, 3)))'
+  end
+  
+  it "should support sql_negate on arrays with all two pairs" do
+    @d.l([[:x, 100],[:y, 'a']].sql_negate).should == '((x != 100) AND (y != \'a\'))'
+    @d.l([[:x, true], [:y, false]].sql_negate).should == '((x != \'t\') AND (y != \'f\'))'
+    @d.l([[:x, nil], [:y, [1,2,3]]].sql_negate).should == '((x IS NOT NULL) AND (y NOT IN (1, 2, 3)))'
+  end
+  
+  it "should support ~ on arrays with all two pairs" do
+    @d.l(~[[:x, 100],[:y, 'a']]).should == '((x != 100) OR (y != \'a\'))'
+    @d.l(~[[:x, true], [:y, false]]).should == '((x != \'t\') OR (y != \'f\'))'
+    @d.l(~[[:x, nil], [:y, [1,2,3]]]).should == '((x IS NOT NULL) OR (y NOT IN (1, 2, 3)))'
+  end
+  
+  it "should support sql_or on arrays with all two pairs" do
+    @d.l([[:x, 100],[:y, 'a']].sql_or).should == '((x = 100) OR (y = \'a\'))'
+    @d.l([[:x, true], [:y, false]].sql_or).should == '((x = \'t\') OR (y = \'f\'))'
+    @d.l([[:x, nil], [:y, [1,2,3]]].sql_or).should == '((x IS NULL) OR (y IN (1, 2, 3)))'
+  end
+  
+  it "should support Array#sql_string_join for concatenation of SQL strings" do
+    @d.lit([:x].sql_string_join).should == '(x)'
+    @d.lit([:x].sql_string_join(', ')).should == '(x)'
+    @d.lit([:x, :y].sql_string_join).should == '(x || y)'
+    @d.lit([:x, :y].sql_string_join(', ')).should == "(x || ', ' || y)"
+    @d.lit([:x.sql_function(1), :y|1].sql_string_join).should == '(x(1) || y[1])'
+    @d.lit([:x.sql_function(1), 'y.z'.lit].sql_string_join(', ')).should == "(x(1) || ', ' || y.z)"
+    @d.lit([:x, 1, :y].sql_string_join).should == "(x || '1' || y)"
+    @d.lit([:x, 1, :y].sql_string_join(', ')).should == "(x || ', ' || '1' || ', ' || y)"
+    @d.lit([:x, 1, :y].sql_string_join(:y__z)).should == "(x || y.z || '1' || y.z || y)"
+    @d.lit([:x, 1, :y].sql_string_join(1)).should == "(x || '1' || '1' || '1' || y)"
+    @d.lit([:x, :y].sql_string_join('y.x || x.y'.lit)).should == "(x || y.x || x.y || y)"
+    @d.lit([[:x, :y].sql_string_join, [:a, :b].sql_string_join].sql_string_join).should == "((x || y) || (a || b))"
+  end
+
+  it "should support StringExpression#+ for concatenation of SQL strings" do
+    @d.lit(:x.sql_string + :y).should == '(x || y)'
+    @d.lit([:x].sql_string_join + :y).should == '((x) || y)'
+    @d.lit([:x, :z].sql_string_join(' ') + :y).should == "((x || ' ' || z) || y)"
+  end
+
+  it "should be supported inside blocks" do
+    @d.l{[[:x, nil], [:y, [1,2,3]]].sql_or}.should == '((x IS NULL) OR (y IN (1, 2, 3)))'
+    @d.l{~[[:x, nil], [:y, [1,2,3]]]}.should == '((x IS NOT NULL) OR (y NOT IN (1, 2, 3)))'
+    @d.l{~(((('x'.lit - :y)/(:x + :y))*:z) <= 100)}.should == '((((x - y) / (x + y)) * z) > 100)'
+    @d.l{{:x => :a} & {:y => :z}}.should == '((x = a) AND (y = z))'
+  end
+
+  it "should support &, |, ^, ~, <<, and >> for NumericExpressions" do
+    @d.l(:x.sql_number & 1 > 100).should == '((x & 1) > 100)'
+    @d.l(:x.sql_number | 1 > 100).should == '((x | 1) > 100)'
+    @d.l(:x.sql_number ^ 1 > 100).should == '((x ^ 1) > 100)'
+    @d.l(~:x.sql_number > 100).should == '(~x > 100)'
+    @d.l(:x.sql_number << 1 > 100).should == '((x << 1) > 100)'
+    @d.l(:x.sql_number >> 1 > 100).should == '((x >> 1) > 100)'
+    @d.l((:x + 1) & 1 > 100).should == '(((x + 1) & 1) > 100)'
+    @d.l((:x + 1) | 1 > 100).should == '(((x + 1) | 1) > 100)'
+    @d.l((:x + 1) ^ 1 > 100).should == '(((x + 1) ^ 1) > 100)'
+    @d.l(~(:x + 1) > 100).should == '(~(x + 1) > 100)'
+    @d.l((:x + 1) << 1 > 100).should == '(((x + 1) << 1) > 100)'
+    @d.l((:x + 1) >> 1 > 100).should == '(((x + 1) >> 1) > 100)'
+    @d.l((:x + 1) & (:x + 2) > 100).should == '(((x + 1) & (x + 2)) > 100)'
+  end
+
+  it "should raise an error if use a Bitwise method on a ComplexExpression that isn't a NumericExpression" do
+    proc{(:x + 1) & (:x & 2)}.should raise_error(Sequel::Error)
+  end
+
+  it "should raise an error if use a Boolean method on a ComplexExpression that isn't a BooleanExpression" do
+    proc{:x & (:x + 2)}.should raise_error(Sequel::Error)
+  end
+
+  it "should raise an error if attempting to invert a ComplexExpression that isn't a BooleanExpression" do
+    proc{Sequel::SQL::BooleanExpression.invert(:x + 2)}.should raise_error(Sequel::Error)
+  end
+
+  it "should return self on .lit" do
+    y = :x + 1
+    y.lit.should == y
+  end
+
+  it "should raise an error if trying to create an invalid complex expression" do
+    proc{Sequel::SQL::ComplexExpression.new(:BANG, 1, 2)}.should raise_error(Sequel::Error)
+  end
+
+  it "should raise an error if trying to literalize an invalid complex expression" do
+    ce = :x + 1
+    ce.instance_variable_set(:@op, :BANG)
+    proc{@d.lit(ce)}.should raise_error(Sequel::Error)
+  end
+
+  it "should support equality comparison of two expressions" do
+    e1 = ~:comment.like('%:hidden:%')
+    e2 = ~:comment.like('%:hidden:%')
+    e1.should == e2
+  end
+  
+  if RUBY_VERSION < '1.9.0'
+    it "should not allow inequality operations on true, false, or nil" do
+      proc{:x > 1}.should_not raise_error
+      proc{:x < true}.should raise_error(Sequel::Error)
+      proc{:x >= false}.should raise_error(Sequel::Error)
+      proc{:x <= nil}.should raise_error(Sequel::Error)
+    end
+
+    it "should not allow inequality operations on boolean complex expressions" do
+      proc{:x > (:y > 5)}.should raise_error(Sequel::Error)
+      proc{:x < (:y < 5)}.should raise_error(Sequel::Error)
+      proc{:x >= (:y >= 5)}.should raise_error(Sequel::Error)
+      proc{:x <= (:y <= 5)}.should raise_error(Sequel::Error)
+      proc{:x > {:y => nil}}.should raise_error(Sequel::Error)
+      proc{:x < ~{:y => nil}}.should raise_error(Sequel::Error)
+      proc{:x >= {:y => 5}}.should raise_error(Sequel::Error)
+      proc{:x <= ~{:y => 5}}.should raise_error(Sequel::Error)
+      proc{:x >= {:y => [1,2,3]}}.should raise_error(Sequel::Error)
+      proc{:x <= ~{:y => [1,2,3]}}.should raise_error(Sequel::Error)
+    end
+    
+    it "should support >, <, >=, and <= via Symbol#>,<,>=,<=" do
+      @d.l(:x > 100).should == '(x > 100)'
+      @d.l(:x < 100.01).should == '(x < 100.01)'
+      @d.l(:x >= 100000000000000000000000000000000000).should == '(x >= 100000000000000000000000000000000000)'
+      @d.l(:x <= 100).should == '(x <= 100)'
+    end
+    
+    it "should support negation of >, <, >=, and <= via Symbol#~" do
+      @d.l(~(:x > 100)).should == '(x <= 100)'
+      @d.l(~(:x < 100.01)).should == '(x >= 100.01)'
+      @d.l(~(:x >= 100000000000000000000000000000000000)).should == '(x < 100000000000000000000000000000000000)'
+      @d.l(~(:x <= 100)).should == '(x > 100)'
+    end
+    
+    it "should support double negation via ~" do
+      @d.l(~~(:x > 100)).should == '(x > 100)'
+    end
+  end
+end

data/spec/sequel_core/migration_spec.rb

@@ -0,0 +1,261 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+context "Migration classes" do
+  setup do
+    Sequel::Migration.descendants.clear
+  end
+
+  specify "should be registred in Migration.descendants" do
+    @class = Class.new(Sequel::Migration)
+    
+    Sequel::Migration.descendants.should == [@class]
+  end
+  
+  specify "should be registered in the right order" do
+    @c1 = Class.new(Sequel::Migration)
+    @c2 = Class.new(Sequel::Migration)
+    @c3 = Class.new(Sequel::Migration)
+    
+    Sequel::Migration.descendants.should == [@c1, @c2, @c3]
+  end
+end
+
+context "Migration#apply" do
+  setup do
+    @c = Class.new do
+      define_method(:one) {|x| [1111, x]}
+      define_method(:two) {|x| [2222, x]}
+    end
+    @db = @c.new
+    
+    @migration = Class.new(Sequel::Migration) do
+      define_method(:up) {one(3333)}
+      define_method(:down) {two(4444)}
+    end
+  end
+  
+  specify "should raise for an invalid direction" do
+    proc {@migration.apply(@db, :hahaha)}.should raise_error(ArgumentError)
+  end
+  
+  specify "should apply the up direction correctly" do
+    @migration.apply(@db, :up).should == [1111, 3333]
+  end
+
+  specify "should apply the down direction correctly" do
+    @migration.apply(@db, :down).should == [2222, 4444]
+  end
+end
+
+class DummyMigrationDataset
+  attr_reader :from
+  
+  def initialize(x); @from = x; end
+  
+  @@version = nil
+  
+  def version; @@version; end
+  def version=(x); @@version = x; end
+  def first; @@version ? {:version => @@version} : nil; end
+  def update(h); @@version = h[:version]; end
+  def <<(h); @@version = h[:version]; end
+end
+
+class DummyMigrationDB
+  attr_reader :creates, :drops, :table_created
+  
+  def initialize
+    @creates = []
+    @drops = []
+  end
+  
+  def create(x); @creates << x; end
+  def drop(x); @drops << x; end
+  
+  def [](x); DummyMigrationDataset.new(x); end
+  
+  def create_table(x); raise if @table_created == x; @table_created = x; end
+  def table_exists?(x); @table_created == x; end
+  
+  def transaction; yield; end
+end
+
+MIGRATION_001 = %[
+  class CreateSessions < Sequel::Migration
+    def up
+      create(1111)
+    end
+    
+    def down
+      drop(1111)
+    end
+  end
+]
+
+MIGRATION_002 = %[
+  class CreateNodes < Sequel::Migration
+    def up
+      create(2222)
+    end
+    
+    def down
+      drop(2222)
+    end
+  end
+]
+
+MIGRATION_003 = %[
+  class CreateUsers < Sequel::Migration
+    def up
+      create(3333)
+    end
+    
+    def down
+      drop(3333)
+    end
+  end
+]
+
+MIGRATION_005 = %[
+  class CreateAttributes < Sequel::Migration
+    def up
+      create(5555)
+    end
+    
+    def down
+      drop(5555)
+    end
+  end
+]
+
+context "Sequel::Migrator" do
+  setup do
+    @db = DummyMigrationDB.new
+    
+    File.open('001_create_sessions.rb', 'w') {|f| f << MIGRATION_001}
+    File.open('002_create_nodes.rb', 'w') {|f| f << MIGRATION_002}
+    File.open('003_create_users.rb', 'w') {|f| f << MIGRATION_003}
+    File.open('005_create_attributes.rb', 'w') {|f| f << MIGRATION_005}
+    
+    @db[:schema_info].version = nil
+  end
+  
+  teardown do
+    Object.send(:remove_const, "CreateSessions") if Object.const_defined?("CreateSessions")
+    Object.send(:remove_const, "CreateNodes") if Object.const_defined?("CreateNodes")
+    Object.send(:remove_const, "CreateUsers") if Object.const_defined?("CreateUsers")
+    Object.send(:remove_const, "CreateAttributes") if Object.const_defined?("CreateAttributes")
+
+    File.delete('001_create_sessions.rb')
+    File.delete('002_create_nodes.rb')
+    File.delete('003_create_users.rb')
+    File.delete('005_create_attributes.rb')
+  end
+  
+  specify "should return the list of files for a specified version range" do
+    Sequel::Migrator.migration_files('.', 1..1).should == \
+      ['./001_create_sessions.rb']
+
+    Sequel::Migrator.migration_files('.', 1..3).should == \
+      ['./001_create_sessions.rb', './002_create_nodes.rb', './003_create_users.rb']
+
+    Sequel::Migrator.migration_files('.', 3..5).should == \
+      ['./003_create_users.rb', './005_create_attributes.rb']
+      
+    Sequel::Migrator.migration_files('.', 7..8).should == []
+  end
+  
+  specify "should return the latest version available" do
+    Sequel::Migrator.latest_migration_version('.').should == 5
+  end
+  
+  specify "should load the migration classes for the specified range" do
+    Sequel::Migrator.migration_classes('.', 3, 0, :up).should == \
+      [CreateSessions, CreateNodes, CreateUsers]
+  end
+  
+  specify "should load the migration classes for the specified range" do
+    Sequel::Migrator.migration_classes('.', 0, 5, :down).should == \
+      [CreateAttributes, CreateUsers, CreateNodes, CreateSessions]
+  end
+  
+  specify "should start from current + 1 for the up direction" do
+    Sequel::Migrator.migration_classes('.', 3, 1, :up).should == \
+      [CreateNodes, CreateUsers]
+  end
+  
+  specify "should end on current + 1 for the down direction" do
+    Sequel::Migrator.migration_classes('.', 2, 5, :down).should == \
+      [CreateAttributes, CreateUsers]
+  end
+  
+  specify "should automatically create the schema_info table" do
+    @db.table_exists?(:schema_info).should be_false
+    Sequel::Migrator.schema_info_dataset(@db)
+    @db.table_exists?(:schema_info).should be_true
+    
+    # should not raise if table already exists
+    proc {Sequel::Migrator.schema_info_dataset(@db)}.should_not raise_error
+  end
+  
+  specify "should return a dataset for the schema_info table" do
+    d = Sequel::Migrator.schema_info_dataset(@db)
+    d.from.should == :schema_info
+  end
+  
+  specify "should get the migration version stored in the database" do
+    # default is 0
+    Sequel::Migrator.get_current_migration_version(@db).should == 0
+    
+    Sequel::Migrator.schema_info_dataset(@db) << {:version => 4321}
+
+    Sequel::Migrator.get_current_migration_version(@db).should == 4321
+  end
+  
+  specify "should set the migration version stored in the database" do
+    Sequel::Migrator.get_current_migration_version(@db).should == 0
+    Sequel::Migrator.set_current_migration_version(@db, 6666)
+    Sequel::Migrator.get_current_migration_version(@db).should == 6666
+  end
+  
+  specify "should apply migrations correctly in the up direction" do
+    Sequel::Migrator.apply(@db, '.', 3, 2)
+    @db.creates.should == [3333]
+    
+    Sequel::Migrator.get_current_migration_version(@db).should == 3
+
+    Sequel::Migrator.apply(@db, '.', 5)
+    @db.creates.should == [3333, 5555]
+
+    Sequel::Migrator.get_current_migration_version(@db).should == 5
+  end
+  
+  specify "should apply migrations correctly in the down direction" do
+    Sequel::Migrator.apply(@db, '.', 1, 5)
+    @db.drops.should == [5555, 3333, 2222]
+
+    Sequel::Migrator.get_current_migration_version(@db).should == 1
+  end
+
+  specify "should apply migrations up to the latest version if no target is given" do
+    Sequel::Migrator.apply(@db, '.')
+    @db.creates.should == [1111, 2222, 3333, 5555]
+
+    Sequel::Migrator.get_current_migration_version(@db).should == 5
+  end
+
+  specify "should apply migrations down to 0 version correctly" do
+    Sequel::Migrator.apply(@db, '.', 0, 5)
+    @db.drops.should == [5555, 3333, 2222, 1111]
+
+    Sequel::Migrator.get_current_migration_version(@db).should == 0
+  end
+  
+  specify "should return the target version" do
+    Sequel::Migrator.apply(@db, '.', 3, 2).should == 3
+
+    Sequel::Migrator.apply(@db, '.', 0).should == 0
+
+    Sequel::Migrator.apply(@db, '.').should == 5
+  end
+end