pg_morph 0.3.0

data/spec/dummy/config/initializers/backtrace_silencers.rb

@@ -0,0 +1,7 @@
+# Be sure to restart your server when you modify this file.
+
+# You can add backtrace silencers for libraries that you're using but don't wish to see in your backtraces.
+# Rails.backtrace_cleaner.add_silencer { |line| line =~ /my_noisy_library/ }
+
+# You can also remove all the silencers if you're trying to debug a problem that might stem from framework code.
+# Rails.backtrace_cleaner.remove_silencers!

data/spec/dummy/config/initializers/filter_parameter_logging.rb

@@ -0,0 +1,4 @@
+# Be sure to restart your server when you modify this file.
+
+# Configure sensitive parameters which will be filtered from the log file.
+Rails.application.config.filter_parameters += [:password]

data/spec/dummy/config/initializers/inflections.rb

@@ -0,0 +1,16 @@
+# Be sure to restart your server when you modify this file.
+
+# Add new inflection rules using the following format. Inflections
+# are locale specific, and you may define rules for as many different
+# locales as you wish. All of these examples are active by default:
+# ActiveSupport::Inflector.inflections(:en) do |inflect|
+#   inflect.plural /^(ox)$/i, '\1en'
+#   inflect.singular /^(ox)en/i, '\1'
+#   inflect.irregular 'person', 'people'
+#   inflect.uncountable %w( fish sheep )
+# end
+
+# These inflection rules are supported but not enabled by default:
+# ActiveSupport::Inflector.inflections(:en) do |inflect|
+#   inflect.acronym 'RESTful'
+# end

data/spec/dummy/config/initializers/mime_types.rb

@@ -0,0 +1,5 @@
+# Be sure to restart your server when you modify this file.
+
+# Add new mime types for use in respond_to blocks:
+# Mime::Type.register "text/richtext", :rtf
+# Mime::Type.register_alias "text/html", :iphone

data/spec/dummy/config/initializers/secret_token.rb

@@ -0,0 +1,12 @@
+# Be sure to restart your server when you modify this file.
+
+# Your secret key is used for verifying the integrity of signed cookies.
+# If you change this key, all old signed cookies will become invalid!
+
+# Make sure the secret is at least 30 characters and all random,
+# no regular words or you'll be exposed to dictionary attacks.
+# You can use `rake secret` to generate a secure secret key.
+
+# Make sure your secret_key_base is kept private
+# if you're sharing your code publicly.
+Dummy::Application.config.secret_key_base = 'b2e679b42a15ee93bd8344916b0a91c888b96a1707b76675e5b6c51b7849c7a41b461cc18ffff1ea9ce025d2bf9f902db938339285a7d66af630dc9a7b9b6eff'

data/spec/dummy/config/initializers/session_store.rb

@@ -0,0 +1,3 @@
+# Be sure to restart your server when you modify this file.
+
+Dummy::Application.config.session_store :cookie_store, key: '_dummy_session'

data/spec/dummy/config/initializers/wrap_parameters.rb

@@ -0,0 +1,14 @@
+# Be sure to restart your server when you modify this file.
+
+# This file contains settings for ActionController::ParamsWrapper which
+# is enabled by default.
+
+# Enable parameter wrapping for JSON. You can disable this by setting :format to an empty array.
+ActiveSupport.on_load(:action_controller) do
+  wrap_parameters format: [:json] if respond_to?(:wrap_parameters)
+end
+
+# To enable root element in JSON for ActiveRecord objects.
+# ActiveSupport.on_load(:active_record) do
+#  self.include_root_in_json = true
+# end

data/spec/dummy/config/locales/en.yml

@@ -0,0 +1,23 @@
+# Files in the config/locales directory are used for internationalization
+# and are automatically loaded by Rails. If you want to use locales other
+# than English, add the necessary files in this directory.
+#
+# To use the locales, use `I18n.t`:
+#
+#     I18n.t 'hello'
+#
+# In views, this is aliased to just `t`:
+#
+#     <%= t('hello') %>
+#
+# To use a different locale, set it with `I18n.locale`:
+#
+#     I18n.locale = :es
+#
+# This would use the information in config/locales/es.yml.
+#
+# To learn more, please read the Rails Internationalization guide
+# available at http://guides.rubyonrails.org/i18n.html.
+
+en:
+  hello: "Hello world"

data/spec/dummy/config/routes.rb

@@ -0,0 +1,56 @@
+Dummy::Application.routes.draw do
+  # The priority is based upon order of creation: first created -> highest priority.
+  # See how all your routes lay out with "rake routes".
+
+  # You can have the root of your site routed with "root"
+  # root 'welcome#index'
+
+  # Example of regular route:
+  #   get 'products/:id' => 'catalog#view'
+
+  # Example of named route that can be invoked with purchase_url(id: product.id)
+  #   get 'products/:id/purchase' => 'catalog#purchase', as: :purchase
+
+  # Example resource route (maps HTTP verbs to controller actions automatically):
+  #   resources :products
+
+  # Example resource route with options:
+  #   resources :products do
+  #     member do
+  #       get 'short'
+  #       post 'toggle'
+  #     end
+  #
+  #     collection do
+  #       get 'sold'
+  #     end
+  #   end
+
+  # Example resource route with sub-resources:
+  #   resources :products do
+  #     resources :comments, :sales
+  #     resource :seller
+  #   end
+
+  # Example resource route with more complex sub-resources:
+  #   resources :products do
+  #     resources :comments
+  #     resources :sales do
+  #       get 'recent', on: :collection
+  #     end
+  #   end
+
+  # Example resource route with concerns:
+  #   concern :toggleable do
+  #     post 'toggle'
+  #   end
+  #   resources :posts, concerns: :toggleable
+  #   resources :photos, concerns: :toggleable
+
+  # Example resource route within a namespace:
+  #   namespace :admin do
+  #     # Directs /admin/products/* to Admin::ProductsController
+  #     # (app/controllers/admin/products_controller.rb)
+  #     resources :products
+  #   end
+end

data/spec/dummy/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment',  __FILE__)
+run Rails.application

data/spec/dummy/db/migrate/20140401124656_create_comments.rb

@@ -0,0 +1,9 @@
+class CreateComments < ActiveRecord::Migration
+  def change
+    create_table :comments do |t|
+      t.string :content
+
+      t.timestamps
+    end
+  end
+end

data/spec/dummy/db/migrate/20140401124705_create_posts.rb

@@ -0,0 +1,9 @@
+class CreatePosts < ActiveRecord::Migration
+  def change
+    create_table :posts do |t|
+      t.string :content
+
+      t.timestamps
+    end
+  end
+end

data/spec/dummy/db/migrate/20140401124727_create_likes.rb

@@ -0,0 +1,10 @@
+class CreateLikes < ActiveRecord::Migration
+  def change
+    create_table :likes do |t|
+      t.string :likeable_type
+      t.integer :likeable_id
+
+      t.timestamps
+    end
+  end
+end

data/spec/dummy/db/schema.rb

@@ -0,0 +1,38 @@
+# encoding: UTF-8
+# This file is auto-generated from the current state of the database. Instead
+# of editing this file, please use the migrations feature of Active Record to
+# incrementally modify your database, and then regenerate this schema definition.
+#
+# Note that this schema.rb definition is the authoritative source for your
+# database schema. If you need to create the application database on another
+# system, you should be using db:schema:load, not running all the migrations
+# from scratch. The latter is a flawed and unsustainable approach (the more migrations
+# you'll amass, the slower it'll run and the greater likelihood for issues).
+#
+# It's strongly recommended that you check this file into your version control system.
+
+ActiveRecord::Schema.define(version: 20140401124727) do
+
+  # These are extensions that must be enabled in order to support this database
+  enable_extension "plpgsql"
+
+  create_table "comments", force: true do |t|
+    t.string   "content"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+  end
+
+  create_table "likes", force: true do |t|
+    t.string   "likeable_type"
+    t.integer  "likeable_id"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+  end
+
+  create_table "posts", force: true do |t|
+    t.string   "content"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+  end
+
+end

data/spec/dummy/db/seeds.rb

@@ -0,0 +1,7 @@
+# This file should contain all the record creation needed to seed the database with its default values.
+# The data can then be loaded with the rake db:seed (or created alongside the db with db:setup).
+#
+# Examples:
+#
+#   cities = City.create([{ name: 'Chicago' }, { name: 'Copenhagen' }])
+#   Mayor.create(name: 'Emanuel', city: cities.first)

data/spec/pg_morph/adapter_integration_spec.rb

@@ -0,0 +1,89 @@
+require 'spec_helper'
+
+describe PgMorph::Adapter do
+  ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.class_eval do
+    include PgMorph::Adapter
+
+    def run(query)
+      ActiveRecord::Base.connection.select_value(query)
+    end
+  end
+
+  before do
+    @adapter = ActiveRecord::Base.connection
+    @comments_polymorphic = PgMorph::Polymorphic.new(:likes, :comments, column: :likeable)
+    @posts_polymorphic = PgMorph::Polymorphic.new(:likes, :posts, column: :likeable)
+    begin
+      Like.destroy_all
+      Comment.destroy_all
+      @adapter.remove_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+      @adapter.remove_polymorphic_foreign_key(:likes, :posts, column: :likeable)
+    rescue
+    end
+  end
+
+  after do
+    begin
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+    rescue
+    end
+  end
+
+  describe '#add_polymorphic_foreign_key' do
+    it 'creates proxy table' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      expect(@adapter.run('SELECT id FROM likes_comments')).to be_nil
+    end
+  end
+
+  describe '#remove_polymorphic_foreign_key' do
+    it 'removes proxy table' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+      expect(@adapter.run('SELECT id FROM likes_comments')).to be_nil
+
+      @adapter.remove_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      -> { @adapter.run('SELECT id FROM likes_comments') }
+        .should raise_error ActiveRecord::StatementInvalid
+    end
+  end
+
+  describe 'assertions to a partition' do
+    it 'works properly' do
+      # new record inserted correctly
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+      comment = Comment.create(content: 'comment')
+      like = Like.create(likeable: comment)
+
+      expect(Like.count).to eq(1)
+      expect(like.id).to eq(Like.last.id)
+
+      # new record with more partition tables inserted correctly
+      @adapter.add_polymorphic_foreign_key(:likes, :posts, column: :likeable)
+      post = Post.create(content: 'content')
+      like2 = Like.create(likeable: post)
+
+      expect(Like.count).to eq(2)
+      expect(like2.id).to eq(Like.last.id)
+
+      # after removing partition row not inserted
+      like.destroy
+      expect(Like.count).to eq(1)
+      @adapter.remove_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      -> {  Like.create(likeable: comment) }
+        .should raise_error ActiveRecord::StatementInvalid
+
+      # if no partitions row inserted correctly
+      like2.destroy
+      expect(Like.count).to eq(0)
+      @adapter.remove_polymorphic_foreign_key(:likes, :posts, column: :likeable)
+      like4 = Like.create(likeable: post)
+
+      expect(Like.count).to eq(1)
+      expect(like4.id).to eq(Like.last.id)
+    end
+  end
+
+end

data/spec/pg_morph/adapter_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+
+describe PgMorph::Adapter do
+  class FakeAdapter
+    include PgMorph::Adapter
+
+    def execute(sql, name = nil)
+      sql_statements << sql
+      sql
+    end
+
+    def sql_statements
+      @sql_statements || []
+    end
+  end
+
+  before do
+    @adapter = FakeAdapter.new
+    @connection = ActiveRecord::Base.connection
+  end
+
+  it 'add_polymorphic_foreign_key for non postgres adapter' do
+    expect { @adapter.add_polymorphic_foreign_key :likes, :posts, column: :likeable }
+      .to raise_error PgMorph::Exception
+  end
+
+  it 'remove_polymorphic_foreign_key for non postgres adapter' do
+    expect { @adapter.remove_polymorphic_foreign_key :likes, :posts, column: :likeable }
+      .to raise_error PgMorph::Exception
+  end
+end

data/spec/pg_morph/naming_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe PgMorph::Naming do
+
+  Fake = Struct.new(:parent_table, :child_table, :column_name) do
+    include PgMorph::Naming
+  end
+
+  before do
+    @fake = Fake.new(:foos, :bars, :baz)
+  end
+
+  it { expect(@fake.type).to eq('Bar') }
+
+  it { expect(@fake.column_name_type).to eq('baz_type') }
+
+  it { expect(@fake.column_name_id).to eq('baz_id') }
+
+  it { expect(@fake.proxy_table).to eq('foos_bars') }
+
+  it { expect(@fake.before_insert_fun_name).to eq('foos_baz_fun') }
+
+  it { expect(@fake.before_insert_trigger_name).to eq('foos_baz_insert_trigger') }
+
+  it { expect(@fake.after_insert_fun_name).to eq('delete_from_foos_master_fun') }
+
+  it { expect(@fake.after_insert_trigger_name).to eq('foos_after_insert_trigger') }
+
+end

data/spec/pg_morph/polymorphic_integration_spec.rb

@@ -0,0 +1,86 @@
+require 'spec_helper'
+
+describe PgMorph::Polymorphic do
+  ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.class_eval do
+    include PgMorph::Adapter
+
+    def run(query)
+      ActiveRecord::Base.connection.select_value(query)
+    end
+  end
+
+  before do
+    @adapter = ActiveRecord::Base.connection
+    @comments_polymorphic = PgMorph::Polymorphic.new(:likes, :comments, column: :likeable)
+    @posts_polymorphic = PgMorph::Polymorphic.new(:likes, :posts, column: :likeable)
+    begin
+      Like.destroy_all
+      Comment.destroy_all
+      @adapter.remove_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+      @adapter.remove_polymorphic_foreign_key(:likes, :posts, column: :likeable)
+    rescue
+    end
+  end
+
+  describe '#create_trigger_body' do
+    before do
+      @adapter.stub(:raise_unless_postgres)
+    end
+
+    it 'raises error for updating trigger with duplicated partition' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      -> { @comments_polymorphic.send(:create_trigger_body) }
+        .should raise_error PG::Error
+    end
+
+    it 'updates trigger with new partition' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      @posts_polymorphic.send(:create_trigger_body).squeeze(' ').should == %Q{
+        IF (NEW.likeable_type = 'Comment') THEN
+          INSERT INTO likes_comments VALUES (NEW.*);
+        ELSIF (NEW.likeable_type = 'Post') THEN
+          INSERT INTO likes_posts VALUES (NEW.*);
+        }.squeeze(' ')
+    end
+  end
+
+  describe '#create_before_insert_trigger_sql' do
+    it 'returns sql' do
+      @comments_polymorphic.create_before_insert_trigger_sql.squeeze(' ').should == %Q{
+      DROP TRIGGER IF EXISTS likes_likeable_insert_trigger ON likes;
+      CREATE TRIGGER likes_likeable_insert_trigger
+        BEFORE INSERT ON likes
+        FOR EACH ROW EXECUTE PROCEDURE likes_likeable_fun();
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#remove_partition_table' do
+    it 'returns sql' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      @comments_polymorphic.remove_partition_table.squeeze(' ').should == %Q{ DROP TABLE IF EXISTS likes_comments; }
+    end
+  end
+
+  describe 'remove_after_insert_trigger_sql' do
+    it 'returns sql' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+
+      @comments_polymorphic.remove_after_insert_trigger_sql.squeeze(' ').should == %Q{
+        DROP TRIGGER likes_after_insert_trigger ON likes;
+        DROP FUNCTION delete_from_likes_master_fun();
+        }.squeeze(' ')
+    end
+
+    it 'returns empty string if there are more partitions' do
+      @adapter.add_polymorphic_foreign_key(:likes, :comments, column: :likeable)
+      @adapter.add_polymorphic_foreign_key(:likes, :posts, column: :likeable)
+
+      @comments_polymorphic.remove_after_insert_trigger_sql.squeeze(' ').should == ''
+    end
+  end
+
+end

data/spec/pg_morph/polymorphic_spec.rb

@@ -0,0 +1,98 @@
+require 'spec_helper'
+
+describe PgMorph::Polymorphic do
+  before do
+    @polymorphic = PgMorph::Polymorphic.new(:foos, :bars, column: :baz)
+  end
+
+  subject { @polymorphic }
+
+  it { expect(@polymorphic.column_name).to eq(:baz) }
+
+  it { expect(@polymorphic.parent_table).to eq(:foos) }
+
+  it { expect(@polymorphic.child_table).to eq(:bars) }
+
+  describe '#create_proxy_table_sql' do
+    it do
+      @polymorphic.create_proxy_table_sql.squeeze(' ').should == %Q{
+      CREATE TABLE foos_bars (
+        CHECK (baz_type = 'Bar'),
+        PRIMARY KEY (id),
+          FOREIGN KEY (baz_id) REFERENCES bars(id)
+      ) INHERITS (foos);
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#create_before_insert_trigger_fun_sql' do
+    it '' do
+      @polymorphic.should_receive(:before_insert_trigger_content)
+
+      @polymorphic.create_before_insert_trigger_fun_sql
+    end
+  end
+
+  describe '#create_trigger_body' do
+    it 'returns proper sql for new trigger' do
+      @polymorphic.send(:create_trigger_body).squeeze(' ').should == %Q{
+      IF (NEW.baz_type = 'Bar') THEN
+        INSERT INTO foos_bars VALUES (NEW.*);
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#before_insert_trigger_content' do
+    it '' do
+      @polymorphic.send(:before_insert_trigger_content) { 'my block' }.squeeze(' ').should == %Q{
+      CREATE OR REPLACE FUNCTION foos_baz_fun() RETURNS TRIGGER AS $$
+        BEGIN
+          my block
+          ELSE
+            RAISE EXCEPTION 'Wrong \"baz_type\"=\"%\" used. Create proper partition table and update foos_baz_fun function', NEW.baz_type;
+          END IF;
+        RETURN NEW;
+        END; $$ LANGUAGE plpgsql;
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#create_after_insert_trigger_fun_sql' do
+    it do
+      @polymorphic.create_after_insert_trigger_fun_sql.squeeze(' ').should == %Q{
+      CREATE OR REPLACE FUNCTION delete_from_foos_master_fun() RETURNS TRIGGER AS $$
+      BEGIN
+        DELETE FROM ONLY foos WHERE id = NEW.id;
+        RETURN NEW;
+      END; $$ LANGUAGE plpgsql;
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#create_after_insert_trigger_sql' do
+    it do
+      @polymorphic.create_after_insert_trigger_sql.squeeze(' ').should == %Q{
+      DROP TRIGGER IF EXISTS foos_after_insert_trigger ON foos;
+      CREATE TRIGGER foos_after_insert_trigger
+        AFTER INSERT ON foos
+        FOR EACH ROW EXECUTE PROCEDURE delete_from_foos_master_fun();
+      }.squeeze(' ')
+    end
+  end
+
+  describe '#remove_before_insert_trigger_sql' do
+    it 'raise error if no function' do
+      -> { @polymorphic.remove_before_insert_trigger_sql }
+        .should raise_error PG::Error
+    end
+
+    it 'returns proper sql for single child table' do
+      @polymorphic.stub(:get_function).with('foos_baz_fun').and_return('')
+
+      @polymorphic.remove_before_insert_trigger_sql.squeeze(' ').should == %Q{
+        DROP TRIGGER foos_baz_insert_trigger ON foos;
+        DROP FUNCTION foos_baz_fun();
+        }.squeeze(' ')
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,16 @@
+ENV["RAILS_ENV"] ||= 'test'
+require File.expand_path("../dummy/config/environment", __FILE__)
+require 'rspec/rails'
+require 'rspec/autorun'
+
+Dir[Rails.root.join("spec/support/**/*.rb")].each { |f| require f }
+
+RSpec.configure do |config|
+  config.fixture_path = "#{::Rails.root}/spec/fixtures"
+
+  config.use_transactional_fixtures = true
+
+  config.infer_base_class_for_anonymous_controllers = false
+
+  config.order = "random"
+end