updateable_views_inheritance 1.1.1

data/test/fixtures/migrations/6_single_table_inheritance_view.rb

@@ -0,0 +1,10 @@
+class SingleTableInheritanceView < ActiveRecord::Migration
+  def self.up
+    rebuild_parent_and_children_views(:locomotives)
+    create_single_table_inheritance_view(:all_locomotives,:locomotives)
+  end
+
+  def self.down
+    drop_view(:all_locomotives)
+  end
+end

data/test/fixtures/migrations/7_second_deep_hierarchy.rb

@@ -0,0 +1,70 @@
+class SecondDeepHierarchy < ActiveRecord::Migration
+  # Create tables and views for the following inheritance hierarchy:
+  #
+  #                               vehicles
+  #       ____________________________|_______________________________
+  #       |                           |                               |
+  # wheeled_vehicles            railed_vehicles                     boats
+  #   ____|_____                      |
+  #   |         |                     |
+  # bicycles  cars                  trains
+  #                   ________________|________________
+  #                   |               |               |
+  #             steam_trains    rack_trains     electric_trains
+  #                                                   |
+  #                                               maglev_trains
+  def self.up
+    create_table :vehicles do |t|
+      t.column :name, :string
+      t.column :vehicle_type, :string
+    end
+    
+    create_child(:wheeled_vehicles, :parent => :vehicles) do |t|
+      t.column :number_of_wheels, :integer
+    end
+    
+    create_child(:bicycles, :parent => :wheeled_vehicles) do |t|
+      t.column :number_of_gears, :integer
+    end
+    
+    create_child(:cars, :parent => :wheeled_vehicles) do |t|
+      t.column :number_of_doors, :integer
+    end
+    
+    create_child(:railed_vehicles, :parent => :vehicles) do |t|
+      t.column :number_of_rails, :integer
+    end
+    
+    create_child(:trains, :parent => :railed_vehicles) do |t|
+      t.column :max_speed, :integer
+    end
+  
+    create_child(:steam_trains, :parent => :trains) do |t|
+      t.column :water_consumption, :decimal, :precision => 6, :scale => 2
+      t.column :coal_consumption,  :decimal, :precision => 6, :scale => 2
+    end
+    
+    create_child(:rack_trains, :parent => :trains) do |t|
+      t.column :bidirectional, :boolean, :default => false
+      t.column :narrow_gauge, :boolean, :default => true
+      t.column :rail_system, :string, :default => 'Abt'
+    end
+
+    create_child(:electric_trains, :parent => :trains) do |t|
+      t.column :electricity_consumption, :decimal, :precision => 6, :scale => 2
+    end
+    
+    create_child(:maglev_trains, :parent => :electric_trains) do |t|
+      t.column :magnetic_field, :integer
+    end
+
+    create_child(:boats, :parent => :vehicles) do |t|
+      t.column :mast_number, :integer
+    end
+  end
+  
+  def self.down
+    %w(boats maglev_trains electric_trains steam_trains rack_trains trains railed_vehicles cars bicycles wheeled_vehicles).each{|child| drop_child(child)}
+    drop_table :vehicles
+  end
+end

data/test/fixtures/migrations/8_second_single_table_inheritance_view.rb

@@ -0,0 +1,10 @@
+class SecondSingleTableInheritanceView < ActiveRecord::Migration
+  def self.up
+    rebuild_parent_and_children_views(:vehicles)
+    create_single_table_inheritance_view(:all_vehicles,:vehicles)
+  end
+
+  def self.down
+    drop_view(:all_vehicles)
+  end
+end

data/test/fixtures/rack_trains.yml

@@ -0,0 +1,7 @@
+rack:
+  id: 4
+  name: 'Rack'
+  vehicle_type: 'RackTrain'
+  number_of_rails: 2
+  max_speed: 100
+  bidirectional: true

data/test/fixtures/steam_locomotives.yml

@@ -0,0 +1,7 @@
+rocket:
+  id: 1
+  name: 'Stephenson Rocket'
+  type: 'SteamLocomotive'
+  max_speed: 20
+  water_consumption: 13
+  coal_consumption: 20

data/test/fixtures/steam_trains.yml

@@ -0,0 +1,8 @@
+steam_tr:
+  id: 3
+  name: 'Steam Train'
+  vehicle_type: 'SteamTrain'
+  number_of_rails: 2
+  max_speed: 21
+  water_consumption: 14
+  coal_consumption: 24

data/test/migration_test.rb

@@ -0,0 +1,30 @@
+require File.join(File.dirname(__FILE__), 'test_helper')
+
+class ClassTableInheritanceMigrationTest < ActiveSupport::TestCase
+  # We use transactional fixtures - migration from the setup is rolled back by Rails on teardown
+  def setup
+    @connection = ActiveRecord::Base.connection
+  end
+
+  def test_create_child_default
+    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 2)
+    assert_equal %w(coal_consumption id max_speed name type water_consumption),
+                 @connection.columns(:steam_locomotives).map{ |c| c.name }.sort
+  end
+
+  def test_create_child_explicit_table
+    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 3)
+    assert_equal %w(electricity_consumption id max_speed name type),
+                 @connection.columns(:electric_locomotives).map{ |c| c.name }.sort
+  end
+  
+  def test_drop_child
+    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 3)
+    ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/', 2)
+    assert_equal %w(steam_locomotives), @connection.views.sort
+    assert_equal %w(locomotives
+                    schema_migrations
+                    steam_locomotives_data
+                    updateable_views_inheritance), @connection.tables.sort
+  end
+end

data/test/schema_test.rb

@@ -0,0 +1,179 @@
+require File.join(File.dirname(__FILE__), 'test_helper')
+
+class ClassTableInheritanceSchemaTest < ActiveSupport::TestCase  
+  def setup
+    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 5)
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  def test_pk_and_sequence_for
+    assert_equal ['id', 'public.locomotives_id_seq'], @connection.pk_and_sequence_for(:maglev_locomotives), "Could not get pk and sequence for child aggregate view"
+  end
+  
+  def test_primary_key
+    assert_equal 'id', @connection.primary_key(:maglev_locomotives), "Wrong or no primary key for child aggregate view"
+  end
+  
+  
+  def test_content_columns
+    assert !SteamLocomotive.content_columns.include?("id")
+  end
+  
+  def test_views
+    assert_equal ["electric_locomotives", "maglev_locomotives", "rack_locomotives", "steam_locomotives"], 
+                 @connection.views.sort
+  end
+  
+  class ParentTableWithOnlyOneColumn < ActiveRecord::Migration
+    def self.up
+      create_table(:parent_pk_only){}
+      create_table :child_data do |t|
+        t.column :name, :string
+      end
+      create_child_view :parent_pk_only, :child
+    end
+  end
+  
+  def test_parent_table_with_only_one_column
+    ParentTableWithOnlyOneColumn.up
+    assert @connection.views.include?('child')
+    assert_equal %w(id name), @connection.columns(:child).map{|c| c.name}.sort
+  end
+  
+  class ChildTableWithOnlyOneColumn < ActiveRecord::Migration
+    def self.up
+      create_table :parent do |t|
+        t.column :name, :string
+      end
+      create_table(:child_pk_only_data){}
+      create_child_view :parent, :child_pk_only
+    end
+  end
+  
+  def test_child_table_with_only_one_column
+    ChildTableWithOnlyOneColumn.up
+    assert @connection.views.include?('child_pk_only'), "Could not create child view when child table has only one column"
+    assert_equal %w(id name), @connection.columns(:child_pk_only).map{|c| c.name}.sort
+  end
+  
+  def test_serial_pk_has_default_in_view
+    assert_equal %q|nextval('locomotives_id_seq'::regclass)|,
+                 @connection.query(<<-end_sql, 'Serial PK has default in view')[0][0]
+                   SELECT pg_catalog.pg_get_expr(d.adbin, d.adrelid) as constraint
+                     FROM pg_catalog.pg_attrdef d, pg_catalog.pg_attribute a, pg_catalog.pg_class c
+                    WHERE d.adrelid = a.attrelid 
+                      AND d.adnum = a.attnum 
+                      AND a.atthasdef 
+                      AND c.relname = 'maglev_locomotives' 
+                      AND a.attrelid = c.oid
+                      AND a.attname = 'id'
+                      AND a.attnum > 0 AND NOT a.attisdropped
+                 end_sql
+  end
+
+  def test_default_value_string_of_view_column
+    RackLocomotive.reset_column_information
+    assert_equal 'Abt', RackLocomotive.new.rail_system
+  end
+  
+  def test_default_value_boolean_of_view_column
+    assert !RackLocomotive.new.bidirectional
+    assert RackLocomotive.new.narrow_gauge
+  end
+    
+  class ChangeDefaultValueOfColumn < ActiveRecord::Migration
+    def self.up
+      remove_parent_and_children_views(:rack_locomotives)
+      change_column_default(:rack_locomotives_data, :rail_system, 'Marsh')
+      rebuild_parent_and_children_views(:rack_locomotives)
+    end
+  end
+  
+  def test_change_default_value_of_column
+    ChangeDefaultValueOfColumn.up
+    RackLocomotive.reset_column_information
+    assert_equal 'Marsh', RackLocomotive.new.rail_system
+  end
+  
+  class RemoveChildrenViews < ActiveRecord::Migration
+    def self.up
+      remove_parent_and_children_views(:locomotives)
+    end
+  end
+    
+  def test_remove_parent_and_children_views
+    RemoveChildrenViews.up
+    assert @connection.views.empty?
+  end
+    
+  class RemoveColumnInParentTable < ActiveRecord::Migration
+    def self.up
+      remove_parent_and_children_views(:locomotives)
+      remove_column(:locomotives, :max_speed)
+      rebuild_parent_and_children_views(:locomotives)
+    end
+  end
+  
+  def test_remove_column_parent_table
+    RemoveColumnInParentTable.up
+    assert_equal %w(coal_consumption id name type water_consumption),
+                 @connection.columns(:steam_locomotives).map{ |c| c.name }.sort
+    assert_equal %w(electricity_consumption id magnetic_field name type),
+                 @connection.columns(:maglev_locomotives).map{ |c| c.name }.sort
+  end
+  
+  class RenameColumnInParentTable < ActiveRecord::Migration
+    def self.up
+      ActiveRecord::Fixtures.create_fixtures(File.dirname(__FILE__) + '/fixtures/', :electric_locomotives)
+      ActiveRecord::Fixtures.create_fixtures(File.dirname(__FILE__) + '/fixtures/', :maglev_locomotives)
+      ActiveRecord::Fixtures.create_fixtures(File.dirname(__FILE__) + '/fixtures/', :steam_locomotives)
+      remove_parent_and_children_views(:locomotives)
+      rename_column(:locomotives, :max_speed, :maximal_speed)
+      rebuild_parent_and_children_views(:locomotives)
+
+    end
+  end
+
+  def test_rename_column_parent_table
+    RenameColumnInParentTable.up
+    assert_equal %w(coal_consumption id maximal_speed name type water_consumption),
+                 @connection.columns(:steam_locomotives).map{ |c| c.name }.sort
+    assert_equal %w(electricity_consumption id magnetic_field maximal_speed name type),
+                 @connection.columns(:maglev_locomotives).map{ |c| c.name }.sort
+    
+  end
+  
+  class AddColumnToParentTable < ActiveRecord::Migration
+    def self.up
+      add_column(:raw_electric_locomotives, :number_of_engines, :integer)
+      rebuild_parent_and_children_views(:electric_locomotives)
+    end
+  end
+  
+  def test_add_column_to_parent_table
+    AddColumnToParentTable.up
+    assert_equal %w(electricity_consumption id max_speed name number_of_engines type),
+                 @connection.columns(:electric_locomotives).map{ |c| c.name }.sort
+    assert_equal %w(electricity_consumption id magnetic_field max_speed name number_of_engines type),
+                 @connection.columns(:maglev_locomotives).map{ |c| c.name }.sort
+    
+  end
+  
+  class ChangeChildRelationView < ActiveRecord::Migration
+    def self.up
+      remove_parent_and_children_views(:electric_locomotives)
+      rename_column(:raw_electric_locomotives, :electricity_consumption, :electric_consumption)
+      rebuild_parent_and_children_views(:electric_locomotives)
+    end
+  end
+  
+  def test_change_child_relation_view
+    ChangeChildRelationView.up
+    assert_equal %w(electric_consumption id max_speed name type),
+                 @connection.columns(:electric_locomotives).map{ |c| c.name }.sort
+  end
+  
+  def test_table_exists
+    #TODO: test table_exists? monkey patch
+  end
+end

data/test/single_table_inheritance.rb

@@ -0,0 +1,141 @@
+require File.join(File.dirname(__FILE__), 'test_helper')
+
+class SingleTableInheritanceAggregateViewTest < ActiveSupport::TestCase  
+  def setup
+    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 6)
+    # order of fixtures is important for the test - last loaded should not be with max(id)
+    %w(electric_locomotives maglev_locomotives steam_locomotives).each do |f|
+      ActiveRecord::Fixtures.create_fixtures(File.dirname(__FILE__) + '/fixtures/', f)
+    end
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  def test_single_table_inheritance_view_schema
+    @connection = ActiveRecord::Base.connection
+    assert_equal %w(coal_consumption id max_speed name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+  end
+  
+  def test_single_table_inheritance_records_number
+    assert_equal Locomotive.count, @connection.query("SELECT count(*) FROM all_locomotives").first.first.to_i
+    assert_equal SteamLocomotive.count, @connection.query("SELECT count(*) FROM all_locomotives WHERE type='SteamLocomotive'").first.first.to_i
+    assert_equal ElectricLocomotive.count - MaglevLocomotive.count, @connection.query("SELECT count(*) FROM all_locomotives WHERE type='ElectricLocomotive'").first.first.to_i
+    assert_equal RackLocomotive.count, @connection.query("SELECT count(*) FROM all_locomotives WHERE type='RackLocomotive'").first.first.to_i
+    assert_equal MaglevLocomotive.count, @connection.query("SELECT count(*) FROM all_locomotives WHERE type='MaglevLocomotive'").first.first.to_i
+  end
+  
+  def test_single_table_inheritance_save
+    electric_locomotive = ElectricLocomotive.new(:name=> 'BoBo', :max_speed => 40, :electricity_consumption => 12)
+    assert electric_locomotive.save
+    assert_equal electric_locomotive.name, @connection.query("SELECT name FROm all_locomotives WHERE id=#{electric_locomotive.id}").first.first
+  end
+  
+  class AddColumnToParentTable < ActiveRecord::Migration
+    def self.up
+      add_column(:raw_electric_locomotives, :number_of_engines, :integer)
+      drop_view(:all_locomotives)
+      rebuild_parent_and_children_views(:electric_locomotives)
+      create_single_table_inheritance_view(:all_locomotives, :locomotives)
+    end
+  end
+  
+  def test_single_table_inheritance_view_add_column_to_parent_table
+    AddColumnToParentTable.up
+    assert_equal %w(coal_consumption id max_speed name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system number_of_engines).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+  end
+  
+  class RemoveColumnInParentTable < ActiveRecord::Migration
+    def self.up
+      drop_view(:all_locomotives)
+      remove_parent_and_children_views(:locomotives)
+      remove_column(:locomotives, :max_speed)
+      rebuild_parent_and_children_views(:locomotives)
+      create_single_table_inheritance_view(:all_locomotives,:locomotives)
+    end
+  end
+  
+  def test_single_table_inheritance_view_remove_column_parent_table
+    RemoveColumnInParentTable.up
+    assert_equal %w(coal_consumption id name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+  end
+  
+  class RenameColumnInParentTable < ActiveRecord::Migration
+    def self.up
+      drop_view(:all_locomotives)
+      remove_parent_and_children_views(:locomotives)
+      rename_column(:locomotives, :max_speed, :maximal_speed)
+      rebuild_parent_and_children_views(:locomotives)
+      create_single_table_inheritance_view(:all_locomotives, :locomotives) 
+    end
+  end
+
+  def test_single_table_inheritance_view_rename_column_parent_table
+    RenameColumnInParentTable.up
+    assert_equal %w(coal_consumption id maximal_speed name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+  end
+  
+  class ChangeChildRelationView < ActiveRecord::Migration
+    def self.up
+      drop_view(:all_locomotives)
+      remove_parent_and_children_views(:electric_locomotives)
+      rename_column(:raw_electric_locomotives, :electricity_consumption, :electric_consumption)
+      rebuild_parent_and_children_views(:electric_locomotives)
+      create_single_table_inheritance_view(:all_locomotives, :locomotives)
+    end
+  end
+  
+  def test_single_table_inheritance_view_change_child_relation_view
+    ChangeChildRelationView.up
+    assert_equal %w(coal_consumption id max_speed name type water_consumption electric_consumption bidirectional narrow_gauge magnetic_field rail_system).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+  end
+  
+  class ConflictColumns < ActiveRecord::Migration
+    def self.up
+      drop_view(:all_locomotives)
+      add_column(:raw_electric_locomotives, :number_of_engines, :integer)
+      add_column(:steam_locomotives_data, :number_of_engines, :string)
+      rebuild_parent_and_children_views(:electric_locomotives)
+      rebuild_parent_and_children_views(:steam_locomotives)
+      create_single_table_inheritance_view(:all_locomotives, :locomotives)
+    end
+  end
+  
+  def test_single_table_inheritance_view_conflict_columns
+    ConflictColumns.up
+    assert_equal %w(coal_consumption id max_speed name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system number_of_engines).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+    assert_equal 'text', @connection.columns(:all_locomotives).detect{|c| c.name == "number_of_engines"}.sql_type
+  end
+  
+  class ConflictColumnsWithValues < ActiveRecord::Migration
+    def self.up
+      add_column(:raw_electric_locomotives, :number_of_engines, :integer)
+      add_column(:steam_locomotives_data, :number_of_engines, :string)
+      execute("UPDATE raw_electric_locomotives SET number_of_engines = 2")
+      execute("UPDATE steam_locomotives_data SET number_of_engines = 'one'")
+      drop_view(:all_locomotives)
+      rebuild_parent_and_children_views(:electric_locomotives)
+      rebuild_parent_and_children_views(:steam_locomotives)
+      create_single_table_inheritance_view(:all_locomotives, :locomotives)
+    end
+  end
+  
+  def test_single_table_inheritance_view_conflict_columns_with_values
+    ConflictColumnsWithValues.up
+    assert_equal %w(coal_consumption id max_speed name type water_consumption electricity_consumption bidirectional narrow_gauge magnetic_field rail_system number_of_engines).sort,
+                 @connection.columns(:all_locomotives).map{ |c| c.name }.sort
+    assert_equal 'text', @connection.columns(:all_locomotives).detect{|c| c.name == "number_of_engines"}.sql_type
+    assert_equal 'one', @connection.query("SELECT number_of_engines FROM all_locomotives WHERE id=#{SteamLocomotive.find(:first).id}").first.first
+    assert_equal '2', @connection.query("SELECT number_of_engines FROM all_locomotives WHERE id=#{ElectricLocomotive.find(:first).id}").first.first
+  end
+  
+  def test_respond_to_missing_attributes
+    Locomotive.table_name = :all_locomotives
+    assert !MaglevLocomotive.new.respond_to?(:non_existing_attribute_in_the_hierarchy), "Active Record is gone haywire - responds to attributes that are never defined"
+    assert !MaglevLocomotive.new.respond_to?(:coal_consumption), "Responds to an attribute not defined in the class' view but in the STI view"
+  end
+end