test_adapter.rb in swift-0.7.0

- old
+ new
@@ -1,124 +1,121 @@
 require_relative 'helper'
 require 'stringio'
 
 describe 'Adapter' do
-  supported_by Swift::DB::Postgres, Swift::DB::Mysql do
+  supported_by Swift::DB::Postgres, Swift::DB::Mysql, Swift::DB::DB2 do
     describe 'db' do
+      before do
+        @db = Swift.db
+        @db.execute('drop table users') rescue nil
+        type = case @db
+          when Swift::DB::DB2 then 'integer not null generated by default as identity'
+          else 'serial'
+        end
+        @db.execute %Q{create table users(id #{type}, name varchar(512), email varchar(512), created_at timestamp)}
+      end
+
       it 'yields db to block' do
         Swift.db do |db|
           assert_kind_of Swift::Adapter, db
         end
       end
-    end
 
-    describe 'execute' do
-      it 'executes without bind values' do
-        assert Swift.db.execute %q{drop table if exists users}
-      end
+      describe 'execute' do
+        it 'executes without bind values' do
+          assert @db.execute %q{select count(*) from users}
+        end
 
-      it 'executes with bind values' do
-        Swift.db.execute %q{drop table if exists users}
-        Swift.db.execute %q{create table users(id serial, name text, created_at timestamp)}
-        assert Swift.db.execute 'insert into users (name, created_at) values (?, now())', 'Benny Arthurton'
+        it 'executes with bind values' do
+          assert @db.execute 'insert into users (name, created_at) values (?, current_timestamp)', 'Benny Arthurton'
+        end
       end
-    end
 
-    describe 'prepared statements' do
-      before do
-        @db = Swift.db do |db|
-          db.execute %q{drop table if exists users}
-          db.execute %q{create table users(id serial, name text, created_at timestamp)}
+      describe 'prepared statements' do
+        it 'executes via Statement#new' do
+          result = []
+          Swift::Statement.new(@db, 'select count(*) as n from users').execute {|r| result << r[:n] }
+          assert_kind_of Integer, result[0]
         end
-      end
 
-      it 'executes without bind values' do
-        assert @db.prepare(%q{insert into users (name, created_at) values ('Apple Arthurton', now())}).execute
-      end
+        it 'executes without bind values' do
+          assert @db.prepare(%q{insert into users (name) values ('Apple Arthurton')}).execute
+        end
 
-      it 'executes with bind values' do
-        assert @db.prepare(%q{insert into users (name, created_at) values (?, now())}).execute('Apple Arthurton')
-      end
+        it 'executes with bind values' do
+          assert @db.prepare(%q{insert into users (name) values (?)}).execute('Apple Arthurton')
+        end
 
-      it 'executes multiple times' do
-        sth = @db.prepare(%q{insert into users (name, created_at) values (?, now())})
-        assert sth.execute('Apple Arthurton')
-        assert sth.execute('Benny Arthurton')
-      end
+        it 'executes multiple times' do
+          sth = @db.prepare(%q{insert into users (name, created_at) values (?, current_timestamp)})
+          assert sth.execute('Apple Arthurton')
+          assert sth.execute('Benny Arthurton')
+        end
 
-      it 'has insert_id' do
-        sql = case @db
-          when Swift::DB::Postgres then %q{insert into users (name, created_at) values (?, now()) returning id}
-          when Swift::DB::Mysql    then %q{insert into users (name, created_at) values (?, now())}
+        it 'has insert_id' do
+          sql = case @db
+            when Swift::DB::Postgres then %q{insert into users (name) values (?) returning id}
+            when Swift::DB::Mysql    then %q{insert into users (name) values (?)}
+            when Swift::DB::DB2      then %q{select id from final table(insert into users(name) values(?))}
+          end
+          assert_kind_of Numeric, @db.prepare(sql).execute('Connie Arthurton').insert_id
         end
-        assert_kind_of Numeric, @db.prepare(sql).execute('Connie Arthurton').insert_id
       end
-    end
 
-    describe 'executed prepared statements' do
-      before do
-        @db = Swift.db do |db|
-          db.execute %q{drop table if exists users}
-          db.execute %q{create table users(id serial, name text, created_at timestamp)}
-          sth = db.prepare(%q{insert into users (name, created_at) values (?, now())})
-          sth.execute('Apple Arthurton')
-          sth.execute('Benny Arthurton')
+      describe 'executed prepared statements' do
+        before do
+          insert = @db.prepare(%q{insert into users (name, created_at) values (?, current_timestamp)})
+          insert.execute('Apple Arthurton')
+          insert.execute('Benny Arthurton')
+          @sth = @db.prepare('select * from users').execute
         end
-        @sth = @db.prepare('select * from users').execute
-      end
 
-      it 'enumerates' do
-        assert_kind_of Enumerable, @sth
-      end
+        it 'enumerates' do
+          assert_kind_of Enumerable, @sth
+        end
 
-      it 'enumerates block' do
-        begin
-          @sth.execute{|row| row}
-        rescue => error
-          flunk error.message
-        else
-          pass
+        it 'enumerates block' do
+          begin
+            @sth.execute{|row| row}
+          rescue => error
+            flunk error.message
+          else
+            pass
+          end
         end
-      end
 
-      it 'returns hash tuples for enumerable methods' do
-        assert_kind_of Hash, @sth.first
-      end
+        it 'returns hash tuples for enumerable methods' do
+          assert_kind_of Hash, @sth.first
+        end
 
-      it 'returns a result set on Adapter#execute{}' do
-        @db.execute('select * from users') {|r| assert_kind_of Hash, r }
-      end
+        it 'returns a result set on Adapter#execute{}' do
+          @db.execute('select * from users') {|r| assert_kind_of Hash, r }
+        end
 
-      it 'returns a result set on Adapter#results' do
-        @db.execute('select * from users')
-        assert_kind_of Swift::Result, @db.results
-      end
+        it 'returns a result set on Adapter#results' do
+          @db.execute('select * from users')
+          assert_kind_of Swift::Result, @db.results
+        end
 
-      it 'returns fields' do
-        assert_equal [ :id, :name, :created_at ], @sth.fields
+        it 'returns fields' do
+          assert_equal [ :id, :name, :email, :created_at ], @sth.fields
+        end
       end
-    end
 
 
-    #--
-    # TODO: Not sure how I feel about the block in write; feels like it's just there to get around the fields in the
-    # argument list. How about write('users', %w{name, email, balance}, data)?
-    describe 'bulk writes!' do
-      before do
-        @db = Swift.db do |db|
-          db.execute %q{drop table if exists users}
-          db.execute %q{create table users(id serial, name text, email text)}
+      #--
+      # TODO: Not sure how I feel about the block in write; feels like it's just there to get around the fields in the
+      # argument list. How about write('users', %w{name, email, balance}, data)?
+      describe 'bulk writes!' do
+        it 'writes from an IO object' do
+          data = StringIO.new "Sally Arthurton\tsally@local\nJonas Arthurton\tjonas@local\n"
+          assert_equal 2, @db.write('users', %w{name email}, data)
         end
-      end
 
-      it 'writes from an IO object' do
-        data = StringIO.new "Sally Arthurton\tsally@local\nJonas Arthurton\tjonas@local\n"
-        assert_equal 2, Swift.db.write('users', %w{name email}, data)
-      end
-
-      it 'writes from a string' do
-        data = "Sally Arthurton\tsally@local\nJonas Arthurton\tjonas@local\n"
-        assert_equal 2, Swift.db.write('users', %w{name email}, data)
+        it 'writes from a string' do
+          data = "Sally Arthurton\tsally@local\nJonas Arthurton\tjonas@local\n"
+          assert_equal 2, @db.write('users', %w{name email}, data)
+        end
       end
     end
   end
 end