ruby-informix 0.6.2 → 0.7.0

data/lib/informix/seqcursor.rb

@@ -0,0 +1,169 @@
+# $Id: seqcursor.rb,v 1.3 2008/03/29 07:35:03 santana Exp $
+#
+# Copyright (c) 2008, Gerardo Santana Gomez Garrido <gerardo.santana@gmail.com>
+# All rights reserved.
+# 
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 
+# 1. Redistributions of source code must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+#    notice, this list of conditions and the following disclaimer in the
+#    documentation and/or other materials provided with the distribution.
+# 3. The name of the author may not be used to endorse or promote products
+#    derived from this software without specific prior written permission.
+# 
+# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
+# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+
+require 'informixc'
+
+module Informix
+  class SequentialCursor < CursorBase
+    include Enumerable
+
+    # Fetches the next record.
+    #
+    # Returns the record fetched as an array, or nil if there are no
+    # records left.
+    #
+    #   cursor.fetch  => array or nil
+    def fetch
+      fetch0(Array, false)
+    end
+
+    # Fetches the next record, storing it in the same Array object every time
+    # it is called.
+    #
+    # Returns the record fetched as an array, or nil if there are no
+    # records left.
+    #
+    #   cursor.fetch!  => array or nil
+    def fetch!
+      fetch0(Array, true)
+    end
+
+    # Fetches the next record.
+    #
+    # Returns the record fetched as a hash, or nil if there are no
+    # records left.
+    #
+    #   cursor.fetch_hash  => hash or nil
+    def fetch_hash
+      fetch0(Hash, false)
+    end
+
+    # Fetches the next record, storing it in the same Hash object every time
+    # it is called.
+    #
+    # Returns the record fetched as a hash, or nil if there are no
+    # records left.
+    #
+    #   cursor.fetch_hash!  => hash or nil
+    def fetch_hash!
+      fetch0(Hash, true)
+    end
+
+    # Reads at most <i>n</i> records.
+    #
+    # Returns the records read as an array of arrays
+    #
+    #   cursor.fetch_many(n)  => array
+    def fetch_many(n)
+      fetch_many0(n, Array)
+    end
+
+    # Reads at most <i>n</i> records.
+    # Returns the records read as an array of hashes.
+    #
+    #   cursor.fetch_hash_many(n)  => array
+    def fetch_hash_many(n)
+      fetch_many0(n, Hash)
+    end
+
+    # Returns all the records left as an array of arrays
+    #
+    #   cursor.fetch_all  => array
+    def fetch_all
+      fetch_many0(nil, Array)
+    end
+
+    # Returns all the records left as an array of hashes
+    #
+    #   cursor.fetch_hash_all  => array
+    def fetch_hash_all
+      fetch_many0(nil, Hash)
+    end
+
+    # Iterates over the remaining records, passing each <i>record</i> to the
+    # <i>block</i> as an array.
+    #
+    # Returns __self__.
+    #
+    #   cursor.each {|record| block } => cursor
+    def each(&block)
+      each0(Array, false, &block)
+    end
+
+    # Iterates over the remaining records, passing each <i>record</i> to the
+    # <i>block</i> as an array. No new Array objects are created for each
+    # record. The same Array object is reused in each call.
+    #
+    # Returns __self__.
+    #
+    #   cursor.each! {|record| block } => cursor
+    def each!(&block)
+      each0(Array, true, &block)
+    end
+
+    # Iterates over the remaining records, passing each <i>record</i> to the
+    # <i>block</i> as a hash.
+    #
+    #   cursor.each_hash {|record| block } => cursor
+    def each_hash(&block)
+      each0(Hash, false, &block)
+    end
+
+    # Iterates over the remaining records, passing each <i>record</i> to the
+    # <i>block</i> as a hash. No new Hash objects are created for each record.
+    # The same Hash object is reused in each call.
+    #
+    # Returns __self__.
+    #
+    #   cursor.each_hash! {|record| block } => cursor
+    def each_hash!(&block)
+      each0(Hash, true, &block)
+    end
+
+    # Iterates over the remaining records, passing at most <i>n</i>
+    # <i>records</i> to the <i>block</i> as arrays.
+    #
+    # Returns __self__.
+    #
+    #   cursor.each_by(n) {|records| block } => cursor
+    def each_by(n, &block)
+      each_by0(n, Array, &block)
+    end
+
+    # Iterates over the remaining records, passing at most <i>n</i>
+    # <i>records</i> to the <i>block</i> as hashes.
+    #
+    # Returns __self__.
+    #
+    #   cursor.each_hash_by(n) {|records| block } => cursor
+    def each_hash_by(n, &block)
+      each_by0(n, Hash, &block)
+    end
+  end # class SequentialCursor
+end # module Informix

data/test/ifx_all.rb

@@ -0,0 +1,31 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+
+class IfxAll
+  def IfxAll.suite
+    suite = Test::Unit::TestSuite.new "Ruby Informix Test Suite"
+    Object.constants.grep(/^IfxTest/).sort.each do |const_name|
+      if (c = Object.const_get(const_name)).kind_of?(Class) && c.respond_to?(:suite)
+        puts "Adding #{const_name}"
+        suite << c.suite
+      end
+    end
+    suite
+  end
+end
+
+if __FILE__ == $0
+  if ARGV.size == 0
+    STDERR.puts "Usage:
+ruby #{$0} database [username password]"
+    exit 1
+  end
+  require 'test/unit/ui/console/testrunner'
+  Dir.glob(File.join(testdir, "ifx_test*.rb")).each do |testcase|
+    require "#{testcase}"
+  end
+  Test::Unit::UI::Console::TestRunner.run(IfxAll)
+end

data/test/ifx_test_create_table.rb

@@ -0,0 +1,26 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+
+class IfxTestCreate < Informix::TestCase
+  def setup
+    connect_string, username, password = ARGV[0..2]
+    super connect_string, username, password
+  end
+  
+  def test_create_table
+    drop_test_table
+
+    assert_nothing_raised(Informix::Error, "Creating table #{TEST_TABLE_NAME}") do
+      begin
+        create_test_table
+      ensure
+        drop_test_table
+      end
+    end
+  end
+end

data/test/ifx_test_errinfo.rb

@@ -0,0 +1,37 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+
+# Informix::ExcInfo is just a struct, not much testing needed
+class IfxTestExcInfo < Informix::TestCase
+  # Don't need an Informix database for these tests
+  def setup; end
+  
+  def test_initialize
+    sql_code, sql_state, class_origin, subclass_origin, message, server_name, connection_name = 
+      -987, "IX111", "IX", "000", "Some message", "server", "conn"
+
+    info = Informix::ExcInfo.new(sql_code,
+                                 sql_state,
+                                 class_origin,
+                                 subclass_origin,
+                                 message,
+                                 server_name,
+                                 connection_name)
+
+    assert_equal sql_code, info.sql_code
+    assert_equal sql_state, info.sql_state
+    assert_equal class_origin, info.class_origin
+    assert_equal subclass_origin, info.subclass_origin
+    assert_equal message, info.message
+    assert_equal server_name, info.server_name
+    assert_equal connection_name, info.connection_name
+  end
+
+end
+
+

data/test/ifx_test_exceptions.rb

@@ -0,0 +1,71 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+
+class IfxTestExceptions < Informix::TestCase
+  def info_params
+    [ 
+      [ -100, "IX111", "IX", "000", "Some message", "server", "conn" ],
+      [ -200, "IX111", "IX", "000", "Some message", "server", "conn" ],
+      [ -300, "IX111", "IX", "000", "Some message", "server", "conn" ]
+    ]
+  end
+
+  # Don't need an Informix database for these tests
+  def setup
+    @info_arr = info_params.map { |arr| Informix::ExcInfo.new(*arr) }
+    @test_exc = Informix::Error.new @info_arr
+  end
+
+  # Also tests Informix::Error#each via each_with_index
+  def test_add_info
+    params = self.info_params
+    exc = Informix::Error.new
+    params.each { |arr| exc.add_info(*arr) }
+    exc.each_with_index { |info, i| assert_equal params[i], info.to_a }
+  end
+
+  def test_each_for_empty_exception
+    exc = Informix::Error.new
+    assert_nothing_raised { exc.each { |x| } }
+  end
+  
+  def test_size
+    assert_equal info_params.size, @test_exc.size
+  end
+
+  def test_length
+    assert_equal info_params.length, @test_exc.length
+  end
+
+  def test_initialize_type_error
+    assert_raise(TypeError) { Informix::Error.new [1, 2, 3] }
+    assert_raise(TypeError) { Informix::Error.new 0 }
+  end
+
+  def test_initialize_with_string
+    err = nil
+    expected_msg = "This is a message"
+    assert_nothing_raised { err = Informix::Error.new expected_msg }
+    assert_equal expected_msg, err.message
+  end
+
+  def test_to_s
+    assert_nothing_raised { @test_exc.to_s }
+  end
+
+  def test_at
+    @test_exc.size.times { |i| assert_equal @info_arr[i], @test_exc[i] }
+  end
+  
+  def test_sqlcode
+    assert_equal -100, @test_exc.sql_code
+    assert_equal 0, Informix::Error.new.sql_code
+  end
+  
+end
+

data/test/ifx_test_fetch_n_each.rb

@@ -0,0 +1,64 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+require 'stringio'
+require 'date'
+
+class IfxTestXFetchNEach < Informix::TestCase
+  def setup
+    connect_string, username, password = ARGV[0..2]
+    super connect_string, username, password
+    drop_test_table
+    create_test_table
+    populate_test_table
+
+    assert_nothing_raised(Informix::Error, "Selecting records") do
+      @c = @db.cursor("select * from #{TEST_TABLE_NAME}").open
+    end
+  end
+
+  def teardown
+    @c.drop
+  end
+
+  def test_fetch_each_each_by
+    fetch = []
+    while r = @c.fetch; fetch << r; end
+    @c.close
+
+    each = []
+    @c.open.each {|r| each << r }.close
+
+    each_by = []
+    @c.open.each_by(2) {|a, b|
+      each_by << a if a
+      each_by << b if b
+    }
+    assert_equal(fetch.flatten, each.flatten, "fetch & each")
+    assert_equal(fetch.flatten, each_by.flatten, "fetch & each_by")
+  end
+
+  def test_fetch_each_each_by_hash
+    fetch_hash = []
+    @c.open
+    while r = @c.fetch_hash; fetch_hash << r; end
+    @c.close
+
+    each_hash = []
+    @c.open.each_hash {|r| each_hash << r }.close
+
+    each_hash_by = []
+    @c.open.each_hash_by(2) {|a, b|
+      each_hash_by << a if a
+      each_hash_by << b if b
+    }
+    assert_equal(fetch_hash.flatten, each_hash.flatten,
+      "fetch_hash & each_hash")
+    assert_equal(fetch_hash.flatten, each_hash_by.flatten,
+      "fetch_hash & each_hash_by")
+  end
+end

data/test/ifx_test_insert.rb

@@ -0,0 +1,27 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+require 'stringio'
+require 'date'
+
+class IfxTestInsert < Informix::TestCase
+  def setup
+    connect_string, username, password = ARGV[0..2]
+    super connect_string, username, password
+    drop_test_table
+    create_test_table
+  end
+
+  def test_insert
+    begin
+      populate_test_table
+    ensure
+      drop_test_table
+    end
+  end
+end
+

data/test/ifx_test_select.rb

@@ -0,0 +1,87 @@
+testdir = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.unshift testdir
+$LOAD_PATH.unshift File.join(testdir, "..")
+
+require 'informix'
+require 'test/unit'
+require 'testcase'
+require 'stringio'
+require 'date'
+
+class IfxTestSelect < Informix::TestCase
+  def setup
+    connect_string, username, password = ARGV[0..2]
+    super connect_string, username, password
+    drop_test_table
+    create_test_table
+    populate_test_table
+  end
+
+  def test_select
+    rows = nil
+
+    assert_nothing_raised(Informix::Error, "Selecting records") do
+      rows = db.cursor('select * from test').open.fetch_all
+    end
+    
+    assert_equal(2, rows.size, "# of records retrieved")
+    
+    exp_rows = @rows
+
+    2.times do |i|
+      assert_equal(exp_rows[i].shift, rows[i].shift, "serial")
+      assert_equal(exp_rows[i].shift, rows[i].shift, "char")
+      assert_equal(exp_rows[i].shift, rows[i].shift, "varchar")
+      assert_equal(exp_rows[i].shift, rows[i].shift, "smallint")
+      assert_equal(exp_rows[i].shift, rows[i].shift, "integer")
+      
+      assert_in_delta(exp_rows[i].shift, rows[i].shift, 1e-6, "smallfloat")
+      assert_in_delta(exp_rows[i].shift, rows[i].shift, 1e-14, "float")
+
+      obj = rows[i].shift
+      obj = obj.strftime("%m/%d/%Y") if obj.respond_to? :strftime
+      exp = exp_rows[i].shift
+      exp = exp.strftime("%m/%d/%Y") if exp.respond_to? :strftime
+
+      assert_equal(exp, obj, "date")
+      
+      obj = rows[i].shift
+      obj = obj.strftime("%Y-%m-%d %H:%M:%S.000") if obj.respond_to? :strftime
+      exp = exp_rows[i].shift
+      exp = exp.strftime("%Y-%m-%d %H:%M:%S.000") if exp.respond_to? :strftime
+
+      assert_equal(exp, obj, "datetime")
+      
+      assert_equal(exp_rows[i].shift, rows[i].shift, "interval year to month")
+      assert_equal(exp_rows[i].shift, rows[i].shift, "interval day to second")
+
+      assert_equal(exp_rows[i].shift, rows[i].shift, "decimal")
+
+      obj = rows[i].shift
+      exp = exp_rows[i].shift
+
+      if obj.class == StringIO
+        obj.rewind 
+        obj = obj.read # Convert to string
+      end
+
+      if exp.class == StringIO
+        exp.rewind 
+        exp = exp.read # Convert to string
+      end
+
+     assert_equal(exp, obj, "text")
+
+      if @supported["boolean"]
+        assert_equal(exp_rows[i].shift, rows[i].shift ? 't': 'f', "boolean")
+      end
+      
+      if @supported["int8"]
+        assert_equal(exp_rows[i].shift, rows[i].shift, "int8")
+      end
+
+    end
+  end
+end
+
+