fx-jets 0.6.3s

data/spec/acceptance_helper.rb

@@ -0,0 +1,62 @@
+require "bundler"
+
+ENV["RAILS_ENV"] = "test"
+
+RSpec.configure do |config|
+  config.around(:each) do |example|
+    Dir.chdir("spec/dummy") do
+      example.run
+    end
+  end
+
+  config.before(:suite) do
+    Dir.chdir("spec/dummy") do
+      system <<-CMD
+        git init 1>/dev/null &&
+        git add -A &&
+        git commit --no-gpg-sign --message 'initial' 1>/dev/null
+      CMD
+    end
+  end
+
+  config.after(:suite) do
+    Dir.chdir("spec/dummy") do
+      ActiveRecord::Base.connection.disconnect!
+      system <<-CMD
+        echo &&
+        rake db:environment:set db:drop db:create &&
+        git add -A &&
+        git reset --hard HEAD 1>/dev/null &&
+        rm -rf .git/ 1>/dev/null
+      CMD
+    end
+  end
+
+  def successfully(command)
+    `RAILS_ENV=test #{command}`
+    expect($?.exitstatus).to eq(0), "'#{command}' was unsuccessful"
+  end
+
+  def write_function_definition(file, contents)
+    write_definition(file, contents, "functions")
+  end
+
+  def write_trigger_definition(file, contents)
+    write_definition(file, contents, "triggers")
+  end
+
+  def write_definition(file, contents, directory)
+    File.open("db/#{directory}/#{file}.sql", File::WRONLY) do |definition|
+      definition.truncate(0)
+      definition.write(contents)
+    end
+  end
+
+  def verify_identical_definitions(def_a, def_b)
+    successfully "cmp #{def_a} #{def_b}"
+  end
+
+  def execute(command)
+    ActiveRecord::Base.connection.execute(command).first
+  end
+end

data/spec/dummy/.gitignore

@@ -0,0 +1,16 @@
+# See https://help.github.com/articles/ignoring-files for more about ignoring files.
+#
+# If you find yourself ignoring temporary files generated by your text editor
+# or operating system, you probably want to add a global ignore instead:
+#   git config --global core.excludesfile '~/.gitignore_global'
+
+# Ignore bundler config.
+/.bundle
+
+# Ignore the default SQLite database.
+/db/*.sqlite3
+/db/*.sqlite3-journal
+
+# Ignore all logfiles and tempfiles.
+/log/*.log
+/tmp

data/spec/dummy/Rakefile

@@ -0,0 +1,13 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+
+Jets.application.load_tasks
+
+unless Rake::Task.task_defined?('db:environment:set')
+  desc 'dummy task for rails versions where this task does not exist'
+  task 'db:environment:set' do
+    # no-op
+  end
+end

data/spec/dummy/bin/bundle

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../Gemfile', __FILE__)
+load Gem.bin_path('bundler', 'bundle')

data/spec/dummy/bin/rails

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+APP_PATH = File.expand_path('../../config/application',  __FILE__)
+require_relative '../config/boot'
+require 'rails/commands'

data/spec/dummy/bin/rake

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative '../config/boot'
+require 'rake'
+Rake.application.run

data/spec/dummy/config/application.rb

@@ -0,0 +1,15 @@
+require File.expand_path("../boot", __FILE__)
+
+# Pick the frameworks you want:
+require "active_record/railtie"
+
+Bundler.require(*Jets.groups)
+require "fx"
+
+module Dummy
+  class Application < Rails::Application
+    config.cache_classes = true
+    config.eager_load = false
+    config.active_support.deprecation = :stderr
+  end
+end

data/spec/dummy/config/boot.rb

@@ -0,0 +1,5 @@
+# Set up gems listed in the Gemfile.
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../../../../Gemfile", __FILE__)
+
+require "bundler/setup" if File.exist?(ENV["BUNDLE_GEMFILE"])
+$LOAD_PATH.unshift File.expand_path("../../../../lib", __FILE__)

data/spec/dummy/config/database.yml

@@ -0,0 +1,9 @@
+development: &default
+  adapter: postgresql
+  database: dummy_development
+  encoding: unicode
+  pool: 5
+
+test:
+  <<: *default
+  database: dummy_test

data/spec/dummy/config/environment.rb

@@ -0,0 +1,5 @@
+# Load the Rails application.
+require File.expand_path("../application", __FILE__)
+
+# Initialize the Rails application.
+Jets.application.initialize!

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 Jets.application

data/spec/features/functions/migrations_spec.rb

@@ -0,0 +1,65 @@
+require "spec_helper"
+
+describe "Function migrations", :db do
+  around do |example|
+    sql_definition = <<-EOS
+      CREATE OR REPLACE FUNCTION test()
+      RETURNS text AS $$
+      BEGIN
+          RETURN 'test';
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    with_function_definition(name: :test, sql_definition: sql_definition) do
+      example.run
+    end
+  end
+
+  it "can run migrations that create functions" do
+    migration = Class.new(migration_class) do
+      def up
+        create_function :test
+      end
+    end
+
+    expect { run_migration(migration, :up) }.not_to raise_error
+  end
+
+  it "can run migrations that drop functions" do
+    connection.create_function(:test)
+
+    migration = Class.new(migration_class) do
+      def up
+        drop_function :test
+      end
+    end
+
+    expect { run_migration(migration, :up) }.not_to raise_error
+  end
+
+  it "can run migrations that updates functions" do
+    connection.create_function(:test)
+
+    sql_definition = <<-EOS
+      CREATE OR REPLACE FUNCTION test()
+      RETURNS text AS $$
+      BEGIN
+          RETURN 'testest';
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    with_function_definition(
+      name: :test,
+      version: 2,
+      sql_definition: sql_definition,
+    ) do
+      migration = Class.new(migration_class) do
+        def change
+          update_function :test, version: 2, revert_to_version: 1
+        end
+      end
+
+      expect { run_migration(migration, :change) }.not_to raise_error
+    end
+  end
+end

data/spec/features/functions/revert_spec.rb

@@ -0,0 +1,75 @@
+require "spec_helper"
+
+describe "Reverting migrations", :db do
+  around do |example|
+    sql_definition = <<-EOS
+      CREATE OR REPLACE FUNCTION test()
+      RETURNS text AS $$
+      BEGIN
+          RETURN 'test';
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    with_function_definition(name: :test, sql_definition: sql_definition) do
+      example.run
+    end
+  end
+
+  it "can run reversible migrations for creating functions" do
+    migration = Class.new(migration_class) do
+      def change
+        create_function :test
+      end
+    end
+
+    expect { run_migration(migration, [:up, :down]) }.not_to raise_error
+  end
+
+  it "can run reversible migrations for dropping functions" do
+    connection.create_function(:test)
+
+    good_migration = Class.new(migration_class) do
+      def change
+        drop_function :test, revert_to_version: 1
+      end
+    end
+    bad_migration = Class.new(migration_class) do
+      def change
+        drop_function :test
+      end
+    end
+
+    expect { run_migration(good_migration, [:up, :down]) }.not_to raise_error
+    expect { run_migration(bad_migration, [:up, :down]) }.
+      to raise_error(
+        ActiveRecord::IrreversibleMigration,
+        /`create_function` is reversible only if given a `revert_to_version`/,
+      )
+  end
+
+  it "can run reversible migrations for updating functions" do
+    connection.create_function(:test)
+
+    sql_definition = <<-EOS
+      CREATE OR REPLACE FUNCTION test()
+      RETURNS text AS $$
+      BEGIN
+        RETURN 'bar';
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    with_function_definition(
+      name: :test,
+      version: 2,
+      sql_definition: sql_definition,
+    ) do
+      migration = Class.new(migration_class) do
+        def change
+          update_function :test, version: 2, revert_to_version: 1
+        end
+      end
+
+      expect { run_migration(migration, [:up, :down]) }.not_to raise_error
+    end
+  end
+end

data/spec/features/triggers/migrations_spec.rb

@@ -0,0 +1,56 @@
+require "spec_helper"
+
+describe "Trigger migrations", :db do
+  around do |example|
+    connection.execute <<-EOS
+      CREATE TABLE users (
+          id int PRIMARY KEY,
+          name varchar(256),
+          upper_name varchar(256)
+      );
+    EOS
+    Fx.database.create_function <<-EOS
+      CREATE OR REPLACE FUNCTION uppercase_users_name()
+      RETURNS trigger AS $$
+      BEGIN
+        NEW.upper_name = UPPER(NEW.name);
+        RETURN NEW;
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    sql_definition = <<-EOS
+      CREATE TRIGGER uppercase_users_name
+          BEFORE INSERT ON users
+          FOR EACH ROW
+          EXECUTE PROCEDURE uppercase_users_name();
+    EOS
+    with_trigger_definition(
+      name: :uppercase_users_name,
+      sql_definition: sql_definition,
+    ) do
+      example.run
+    end
+  end
+
+  it "can run migrations that create triggers" do
+    migration = Class.new(migration_class) do
+      def up
+        create_trigger :uppercase_users_name
+      end
+    end
+
+    expect { run_migration(migration, :up) }.not_to raise_error
+  end
+
+  it "can run migrations that drop triggers" do
+    connection.create_trigger(:uppercase_users_name)
+
+    migration = Class.new(migration_class) do
+      def up
+        drop_trigger :uppercase_users_name, on: :users
+      end
+    end
+
+    expect { run_migration(migration, :up) }.not_to raise_error
+  end
+end

data/spec/features/triggers/revert_spec.rb

@@ -0,0 +1,95 @@
+require "spec_helper"
+
+describe "Reverting migrations", :db do
+  around do |example|
+    connection.execute <<-EOS
+      CREATE TABLE users (
+          id int PRIMARY KEY,
+          name varchar(256),
+          upper_name varchar(256)
+      );
+    EOS
+    Fx.database.create_function <<-EOS
+      CREATE OR REPLACE FUNCTION uppercase_users_name()
+      RETURNS trigger AS $$
+      BEGIN
+        NEW.upper_name = UPPER(NEW.name);
+        RETURN NEW;
+      END;
+      $$ LANGUAGE plpgsql;
+    EOS
+    sql_definition = <<-EOS
+      CREATE TRIGGER uppercase_users_name
+          BEFORE INSERT ON users
+          FOR EACH ROW
+          EXECUTE PROCEDURE uppercase_users_name();
+    EOS
+    with_trigger_definition(
+      name: :uppercase_users_name,
+      sql_definition: sql_definition,
+    ) do
+      example.run
+    end
+  end
+
+  it "can run reversible migrations for creating triggers" do
+    migration = Class.new(migration_class) do
+      def change
+        create_trigger :uppercase_users_name, on: :users
+      end
+    end
+
+    expect { run_migration(migration, [:up, :down]) }.not_to raise_error
+  end
+
+  it "can run reversible migrations for dropping triggers" do
+    connection.create_trigger(:uppercase_users_name, on: :users)
+
+    good_migration = Class.new(migration_class) do
+      def change
+        drop_trigger :uppercase_users_name, on: :users, revert_to_version: 1
+      end
+    end
+    bad_migration = Class.new(migration_class) do
+      def change
+        drop_trigger :uppercase_users_name, on: :users
+      end
+    end
+
+    expect { run_migration(good_migration, [:up, :down]) }.not_to raise_error
+    expect { run_migration(bad_migration, [:up, :down]) }.
+      to raise_error(
+        ActiveRecord::IrreversibleMigration,
+        /`create_trigger` is reversible only if given a `revert_to_version`/,
+      )
+  end
+
+  it "can run reversible migrations for updating triggers" do
+    connection.create_trigger(:uppercase_users_name)
+
+    sql_definition = <<-EOS
+      CREATE TRIGGER uppercase_users_name
+          BEFORE UPDATE ON users
+          FOR EACH ROW
+          EXECUTE PROCEDURE uppercase_users_name();
+    EOS
+    with_trigger_definition(
+      name: :uppercase_users_name,
+      sql_definition: sql_definition,
+      version: 2,
+    ) do
+      migration = Class.new(migration_class) do
+        def change
+          update_trigger(
+            :uppercase_users_name,
+            on: :users,
+            version: 2,
+            revert_to_version: 1,
+          )
+        end
+      end
+
+      expect { run_migration(migration, [:up, :down]) }.not_to raise_error
+    end
+  end
+end

data/spec/fx/adapters/postgres/functions_spec.rb

@@ -0,0 +1,37 @@
+require "spec_helper"
+
+module Fx
+  module Adapters
+    describe Postgres::Functions, :db do
+      describe ".all" do
+        it "returns `Function` objects" do
+          connection = ActiveRecord::Base.connection
+          connection.execute <<-EOS.strip_heredoc
+            CREATE OR REPLACE FUNCTION test()
+            RETURNS text AS $$
+            BEGIN
+                RETURN 'test';
+            END;
+            $$ LANGUAGE plpgsql;
+          EOS
+
+          functions = Postgres::Functions.new(connection).all
+
+          first = functions.first
+          expect(functions.size).to eq 1
+          expect(first.name).to eq "test"
+          expect(first.definition).to eq <<-EOS.strip_heredoc
+            CREATE OR REPLACE FUNCTION public.test()
+             RETURNS text
+             LANGUAGE plpgsql
+            AS $function$
+            BEGIN
+                RETURN 'test';
+            END;
+            $function$
+          EOS
+        end
+      end
+    end
+  end
+end

data/spec/fx/adapters/postgres/triggers_spec.rb

@@ -0,0 +1,45 @@
+require "spec_helper"
+
+module Fx
+  module Adapters
+    describe Postgres::Triggers, :db do
+      describe ".all" do
+        it "returns `Trigger` objects" do
+          connection = ActiveRecord::Base.connection
+          connection.execute <<-EOS.strip_heredoc
+            CREATE TABLE users (
+                id int PRIMARY KEY,
+                name varchar(256),
+                upper_name varchar(256)
+            );
+          EOS
+          connection.execute <<-EOS.strip_heredoc
+            CREATE OR REPLACE FUNCTION uppercase_users_name()
+            RETURNS trigger AS $$
+            BEGIN
+              NEW.upper_name = UPPER(NEW.name);
+              RETURN NEW;
+            END;
+            $$ LANGUAGE plpgsql;
+          EOS
+          connection.execute <<-EOS.strip_heredoc
+            CREATE TRIGGER uppercase_users_name
+                BEFORE INSERT ON users
+                FOR EACH ROW
+                EXECUTE PROCEDURE uppercase_users_name();
+          EOS
+
+          triggers = Postgres::Triggers.new(connection).all
+
+          first = triggers.first
+          expect(triggers.size).to eq 1
+          expect(first.name).to eq "uppercase_users_name"
+          expect(first.definition).to include("BEFORE INSERT")
+          expect(first.definition).to match(/ON [public\.users|users]/)
+          expect(first.definition).to include("FOR EACH ROW")
+          expect(first.definition).to include("EXECUTE PROCEDURE uppercase_users_name()")
+        end
+      end
+    end
+  end
+end

data/spec/fx/adapters/postgres_spec.rb

@@ -0,0 +1,146 @@
+require "spec_helper"
+
+module Fx::Adapters
+  describe Postgres, :db do
+    describe "#create_function" do
+      it "successfully creates a function" do
+        adapter = Postgres.new
+        adapter.create_function(
+          <<-EOS
+            CREATE OR REPLACE FUNCTION test()
+            RETURNS text AS $$
+            BEGIN
+                RETURN 'test';
+            END;
+            $$ LANGUAGE plpgsql;
+          EOS
+        )
+
+        expect(adapter.functions.map(&:name)).to include("test")
+      end
+    end
+
+    describe "#create_trigger" do
+      it "successfully creates a trigger" do
+        connection.execute <<-EOS
+          CREATE TABLE users (
+              id int PRIMARY KEY,
+              name varchar(256),
+              upper_name varchar(256)
+          );
+        EOS
+        adapter = Postgres.new
+        adapter.create_function <<-EOS
+          CREATE OR REPLACE FUNCTION uppercase_users_name()
+          RETURNS trigger AS $$
+          BEGIN
+            NEW.upper_name = UPPER(NEW.name);
+            RETURN NEW;
+          END;
+          $$ LANGUAGE plpgsql;
+        EOS
+        adapter.create_trigger(
+          <<-EOS
+            CREATE TRIGGER uppercase_users_name
+                BEFORE INSERT ON users
+                FOR EACH ROW
+                EXECUTE PROCEDURE uppercase_users_name();
+          EOS
+        )
+
+        expect(adapter.triggers.map(&:name)).to include("uppercase_users_name")
+      end
+    end
+
+    describe "#drop_function" do
+      context "when the function has arguments" do
+        it "successfully drops a function with the entire function signature" do
+          adapter = Postgres.new
+          adapter.create_function(
+            <<-EOS
+              CREATE FUNCTION adder(x int, y int)
+              RETURNS int AS $$
+              BEGIN
+                  RETURN $1 + $2;
+              END;
+              $$ LANGUAGE plpgsql;
+            EOS
+          )
+
+          adapter.drop_function(:adder)
+
+          expect(adapter.functions.map(&:name)).not_to include("adder")
+        end
+      end
+
+      context "when the function does not have arguments" do
+        it "successfully drops a function" do
+          adapter = Postgres.new
+          adapter.create_function(
+            <<-EOS
+              CREATE OR REPLACE FUNCTION test()
+              RETURNS text AS $$
+              BEGIN
+                  RETURN 'test';
+              END;
+              $$ LANGUAGE plpgsql;
+            EOS
+          )
+
+          adapter.drop_function(:test)
+
+          expect(adapter.functions.map(&:name)).not_to include("test")
+        end
+      end
+    end
+
+    describe "#functions" do
+      it "finds functions and builds Fx::Function objects" do
+        adapter = Postgres.new
+        adapter.create_function(
+          <<-EOS
+            CREATE OR REPLACE FUNCTION test()
+            RETURNS text AS $$
+            BEGIN
+                RETURN 'test';
+            END;
+            $$ LANGUAGE plpgsql;
+          EOS
+        )
+
+        expect(adapter.functions.map(&:name)).to eq ["test"]
+      end
+    end
+
+    describe "#triggers" do
+      it "finds triggers and builds Fx::Trigger objects" do
+        connection.execute <<-EOS
+          CREATE TABLE users (
+              id int PRIMARY KEY,
+              name varchar(256),
+              upper_name varchar(256)
+          );
+        EOS
+        adapter = Postgres.new
+        adapter.create_function <<-EOS
+          CREATE OR REPLACE FUNCTION uppercase_users_name()
+          RETURNS trigger AS $$
+          BEGIN
+            NEW.upper_name = UPPER(NEW.name);
+            RETURN NEW;
+          END;
+          $$ LANGUAGE plpgsql;
+        EOS
+        sql_definition = <<-EOS
+          CREATE TRIGGER uppercase_users_name
+              BEFORE INSERT ON users
+              FOR EACH ROW
+              EXECUTE PROCEDURE uppercase_users_name()
+        EOS
+        adapter.create_trigger(sql_definition)
+
+        expect(adapter.triggers.map(&:name)).to eq ["uppercase_users_name"]
+      end
+    end
+  end
+end