working with the connection more

.gitignore

@@ -2,3 +2,4 @@ ext/Makefile
 ext/mkmf.log
 ext/*.o
 ext/*.so
+spec/configuration.rb

Rakefile

@@ -1,12 +1,12 @@
 require 'rubygems'
 require 'rake/gempackagetask'
 require 'rake/rdoctask'
-require 'rake/testtask'
+require 'spec/rake/spectask'
 
 load 'vertica.gemspec'
 
 Rake::GemPackageTask.new(VERTICA_SPEC) do |pkg|
-    pkg.need_tar = true
+  pkg.need_tar = true
 end
 
 task :default => "test"
@@ -16,15 +16,12 @@ task :clean do
   include FileUtils
   rm_rf 'pkg'
 end
-
-desc "Run tests"
-Rake::TestTask.new("test") do |t|
-  t.libs << ["test", "ext"]
-  t.pattern = 'test/*_test.rb'
-  t.verbose = true
+
+Spec::Rake::SpecTask.new do |t|
   t.warning = true
+  t.rcov = false
 end
- 
+
 task :doc => [:rdoc]
 namespace :doc do
   Rake::RDocTask.new do |rdoc|

lib/vertica/column.rb

@@ -23,7 +23,7 @@ class Column
       :timestamp_tz,
       :interval,
       :time_tz,
-      :numberic,
+      :numeric,
       :bytea,
       :rle_tuple
     ]

lib/vertica/connection.rb

@@ -6,6 +6,7 @@
 
 module Vertica
   class Connection
+    attr_accessor :conn
     attr_accessor :host
     attr_accessor :port
     attr_accessor :database
@@ -247,10 +248,5 @@ def convert_transaction_status_to_sym(status)
         nil
       end
     end
-
-    def conn
-      @conn
-    end
-
   end
 end

lib/vertica/result.rb

@@ -8,7 +8,10 @@ def initialize(field_descriptions, field_values)
     end
 
     def columns
-      @columns ||= @field_descriptions.map { |fd| Column.new(fd[:type_modifier], fd[:format_code], fd[:table_oid], fd[:name], fd[:attribute_number], fd[:data_type_oid], fd[:data_type_size]) }
+      @columns ||= @field_descriptions.map do |fd| 
+        Column.new(fd[:type_modifier], fd[:format_code], fd[:table_oid],
+          fd[:name], fd[:attribute_number], fd[:data_type_oid], fd[:data_type_size])
+      end
     end
 
     def rows
@@ -42,5 +45,13 @@ def each
         yield row if block_given?
       end
     end
+
+    # Some compatibility with the pg gem
+
+    def getvalue(row, column)
+      rows[row][column]
+    end
+
+
   end
 end

spec/configuration.template.rb

@@ -0,0 +1,14 @@
+class Configuration
+
+  class << self
+    def host; ''; end
+    def port; 5433; end
+    def database; ''; end
+    def username; 'dbadmin'; end
+    def password; ''; end
+
+    def [](key)
+      send(key)
+    end
+  end
+end

spec/pg_compatibility_spec.rb

@@ -0,0 +1,24 @@
+require File.join(File.dirname(__FILE__), "spec_helper")
+
+describe Vertica::Connection do
+  before(:each) do
+    @connection = Vertica::Connection.new(
+      Configuration[:host], Configuration[:port],
+      Configuration[:database], Configuration[:username],
+      Configuration[:password], false, false
+    )
+  end
+
+  it "should" do
+    r = c.query("SELECT * FROM test_table")
+    assert_equal 1, r.row_count
+    assert_equal 2, r.columns.length
+    assert_equal :in, r.columns[0].data_type
+    assert_equal 'id', r.columns[0].name
+    assert_equal :varchar, r.columns[1].data_type
+    assert_equal 'name', r.columns[1].name
+    assert_equal [[1, 'matt']], r.rows
+    c.close
+
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1,2 @@
+require File.join(File.dirname(__FILE__), 'configuration')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'vertica')

test/test_helper.rb

@@ -5,10 +5,10 @@
 class Test::Unit::TestCase
 
   TEST_CONNECTION_USER     = 'dbadmin'
-  TEST_CONNECTION_PASSWORD = 'cl3v3rcstore'
-  TEST_CONNECTION_HOST     = 'ec2-174-129-157-242.compute-1.amazonaws.com'
+  TEST_CONNECTION_PASSWORD = 'yamittome'
+  TEST_CONNECTION_HOST     = 'edb-001'
   TEST_CONNECTION_PORT     = 5433
-  TEST_CONNECTION_DATABASE = 'db'
+  TEST_CONNECTION_DATABASE = 'yamhouse'
 
   TEST_CONNECTION_HASH = {
     :user     => TEST_CONNECTION_USER,