Rubernate 0.1.0

data/tests/rubernate/impl/dbigeneric_stub.rb

@@ -0,0 +1,635 @@
+$:.unshift(File.expand_path('../../..', __FILE__)) unless 
+  $:.include?(File.expand_path('../../..', __FILE__))
+
+
+require 'config'
+require 'rubernate/fixtures' 
+require 'set'
+
+module Rubernate
+module DBI::GenericTests
+  include Rubernate
+  include DBI
+  include FixtureClasses
+  
+  def init rt_class, db_url, db_user, db_password
+    @factory = RuntimeFactory.new rt_class, db_url, db_user, db_password
+    @factory.connect {|dbh| 
+      dbh.do 'delete from r_params'
+      dbh.do 'delete from r_objects'
+      dbh.commit
+    }
+    @runtime = @factory.create
+    Thread.current[:on_save] = Thread.current[:on_load] = nil
+  end
+  
+  def teardown
+    @runtime.dbh.disconnect if @runtime and @runtime.dbh
+  end  
+      
+  def test_connect
+    assert @runtime
+    assert @runtime.dbh
+    assert_nothing_raised { 
+      assert_equal 0, @runtime.dbh.select_all('select * from r_objects').size
+    }
+    assert_nothing_raised { 
+      assert_equal 0, @runtime.dbh.select_all('select * from r_params').size
+    }
+  end
+  
+  def test_create
+    o = C1.new
+    o.peer = Peer.new
+    @runtime.create o
+    rows = @runtime.dbh.select_all 'select * from r_objects'
+    assert_equal 1, rows.size
+    assert_equal C1.name, rows[0][1]
+    assert o.peer.dirty?
+  end
+  
+  def test_save_objects
+    objects = [C1.new, C2.new, C3.new]        
+    objects.each {|object| @runtime.create object}
+    
+    @runtime.save objects
+    
+    for object in objects
+      assert !object.peer.dirty?
+    end        
+    
+    rows = @runtime.dbh.select_all 'select * from r_objects order by object_pk'
+    assert_equal 3, rows.size
+    
+    for object in objects
+      row = rows.shift
+      assert_equal object.primary_key, row[0].to_i
+      assert_equal object.class.name, row[1]
+    end        
+  end
+  
+  def test_save_param_string    
+    o, o.p1 = C1.new, 'test'
+    @runtime.create o
+    @runtime.save o
+    rows = @runtime.dbh.select_all 'select * from r_params' 
+    assert_equal 1, rows.size
+    assert_equal o.primary_key, rows[0]['OBJECT_PK'].to_i
+    assert_equal PARAM_FLAG_STRING, rows[0]['FLAGS'].to_i
+    assert_nil rows[0]['INT_VALUE']
+    assert_equal 'test', rows[0]['STR_VALUE']
+    assert_nil rows[0]['DAT_VALUE']
+    assert_nil rows[0]['REF_VALUE']
+  end
+  
+  def test_save_param_int
+    o, o.p1 = C1.new, 777
+    @runtime.create o
+    @runtime.save o
+    rows = @runtime.dbh.select_all 'select * from r_params'
+    assert_equal  1, rows.size
+    assert_equal  o.primary_key, rows[0]['OBJECT_PK'].to_i
+    assert_equal PARAM_FLAG_INT, rows[0]['FLAGS'].to_i
+    assert_equal 777, rows[0]['INT_VALUE'].to_i
+    assert_nil rows[0]['STR_VALUE']
+    assert_nil rows[0]['DAT_VALUE']
+    assert_nil rows[0]['REF_VALUE']
+  end
+  
+  def test_save_param_ref
+    o0, o1 = C2.new, C1.new
+    o1.p1 = o0
+    @runtime.create o0 
+    @runtime.create o1
+    @runtime.save [o0, o1]
+    
+    rows = @runtime.dbh.select_all 'select * from r_params' 
+    assert_equal 1, rows.size
+    assert_equal o1.primary_key, rows[0]['OBJECT_PK'].to_i
+    assert_equal PARAM_FLAG_REF, rows[0]['FLAGS'].to_i
+    assert_nil rows[0]['INT_VALUE']
+    assert_nil rows[0]['STR_VALUE']
+    assert_nil rows[0]['DAT_VALUE']
+    assert_equal o0.primary_key, rows[0]['REF_VALUE'].to_i
+  end
+      
+  def test_save_param_int_hash
+    hash = {}
+    for i in 1..10
+      hash[i] = C2.new
+      @runtime.create hash[i]
+    end
+    o, o.p1 = C1.new, hash
+    
+    @runtime.create o
+    @runtime.save o
+    
+    rows = @runtime.dbh.select_all 'select * from r_params order by int_value' 
+    assert_equal 10, rows.size
+    for row in rows
+      assert_equal o.primary_key, row['OBJECT_PK'].to_i
+      assert_equal PARAM_FLAG_INT | PARAM_FLAG_HASH, row['FLAGS'].to_i
+      assert hash.has_key?(row['INT_VALUE'].to_i)
+      assert_nil row['STR_VALUE']
+      assert_nil row['DAT_VALUE']
+      assert_equal hash[row['INT_VALUE'].to_i].primary_key, row['REF_VALUE'].to_i
+    end
+  end
+  
+  def test_save_param_si_hash
+    hash = {777 => C2.new}
+
+    @runtime.create hash[777]
+    
+    hash['key'] = C2.new
+    @runtime.create hash['key']
+    
+    o, o.p1 = C1.new, hash
+    
+    @runtime.create o
+    @runtime.save [o]
+    
+    rows = @runtime.dbh.select_all 'select * from r_params order by int_value'
+    assert_equal 2, rows.size
+    
+    for row in rows
+      if row['FLAGS'].to_i == HASH_INT_REF
+        assert_equal o.primary_key, row['OBJECT_PK'].to_i
+        assert hash.has_key?(row['INT_VALUE'].to_i)
+        assert_nil row['STR_VALUE']
+        assert_nil row['DAT_VALUE']
+        assert_equal hash[row['INT_VALUE'].to_i].primary_key, row['REF_VALUE'].to_i
+      elsif row['FLAGS'].to_i == HASH_STRING_REF
+        assert_equal o.primary_key, row['OBJECT_PK'].to_i
+        assert_nil row['INT_VALUE']
+        assert hash.has_key?(row['STR_VALUE'])
+        assert_nil row['DAT_VALUE']
+        assert_equal hash[row['STR_VALUE']].primary_key, row['REF_VALUE'].to_i
+      else 
+        flunk 'Invalid flags ' + row['FLAGS'].to_i
+      end
+    end
+  end
+  
+  def test_save_param_array
+    array = []
+    for i in 0..9
+      o = C2.new
+      @runtime.create o
+      array[i] = o
+    end
+    o, o.p1 = C1.new, array
+    
+    @runtime.create o
+    @runtime.save o
+    
+    rows = @runtime.dbh.select_all 'select * from r_params order by int_value'
+    assert_equal 10, rows.size
+    for i in 0..9
+      row = rows[i]
+      assert_equal o.primary_key, row['OBJECT_PK'].to_i
+      assert_equal 'p1', row['NAME']
+      assert_equal ARRAY_INT_REF, row['FLAGS'].to_i
+      assert_equal i, row['INT_VALUE'].to_i
+      assert_nil row['STR_VALUE']
+      assert_nil row['DAT_VALUE']
+      assert_equal array[row['INT_VALUE'].to_i].primary_key, row['REF_VALUE'].to_i
+    end
+  end
+  
+  def test_delete        
+    o1, o2 = C1.new, C2.new
+    
+    @runtime.create o1
+    @runtime.create o2
+    
+    o1.p1 = o2
+    @runtime.save o1
+    @runtime.save o2
+    
+    assert 1, @runtime.dbh.select_all(
+      'select * from r_objects where object_pk = ?', o1.primary_key).size
+    assert 1, @runtime.dbh.select_all(
+      'select * from r_objects where object_pk = ?', o2.primary_key).size
+    assert 1, @runtime.dbh.select_all(
+      'select * from r_params where object_pk = ?', o1.primary_key).size     
+    @runtime.delete o1
+    
+    assert 0, @runtime.dbh.select_all(
+      'select * from r_objects where object_pk = ?', o1.primary_key).size
+    assert 1, @runtime.dbh.select_all(
+      'select * from r_objects where object_pk = ?', o2.primary_key).size
+    assert 0, @runtime.dbh.select_all(
+      'select * from r_params where object_pk = ?', o1.primary_key).size             
+  end
+  
+  def test_find_by_query_empty
+    peers = @runtime.find_by_query "select * from r_objects"
+    assert peers
+    assert_equal 0, peers.size
+  end
+  
+  def test_find_by_query_order_by_int
+    for i in 1..10
+      o = C1.new
+      @runtime.create o
+      o.p1 = i
+      @runtime.save [o]
+    end
+    objects = @runtime.find_by_query %{select  o.* from r_objects o, r_params p
+                           where o.object_pk = p.object_pk order by p.int_value}
+    assert_equal 10, objects.size
+    for i in 1..10
+      object = objects[i - 1]
+      assert_equal C1, object.class
+      assert_equal  i, object.p1
+      assert !object.peer.dirty?
+    end
+  end
+  
+  def test_load_by_pk
+    assert_nil @runtime.load_by_pk(0)
+    o0, o1 =  C2.new, C1.new
+    @runtime.create o0
+    @runtime.create o1
+    o0.p1 = 'o0.p1'
+    o0.p2 = o1
+    @runtime.save [o0, o1]
+    
+    o0 = @runtime.load_by_pk o0.primary_key
+    assert o0
+    assert_equal 2, o0.peer.size
+    assert_equal 'o0.p1', o0.p1
+    assert_equal o0.p2, o1
+  end  
+  
+  def test_param_type_time
+    pk, time = nil, Time.new
+    Rubernate.session {o = C1.new.attach;  o.p1 = time; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_equal time.to_a[0..5], o.p1.to_a[0..5] # Ignore time-zone
+    }
+  end
+  
+  def test_param_type_date
+    pk, date = nil, Date.today    
+    Rubernate.session {o = C1.new.attach;  o.p1 = date; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_equal date, o.p1
+    }
+  end
+  
+  def test_param_hash_time
+    pk, time = nil, Time.now
+    Rubernate.session {o = C1.new.attach; o.p1 = {time => o}; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_equal 1, o.p1.size
+      _time, _obj = o.p1.shift
+      assert_equal time.to_a[0..5], _time.to_a[0..5] # Ignore time-zone
+      assert_equal o, _obj
+    }
+  end
+
+  def test_param_hash_date
+    pk, date = nil, Date.today
+    Rubernate.session {o = C1.new.attach; o.p1 = {date => o}; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_equal 1, o.p1.size
+      _date, _obj = o.p1.shift
+      assert_equal date, _date 
+      assert_equal o, _obj
+    }
+  end  
+  
+  def test_param_hash_nil
+    pk = nil
+    Rubernate.session {o = C1.new.attach; o.p1 = {nil => o}; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_equal 1, o.p1.size
+      assert_equal o, o.p1[nil]
+    }
+  end
+  
+  def test_param_nil
+    pk = nil
+    Rubernate.session {o = C1.new.attach; o.p1 = 'not nil'; pk = o.primary_key}
+    Rubernate.session {
+      o = find_by_pk pk 
+      assert_equal 'not nil', o.p1
+      o.p1 = nil
+    }
+    Rubernate.session {
+      o = find_by_pk pk
+      assert_nil o.p1
+    }
+  end
+  
+  def test_empty_array
+    pk = nil
+    Rubernate.session {o = C1.new.attach; o.p1 = []; pk = o.primary_key}
+    Rubernate.session do
+      o = find_by_pk pk
+      assert_equal [], o.p1
+    end
+  end
+  
+  def test_empty_hash
+    pk = nil
+    Rubernate.session {o = C1.new.attach; o.p1 = {}; pk = o.primary_key}
+    Rubernate.session do
+      o = find_by_pk pk
+      assert_equal ({}), o.p1
+    end
+  end  
+end
+
+module GenericRuntimeTests
+  include Rubernate
+  include DBI
+  include Queries
+  include FixtureClasses
+  
+  def init rt_class, db_url, db_user, db_password
+    @factory = RuntimeFactory.new rt_class, db_url, db_user, db_password
+    @factory.connect {|dbh| 
+      dbh.do 'delete from r_params'
+      dbh.do 'delete from r_objects'
+      dbh.commit
+    }
+    Rubernate.runtime_factory = @factory
+    Thread.current[:on_save] = Thread.current[:on_load] = nil
+  end
+  
+  def test_session
+    Rubernate.session {|runtime|
+      assert runtime
+      assert runtime.dbh
+      assert_equal 0, runtime.dbh.select_all('select * from r_objects').size
+    }
+  end
+  
+  def test_session_commit
+    Rubernate.session{|runtime|
+      runtime.dbh.do "insert into r_objects values (3, 'C1')"
+    }
+    rows = @factory.connect.select_all('select * from r_objects')
+
+    assert_equal 1, rows.size
+    assert_equal 3, rows[0][0].to_i
+    assert_equal 'C1', rows[0][1].to_s
+  end
+  
+  def test_session_rollback        
+    assert_raise(RuntimeError) {
+      Rubernate.session{|runtime|
+        runtime.dbh.do "insert into r_objects values (4, 'C1')"
+        raise 'error'
+      }
+    }
+    assert_equal 0, @factory.connect.select_all('select * from r_objects').size
+  end
+  
+  def test_find_and_load_lazy
+    k1, k2 = nil
+    Rubernate.session{|runtime|
+      o1, o2 = C1.new.attach, C2.new.attach
+      o1.p1, o2.p1, o2.p2 = 'o1.p1', 'o2.p1', o1
+      k1, k2 = o1.primary_key, o2.primary_key
+    }
+    Rubernate.session{|runtime|
+      o2 = runtime.find_by_pk k2
+      assert 'o2.p1', o2.p1
+      assert 'o1.p1', o2.p2.p1 # o1 should be loaded lazy
+    }
+  end
+  
+  def test_find_flushed
+    k1, k2 = nil
+    Rubernate.session{|runtime|
+      o1, o2 = C1.new.attach, C2.new.attach            
+      o1.p1, o2.p1, o2.p2 = 'o1.p1', 'o2.p1', o1
+      result = runtime.find_by_query %{select o.*
+                  from r_objects o, r_params p
+                 where o.object_pk = p.object_pk and p.ref_value = ?}, o1
+      assert_equal 1, result.size
+      assert o2.equal?(result[0])
+      assert o1.equal?(o2.p2)
+      assert 'o1.p1', o1.p1
+      assert 'o2.p1', o2.p1
+    }
+  end
+  
+  def test_query_building
+    Rubernate.session do |runtime|
+      for i in 1..10
+        o, o.p1 = C2.new.attach, i
+      end      
+      for i in 1..10
+        res = runtime.find_by_query 'Select :o; Where(o.p1.int <= :value)', {:value => i}
+        assert_equal i, res.size
+      end
+    end
+  end
+  
+  # Test find_by_pk for object with no params
+  def test_find_by_pk
+    pk = nil
+    Rubernate.session { |runtime| pk = C3.new.attach.primary_key }
+    Thread.current[:on_load], Thread.current[:on_save] = nil
+    assert pk
+    Rubernate.session { |runtime| 
+      o = runtime.find_by_pk pk
+      assert !o.peer.dirty?
+      assert_not_nil Thread.current[:on_load]
+      o.p0 = 'o.p0'
+    }
+    assert_not_nil Thread.current[:on_save] 
+    Rubernate.session { |runtime| 
+      o = runtime.find_by_pk pk
+      assert !o.peer.dirty?
+      assert_equal 'o.p0', o.p0 
+    }    
+  end
+ 
+  def test_remove
+    pk = nil
+    Rubernate.session { |runtime| pk = C3.new.attach.primary_key }
+    Rubernate.session { |runtime| 
+      o = runtime.find_by_pk pk
+      assert o
+      o.remove!
+      assert o.removed?
+      assert_raise(RuntimeError) { o.primary_key }      
+    }
+    Rubernate.session{ |runtime|
+      assert_raise(Rubernate::ObjectNotFoundException) {runtime.find_by_pk pk}
+    }
+  end
+  
+  def test_find_by_class
+    Rubernate.session do |runtime|
+      for i in 1..10
+        o, o.p1 = C2.new.attach, i
+      end
+      res = runtime.find_by_query 'Select :o; Where(o.klass == :c)', {:c => C2}
+      for o in res
+        o.p2 = o.p1 * o.p1
+      end
+      res = runtime.find_by_query 'Select :o; Where o.klass == :c; OrderBy o.p2.int', {:c => C2}
+      for i in 1..10
+        o = res[i - 1]
+        assert_equal i, o.p1
+        assert_equal i * i, o.p2
+      end      
+    end
+  end
+  
+  def test_ref_store
+    o1pk, o2pk = nil
+    Rubernate.session do
+      o1 = C2.new.attach
+      o2 = C1.new.attach
+      o1.p1 = o1
+      o1.p2 = o2
+      o1pk = o1.primary_key
+      o2pk = o2.primary_key 
+    end
+    Rubernate.session do
+      o1 = find_by_pk o1pk
+      o2 = find_by_pk o2pk
+      assert_equal o1, o1.p1
+      assert_equal o2, o1.p2
+    end
+  end
+
+  def test_array_store
+    o1pk, o2pk = nil
+    Rubernate.session do
+      o1 = C1.new.attach
+      o2 = C2.new.attach
+      o1.p1 = [o2, o1]      
+      o1pk, o2pk = o1.primary_key, o2.primary_key 
+    end
+    
+    Rubernate.session do
+      o1 = find_by_pk o1pk
+      o2 = find_by_pk o2pk
+      assert_equal [o1, o2].to_set, o1.p1.to_set
+    end
+  end  
+
+  def test_array_change
+    o1pk, o2pk = nil
+    Rubernate.session do
+      o1 = C1.new.attach
+      o1pk = o1.primary_key
+      o1.p1 = [o1]
+    end
+    Rubernate.session do
+      o1 = find_by_pk o1pk
+      assert_equal [o1], o1.p1
+      o2 = C2.new.attach
+      o2pk = o2.primary_key
+      o1.p1 << o2
+    end    
+    Rubernate.session do 
+      o1 = find_by_pk o1pk
+      o2 = find_by_pk o2pk
+      assert_equal [o1, o2].to_set, o1.p1.to_set
+    end
+  end
+
+  def test_remove_referred
+    pk1, pk2 = nil
+    Rubernate.session do
+      o1,  o2  = C1.new.attach, C1.new.attach
+      pk1, pk2 = o1.primary_key, o2.primary_key
+      o1.p1 = o2
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      o2 = find_by_pk pk2
+      assert_equal o2, o1.p1
+      o2.remove!
+      assert o1.p1.removed?
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      assert_nil o1.p1
+    end
+  end
+
+  def test_remove_referred_object_array
+    pk1, pk2 = nil
+    Rubernate.session do
+      o1,  o2  = C1.new.attach, C1.new.attach
+      pk1, pk2 = o1.primary_key, o2.primary_key
+      o1.p1 = [o2]
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      o2 = find_by_pk pk2
+      assert_equal [o2], o1.p1
+      o2.remove!
+      o1.p1 << o1
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      assert_equal [o1], o1.p1
+    end
+  end
+  
+  def test_remove_referred_object_hash
+    pk1, pk2 = nil
+    Rubernate.session do
+      o1,  o2  = C1.new.attach, C1.new.attach
+      pk1, pk2 = o1.primary_key, o2.primary_key
+      o1.p1 = {1 => o2}
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      o2 = find_by_pk pk2
+      assert_equal ({1 => o2}), o1.p1
+      o2.remove!
+      o1.p1[2] = o1
+    end
+    Rubernate.session do
+      o1 = find_by_pk pk1
+      assert_equal ({2 => o1}), o1.p1
+    end
+  end
+  
+  def test_remove_referred_externally
+    pk1, pk2 = nil
+    Rubernate.session do
+      o1,  o2  = C1.new.attach, C1.new.attach
+      pk1, pk2 = o1.primary_key, o2.primary_key
+      o1.p1 = o2, o1
+    end
+    Rubernate.session do
+      dbh.do 'delete from r_params where object_pk = ? and ref_value = ?', pk1, pk2
+      o1 = find_by_pk pk1
+      assert_equal [o1], o1.p1
+    end
+  end
+  
+  def test_attach_initialized
+    o1, o1.p1, pk1 = C1.new, 'obj1'
+    o2, o2.p1, pk2 = C2.new, o1
+    o2.p2 = [o2]
+    Rubernate.session {o1.attach; pk1 = o1.primary_key; o2.attach; pk2 = o2.primary_key}
+    Rubernate.session do 
+      o1 = find_by_pk pk1
+      o2 = find_by_pk pk2
+      assert_equal 'obj1', o1.p1
+      assert_equal o1, o2.p1
+      assert_equal [o2], o2.p2
+    end
+  end  
+end
+end

data/tests/rubernate/impl/dbimysql_test.rb

@@ -0,0 +1,19 @@
+$:.unshift(File.expand_path('../../..', __FILE__)) unless 
+  $:.include?(File.expand_path('../../..', __FILE__))
+
+require 'rubernate/fixtures' 
+require 'rubernate/impl/dbigeneric_stub'
+
+class Rubernate::DBI::MysqlTest < Test::Unit::TestCase
+  include Rubernate::DBI::GenericTests  
+  def setup
+    init MYSQL_RUNTIME_CLASS, MYSQL_DB_URL, MYSQL_DB_USER, MYSQL_DB_PWD
+  end
+end
+
+class Rubernate::DBI::MysqlRuntimeTest < Test::Unit::TestCase
+  include Rubernate::GenericRuntimeTests  
+  def setup
+    init MYSQL_RUNTIME_CLASS, MYSQL_DB_URL, MYSQL_DB_USER, MYSQL_DB_PWD
+  end
+end

data/tests/rubernate/impl/dbioracle_test.rb

@@ -0,0 +1,19 @@
+$:.unshift(File.expand_path('../../..', __FILE__)) unless 
+  $:.include?(File.expand_path('../../..', __FILE__))
+
+require 'rubernate/fixtures' 
+require 'rubernate/impl/dbigeneric_stub'
+
+class Rubernate::DBI::OracleTest < Test::Unit::TestCase
+  include Rubernate::DBI::GenericTests 
+  def setup
+    init ORA_RUNTIME_CLASS, ORA_DB_URL, ORA_DB_USER, ORA_DB_PWD      
+  end
+end
+ 
+class Rubernate::DBI::OracleRuntimeTest < Test::Unit::TestCase
+  include Rubernate::GenericRuntimeTests  
+  def setup
+    init ORA_RUNTIME_CLASS, ORA_DB_URL, ORA_DB_USER, ORA_DB_PWD
+  end
+end