rdbi 0.9.0

data/test/test_rdbi.rb

@@ -0,0 +1,85 @@
+require 'helper'
+
+class TestRDBI < Test::Unit::TestCase
+  def test_01_connect
+    dbh = RDBI.connect(:Mock, :username => :foo, :password => :bar)
+    assert(dbh)
+    assert_kind_of(RDBI::Database, dbh)
+
+    dbh = RDBI.connect(RDBI::Driver::Mock, :username => :foo, :password => :bar)
+    assert(dbh)
+    assert_kind_of(RDBI::Database, dbh)
+
+    dbh = RDBI.connect("Mock", :username => :foo, :password => :bar)
+    assert(dbh)
+    assert_kind_of(RDBI::Database, dbh)
+
+    assert_raise(ArgumentError) { RDBI.connect(1, :user => :blah) }
+
+    assert_nothing_raised { dbh = RDBI.connect(:Mock) }
+  end
+
+  def test_02_last_dbh
+    dbh = mock_connect
+
+    assert(RDBI.last_dbh)
+    assert(dbh.object_id == RDBI.last_dbh.object_id)
+  end
+
+  def test_03_ping
+    assert_equal(10, RDBI.ping(:Mock, :some => :arg))
+    assert_equal(10, mock_connect.ping)
+  end
+
+  def test_04_connect_cached
+    dbh = RDBI.connect_cached(:Mock, :username => :foo)
+
+    assert(dbh)
+    assert_kind_of(RDBI::Database, dbh)
+    assert_equal(RDBI::Pool[:default].handles[0], dbh)
+
+    new_dbh = RDBI.connect_cached(:Mock, :username => :foo)
+
+    assert(new_dbh)
+    assert_kind_of(RDBI::Database, new_dbh)
+    assert_equal(RDBI::Pool[:default].handles[1], new_dbh)
+    assert_not_equal(dbh, new_dbh)
+
+    3.times { RDBI.connect_cached(:Mock, :username => :foo) }
+
+    assert_equal(dbh.object_id, RDBI.connect_cached(:Mock, :username => :foo).object_id)
+    assert_equal(new_dbh.object_id, RDBI.connect_cached(:Mock, :username => :foo).object_id)
+
+    # different pool
+
+    pool_dbh = RDBI.connect_cached(:Mock, :username => :foo, :pool_name => :foo)
+
+    assert_not_equal(dbh, pool_dbh)
+    assert_not_equal(new_dbh, pool_dbh)
+    assert_equal(RDBI::Pool[:foo].handles[0], pool_dbh)
+  end
+
+  def test_05_pool
+    dbh = RDBI.connect_cached(:Mock, :username => :foo, :pool_name => :test_05)
+    assert_equal(RDBI.pool(:test_05).handles[0], dbh)
+  end
+
+  def test_06_re_disconnect_all
+    RDBI.disconnect_all
+    connected_size = RDBI.all_connections.select(&:connected).size
+    assert_equal(0, connected_size)
+
+    total_size = RDBI.all_connections.size
+    RDBI.reconnect_all
+    connected_size = RDBI.all_connections.select(&:connected).size
+    assert_equal(connected_size, total_size)
+  end
+
+  def test_07_all_connections
+    total_size = RDBI.all_connections.size
+    mock_connect
+    assert_equal(RDBI.all_connections.size, total_size + 1)
+  end
+end
+
+# vim: syntax=ruby ts=2 et sw=2 sts=2

data/test/test_result.rb

@@ -0,0 +1,269 @@
+require 'helper'
+
+class TestResult < Test::Unit::TestCase
+  def setup
+    @dbh = mock_connect 
+  end
+
+  def teardown
+    @dbh.disconnect
+  end
+
+  def generate_data
+    RDBI::Driver::Mock::Cursor.new((0..9).to_a.map { |x| [x-1, x, x+1] })
+  end
+  
+  def mock_result
+    names = [:zero, :one, :two]
+    res = RDBI::Result.new(@dbh.prepare("foo"), [1], generate_data, RDBI::Schema.new((0..2).to_a.map { |x| RDBI::Column.new(names[x], :integer, :default) }), { :default => RDBI::Type.filterlist() })
+  end
+
+  def mock_empty_result
+    names = [:zero, :one, :two]
+    res = RDBI::Result.new(@dbh.prepare("foo"), [1], RDBI::Driver::Mock::Cursor.new([]), RDBI::Schema.new((0..2).to_a.map { |x| RDBI::Column.new(names[x], :integer, :default) }), { :default => RDBI::Type.filterlist() })
+  end
+
+  def get_index(res)
+    get_guts(get_guts(res)[:data])[:index]
+  end
+
+  def get_guts(res)
+    h = { }
+    %W[index schema binds sth data].collect(&:to_sym).each do |sym|
+      h[sym] = res.instance_variable_get("@#{sym}") || res.instance_variable_get("@#{sym}".to_sym)
+    end
+
+    return h
+  end
+
+  def test_01_init
+    res = mock_result
+    assert(res)
+    assert_kind_of(RDBI::Result, res)
+
+    guts = get_guts(res)
+
+    assert_kind_of(RDBI::Statement, guts[:sth])
+    assert_kind_of(RDBI::Statement, res.sth)
+    assert_equal(res.sth, guts[:sth])
+    res.sth.finish
+
+    assert_kind_of(RDBI::Schema, guts[:schema])
+    assert_kind_of(RDBI::Schema, res.schema)
+    assert_equal(res.schema, guts[:schema])
+
+    assert_kind_of(Array, guts[:binds])
+    assert_equal(res.binds, guts[:binds])
+    assert_equal([1], guts[:binds])
+  end
+
+  def test_02_responds
+    res = mock_result
+
+    %W[
+      schema
+      sth
+      driver
+      result_count
+      affected_count
+      complete
+      complete?
+      has_data
+      has_data?
+      each
+      inject
+      rewind
+      as
+      fetch
+      read
+      raw_fetch
+      finish 
+    ].collect(&:to_sym).each do |sym|
+      assert_respond_to(res, sym)
+    end
+
+    res.sth.finish
+  end
+
+  def test_03_fetch
+    res = mock_result
+    assert_equal([[-1,0,1]], res.fetch)
+    assert_equal(1, get_index(res))
+
+    assert_equal(generate_data[1..9], res.fetch(9))
+    assert_equal(10, get_index(res))
+    assert_equal([], res.fetch)
+    res.sth.finish
+
+    res = mock_result
+    assert_equal(generate_data.all, res.fetch(:all))
+    res.sth.finish
+
+    res = mock_result
+    res.fetch
+    assert_equal(generate_data.all, res.fetch(:all))
+    assert_equal(1, get_index(res))
+    assert_equal(generate_data[1..9], res.fetch(:rest))
+    assert_equal(10, get_index(res))
+
+    assert_equal(generate_data[0], res.fetch(:first))
+    assert_equal(generate_data[-1], res.fetch(:last))
+
+    res.sth.finish
+  end
+
+  def test_04_finish
+    res = mock_result
+    res.finish
+
+    guts = get_guts(res)
+
+    guts.values.each { |value| assert_nil(value) } 
+  end
+
+  def test_05_enumerable_and_index_predicates
+    res = mock_result
+
+    assert(res.has_data?)
+    assert(res.has_data)
+
+    assert(res.complete?)
+    assert(res.complete)
+
+    expected = generate_data
+    
+    res.each_with_index do |x, i|
+      assert_equal(expected[i], x)
+    end
+    
+    assert(res.complete?)
+    assert(res.complete)
+    assert(res.has_data?)
+    assert(res.has_data)
+    res.sth.finish
+  end
+
+  def test_06_as
+    res = mock_result
+    res.as(RDBI::Result::Driver::CSV)
+    assert_equal(
+      "-1,0,1\n",
+      res.fetch(1)
+    )
+
+    assert_equal(
+      "0,1,2\n1,2,3\n2,3,4\n3,4,5\n4,5,6\n5,6,7\n6,7,8\n7,8,9\n8,9,10\n",
+      res.fetch(:rest)
+    )
+
+    assert_equal(
+      "-1,0,1\n0,1,2\n1,2,3\n2,3,4\n3,4,5\n4,5,6\n5,6,7\n6,7,8\n7,8,9\n8,9,10\n",
+      res.fetch(:all)
+    )
+
+    # XXX reset intentionally.
+    res.as(:Array)
+    assert_equal([[-1, 0, 1]], res.fetch(1))
+
+    res.sth.finish
+    res.rewind
+
+    assert_equal(
+      "-1,0,1\n",
+      res.fetch(1, RDBI::Result::Driver::CSV)
+    )
+
+    # XXX this is intentional behavior, because I'm lazy. maybe we'll change it.
+    assert_equal(
+      "-1,0,1\n0,1,2\n1,2,3\n2,3,4\n3,4,5\n4,5,6\n5,6,7\n6,7,8\n7,8,9\n8,9,10\n",
+      res.fetch(:rest, RDBI::Result::Driver::CSV)
+    )
+
+    assert_equal(
+      "-1,0,1\n0,1,2\n1,2,3\n2,3,4\n3,4,5\n4,5,6\n5,6,7\n6,7,8\n7,8,9\n8,9,10\n",
+      res.fetch(:all, RDBI::Result::Driver::CSV)
+    )
+    res.sth.finish
+  end
+
+  def test_07_as_struct
+    res = mock_result
+    res.as(RDBI::Result::Driver::Struct)
+
+    results = res.fetch(1)
+    assert_kind_of(Array, results)
+    assert(results[0].kind_of?(::Struct)) 
+
+    hash = results[0]
+
+    assert_raises(NameError) { hash[:test] = "something" }
+    assert_raises(NameError) { hash['test'] = "something" }
+    assert_raises(NoMethodError) { hash.test = "something" }
+
+    assert_equal(-1, hash.zero)
+    assert_equal(0, hash.one)
+    assert_equal(1, hash.two)
+
+    hash = res.fetch(1)[0]
+
+    assert_equal(0, hash.zero)
+    assert_equal(1, hash.one)
+    assert_equal(2, hash.two)
+    
+    hash = res.fetch(:first)
+   
+    assert_equal(-1, hash.zero)
+    assert_equal(0, hash.one)
+    assert_equal(1, hash.two)
+    
+    res.sth.finish
+
+    res = mock_empty_result
+
+    hash = res.fetch(:first, :Struct)
+    assert_nil(hash)
+    res.sth.finish
+  end
+
+  def test_08_reload
+    res = mock_result 
+
+    assert_equal([-1, 0, 1], res.fetch(1)[0])
+    assert_equal(10, res.result_count)
+    assert_equal([:zero, :one, :two], res.schema.columns.map(&:name))
+
+    res.reload
+
+    # this will actually come back from the Mock driver, which will be
+    # completely different.  not the best test, but it gets the job done.
+    assert_equal(%W[10], res.fetch(1)[0])
+    assert_equal(5, res.result_count)
+    assert_equal((0..9).to_a, res.schema.columns.map(&:name))
+    res.finish
+  end
+
+  def test_09_insert_results
+    dbh = mock_connect
+    sth = dbh.prepare('insert into blah (foo, bar) values (?, ?)')
+    sth.result = [[1,2,3], [1,2,3]]
+    sth.affected_count = 10
+
+    res = sth.execute(1,2)
+
+    assert_equal(16, res.affected_count)
+    assert_equal(2,  res.result_count)
+    sth.finish
+  end
+
+  def test_10_null_results
+    res = RDBI::Result.new(@dbh.prepare("foo"), [1], RDBI::Driver::Mock::Cursor.new([]), [], 0)
+    assert_equal(nil, res.fetch(:first))
+    assert_equal(nil, res.fetch(:last))
+    assert_equal([], res.fetch(:all))
+    assert_equal([], res.fetch(:rest))
+    assert_equal([], res.fetch(1))
+    res.sth.finish
+  end
+end
+
+# vim: syntax=ruby ts=2 et sw=2 sts=2

data/test/test_statement.rb

@@ -0,0 +1,66 @@
+require 'helper'
+
+class TestStatement < Test::Unit::TestCase
+  attr_accessor :dbh
+
+  def setup
+    @dbh = mock_connect
+  end
+
+  def teardown
+    @dbh.disconnect
+  end
+
+  def test_01_allocation
+    sth = dbh.new_statement("some query")
+    assert(sth)
+    assert_kind_of(RDBI::Statement, sth)
+    assert_kind_of(RDBI::Database, sth.dbh)
+    assert_equal(sth.query, "some query")
+    sth.finish
+  end
+
+  def test_02_accessors
+    sth = dbh.new_statement("some query")
+    assert(sth)
+    assert_kind_of(Mutex, sth.mutex)
+    assert(!sth.finished?)
+    assert(!sth.finished)
+    sth.finish
+  end
+
+  def test_03_execute
+    res = dbh.execute("select * from foo where bar=?", 1)
+    assert_kind_of(RDBI::Result, res)
+    assert_equal(%W[10 11 12 13 14].collect { |x| [x] }, res.fetch(:all))
+
+    sth = dbh.prepare("select * from foo where bar=?")
+    assert_kind_of(RDBI::Statement, sth)
+    assert_kind_of(RDBI::Driver::Mock::STH, sth)
+    assert_respond_to(sth, :new_execution)
+    assert_respond_to(sth, :execute)
+    res = sth.execute(1)
+    assert_equal(%W[10 11 12 13 14].map { |x| [x] }, res.fetch(:all))
+    sth.finish
+    
+    sth = dbh.prepare("select * from foo where bar=?bar and foo=?")
+    res = sth.execute({:bar => "10"}, "1")
+    assert_equal(
+      [ { :bar => "10" }, "1" ],
+      res.binds
+    )
+    sth.finish
+  end
+
+  def test_04_finish
+    sth = dbh.prepare("select * from foo where bar=?")
+
+    assert(!sth.finished?)
+    sth.finish
+    assert(sth.finished?)
+    assert_raises(StandardError.new("you may not execute a finished handle")) { sth.execute }
+    sth.finish
+  end
+end
+
+# vim: syntax=ruby ts=2 et sw=2 sts=2

data/test/test_types.rb

@@ -0,0 +1,114 @@
+require 'helper'
+
+class TestTypes < Test::Unit::TestCase
+  def setup
+    @out_types = RDBI::Type.create_type_hash(RDBI::Type::Out)
+    @in_types  = RDBI::Type.create_type_hash(RDBI::Type::In)
+  end
+
+  def test_01_basic
+    assert_respond_to(RDBI::Type, :create_type_hash)
+
+    assert(@out_types)
+    assert_kind_of(Hash, @out_types)
+    assert(@out_types.keys.include?(:integer))
+    assert(@out_types.keys.include?(:decimal))
+    assert(@out_types.keys.include?(:datetime))
+    assert(@out_types.keys.include?(:boolean))
+    assert(@out_types.keys.include?(:default))
+    assert_respond_to(RDBI::Type::Out, :convert)
+
+    assert(@in_types)
+    assert_kind_of(Hash, @in_types)
+    assert(@in_types.keys.include?(Integer))
+    assert(@in_types.keys.include?(BigDecimal))
+    assert(@in_types.keys.include?(DateTime))
+    assert(@in_types.keys.include?(TrueClass))
+    assert(@in_types.keys.include?(FalseClass))
+    assert(@in_types.keys.include?(:default))
+    assert_respond_to(RDBI::Type::In, :convert)
+  end
+
+  def test_02_out_basic_convert
+    assert_equal(1,   out_convert("1", tcc(:integer), @out_types))
+    assert_equal(nil, out_convert(nil, tcc(:integer), @out_types))
+
+    assert_equal(BigDecimal("1.0"), out_convert("1.0", tcc(:decimal), @out_types))
+    assert_equal(nil,               out_convert(nil, tcc(:decimal), @out_types))
+
+    assert_kind_of(DateTime, out_convert(DateTime.now, tcc(:default), @out_types))
+    assert_kind_of(Float,    out_convert(1.0, tcc(:default), @out_types))
+
+    assert_equal(true,  out_convert("TRUE", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("true", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("TRue", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("trUE", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("T", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("t", tcc(:boolean), @out_types))
+    assert_equal(true,  out_convert("1", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("FALSE", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("false", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("FAlse", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("faLSE", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("F", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("f", tcc(:boolean), @out_types))
+    assert_equal(false, out_convert("0", tcc(:boolean), @out_types))
+    assert_equal(nil,   out_convert(nil, tcc(:boolean), @out_types))
+  end
+
+  def test_03_out_datetime_convert
+    format = RDBI::Type::DEFAULT_STRFTIME_FILTER
+    dt = DateTime.now
+
+    conv      = out_convert(dt, tcc(:datetime), @out_types).strftime(format)
+    formatted = dt.strftime(format)
+
+    assert_equal(formatted, conv)
+  end
+
+  def test_04_in_basic_convert
+    assert_equal("1", in_convert(1, @in_types))
+    assert_equal("1", in_convert(Integer("1"), @in_types))
+    assert_equal("1.0", in_convert(1.0, @in_types))
+    assert_equal("1.0", in_convert(BigDecimal("1.0"), @in_types))
+    assert_equal("artsy", in_convert("artsy", @in_types))
+    assert_equal("TRUE", in_convert(true, @in_types))
+    assert_equal("FALSE", in_convert(false, @in_types))
+    assert_equal(nil, in_convert(nil, @in_types))
+  end
+
+  def test_05_in_datetime_convert
+    format = RDBI::Type::DEFAULT_STRFTIME_FILTER
+    dt = DateTime.now
+
+    conv      = in_convert(dt, @in_types)
+    formatted = dt.strftime(format)
+
+    assert_equal(formatted, conv)
+  end
+
+  def test_06_out_integration
+    dt = DateTime.now
+    dbh = mock_connect
+    sth = dbh.prepare("select * from foo")
+    sth.set_schema = RDBI::Schema.new(
+      [
+        RDBI::Column.new(:foo, :integer, :integer),
+        RDBI::Column.new(:bar, :datetime, :datetime),
+        RDBI::Column.new(:quux, :varchar, :string),
+        RDBI::Column.new(:baz, :decimal, :decimal)
+      ]
+    )
+
+    sth.result = [["1", dt.strftime(RDBI::Type::DEFAULT_STRFTIME_FILTER), "hello, world!", "1.0"]]
+
+    res = sth.execute().fetch[0]
+
+    # we can't test datetime objects directly... because datetime is fail.
+    # so, what we do here is compare each result individually.
+    assert_equal(1, res[0])
+    assert_equal(dt.strftime(RDBI::Type::DEFAULT_STRFTIME_FILTER), res[1].strftime(RDBI::Type::DEFAULT_STRFTIME_FILTER))
+    assert_equal("hello, world!", res[2])
+    assert_equal(BigDecimal("1.0"), res[3])
+  end
+end