fx-jets 0.6.3s

data/spec/fx/statements/trigger_spec.rb

@@ -0,0 +1,132 @@
+require "spec_helper"
+require "fx/statements/trigger"
+
+describe Fx::Statements::Trigger, :db do
+  describe "#create_trigger" do
+    it "creates a trigger from a file" do
+      database = stubbed_database
+      definition = stubbed_definition
+
+      connection.create_trigger(:test)
+
+      expect(database).to have_received(:create_trigger).
+        with(definition.to_sql)
+      expect(Fx::Definition).to have_received(:new).
+        with(name: :test, version: 1, type: "trigger")
+    end
+
+    it "allows creating a trigger with a specific version" do
+      database = stubbed_database
+      definition = stubbed_definition
+
+      connection.create_trigger(:test, version: 2)
+
+      expect(database).to have_received(:create_trigger).
+        with(definition.to_sql)
+      expect(Fx::Definition).to have_received(:new).
+        with(name: :test, version: 2, type: "trigger")
+    end
+
+    it "raises an error if both arguments are set" do
+      stubbed_database
+
+      expect {
+        connection.create_trigger(
+          :whatever,
+          version: 1,
+          sql_definition: "a definition",
+        )
+      }.to raise_error(
+        ArgumentError,
+        /cannot both be set/,
+      )
+    end
+  end
+
+  describe "#drop_trigger" do
+    it "drops the trigger" do
+      database = stubbed_database
+
+      connection.drop_trigger(:test, on: :users)
+
+      expect(database).to have_received(:drop_trigger).
+        with(:test, on: :users)
+    end
+  end
+
+  describe "#update_trigger" do
+    it "updates the trigger" do
+      database = stubbed_database
+      definition = stubbed_definition
+
+      connection.update_trigger(:test, on: :users, version: 3)
+
+      expect(database).to have_received(:update_trigger).with(
+        :test,
+        on: :users,
+        sql_definition: definition.to_sql,
+      )
+      expect(Fx::Definition).to have_received(:new).with(
+        name: :test,
+        version: 3,
+        type: "trigger",
+      )
+    end
+
+    it "updates a trigger from a text definition" do
+      database = stubbed_database
+
+      connection.update_trigger(
+        :test,
+        on: :users,
+        sql_definition: "a definition",
+      )
+
+      expect(database).to have_received(:update_trigger).with(
+        :test,
+        on: :users,
+        sql_definition: "a definition",
+      )
+    end
+
+    it "raises an error if not supplied a version" do
+      expect {
+        connection.update_trigger(
+          :whatever,
+          version: nil,
+          sql_definition: nil,
+        )
+      }.to raise_error(
+        ArgumentError,
+        /version or sql_definition must be specified/,
+      )
+    end
+
+    it "raises an error if both arguments are set" do
+      stubbed_database
+
+      expect {
+        connection.update_trigger(
+          :whatever,
+          version: 1,
+          sql_definition: "a definition",
+        )
+      }.to raise_error(
+        ArgumentError,
+        /cannot both be set/,
+      )
+    end
+  end
+
+  def stubbed_database
+    instance_spy("StubbedDatabase").tap do |stubbed_database|
+      allow(Fx).to receive(:database).and_return(stubbed_database)
+    end
+  end
+
+  def stubbed_definition
+    instance_double("Fx::Definition", to_sql: nil).tap do |stubbed_definition|
+      allow(Fx::Definition).to receive(:new).and_return(stubbed_definition)
+    end
+  end
+end

data/spec/fx/trigger_spec.rb

@@ -0,0 +1,55 @@
+require "spec_helper"
+require "fx/trigger"
+
+module Fx
+  describe Trigger do
+    describe "#<=>" do
+      it "delegates to `name`" do
+        trigger_a = Trigger.new(
+          "name" => "name_a",
+          "definition" => "some defintion",
+        )
+        trigger_b = Trigger.new(
+          "name" => "name_b",
+          "definition" => "some defintion",
+        )
+        trigger_c = Trigger.new(
+          "name" => "name_c",
+          "definition" => "some defintion",
+        )
+
+        expect(trigger_b).to be_between(trigger_a, trigger_c)
+      end
+    end
+
+    describe "#==" do
+      it "compares `name` and `definition`" do
+        trigger_a = Trigger.new(
+          "name" => "name_a",
+          "definition" => "some defintion",
+        )
+        trigger_b = Trigger.new(
+          "name" => "name_b",
+          "definition" => "some other defintion",
+        )
+
+        expect(trigger_a).not_to eq(trigger_b)
+      end
+    end
+
+    describe "#to_schema" do
+      it "returns a schema compatible version of the trigger" do
+        trigger = Trigger.new(
+          "name" => "uppercase_users_name",
+          "definition" => "CREATE TRIGGER uppercase_users_name ...",
+        )
+
+        expect(trigger.to_schema).to eq <<-EOS
+  create_trigger :uppercase_users_name, sql_definition: <<-\SQL
+      CREATE TRIGGER uppercase_users_name ...
+  SQL
+        EOS
+      end
+    end
+  end
+end

data/spec/generators/fx/function/function_generator_spec.rb

@@ -0,0 +1,46 @@
+require "spec_helper"
+require "generators/fx/function/function_generator"
+
+describe Fx::Generators::FunctionGenerator, :generator do
+  it "creates a function definition file, and a migration" do
+    migration = file("db/migrate/create_function_test.rb")
+    function_definition = file("db/functions/test_v01.sql")
+
+    run_generator ["test"]
+
+    expect(function_definition).to exist
+    expect(migration).to be_a_migration
+    expect(migration_file(migration)).to contain "CreateFunctionTest"
+  end
+
+  context "when passed --no-migration" do
+    it "creates a only function definition file" do
+      migration = file("db/migrate/create_function_test.rb")
+      function_definition = file("db/functions/test_v01.sql")
+
+      run_generator ["test", "--no-migration"]
+
+      expect(function_definition).to exist
+      expect(migration_file(migration)).not_to exist
+    end
+  end
+
+  it "updates an existing function" do
+    with_function_definition(
+      name: "test",
+      version: 1,
+      sql_definition: "hello",
+    ) do
+      allow(Dir).to receive(:entries).and_return(["test_v01.sql"])
+      migration = file("db/migrate/update_function_test_to_version_2.rb")
+      function_definition = file("db/functions/test_v02.sql")
+
+      run_generator ["test"]
+
+      expect(function_definition).to exist
+      expect(migration).to be_a_migration
+      expect(migration_file(migration)).
+        to contain("UpdateFunctionTestToVersion2")
+    end
+  end
+end

data/spec/generators/fx/trigger/trigger_generator_spec.rb

@@ -0,0 +1,59 @@
+require "spec_helper"
+require "generators/fx/trigger/trigger_generator"
+
+describe Fx::Generators::TriggerGenerator, :generator do
+  it "creates a trigger definition file, and a migration" do
+    migration = file("db/migrate/create_trigger_test.rb")
+    trigger_definition = file("db/triggers/test_v01.sql")
+
+    run_generator ["test", "table_name" => "some_table"]
+
+    expect(trigger_definition).to exist
+    expect(migration).to be_a_migration
+    expect(migration_file(migration)).to contain "CreateTriggerTest"
+    expect(migration_file(migration)).to contain "on: :some_table"
+  end
+
+  context "when passed --no-migration" do
+    it "creates a only trigger definition file" do
+      migration = file("db/migrate/create_trigger_test.rb")
+      trigger_definition = file("db/triggers/test_v01.sql")
+
+      run_generator ["test", {"table_name" => "some_table"}, "--no-migration"]
+
+      expect(trigger_definition).to exist
+      expect(migration_file(migration)).not_to exist
+    end
+  end
+
+  it "supports naming the table as `on` aswell as `table_name`" do
+    migration = file("db/migrate/create_trigger_test.rb")
+    trigger_definition = file("db/triggers/test_v01.sql")
+
+    run_generator ["test", "on" => "some_table"]
+
+    expect(trigger_definition).to exist
+    expect(migration).to be_a_migration
+    expect(migration_file(migration)).to contain "CreateTriggerTest"
+    expect(migration_file(migration)).to contain "on: :some_table"
+  end
+
+  it "requires `table_name` or `on` to be specified" do
+    expect {
+      run_generator ["test", "foo" => "some_table"]
+    }.to raise_error ArgumentError
+  end
+
+  it "updates an existing trigger" do
+    allow(Dir).to receive(:entries).and_return(["test_v01.sql"])
+    migration = file("db/migrate/update_trigger_test_to_version_2.rb")
+    trigger_definition = file("db/triggers/test_v02.sql")
+
+    run_generator ["test", "table_name" => "some_table"]
+
+    expect(trigger_definition).to exist
+    expect(migration).to be_a_migration
+    expect(migration_file(migration)).to contain "UpdateTriggerTestToVersion2"
+    expect(migration_file(migration)).to contain "on: :some_table"
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+ENV["RAILS_ENV"] = "test"
+require "database_cleaner"
+
+require File.expand_path("../dummy/config/environment", __FILE__)
+Dir["spec/support/**/*.rb"].sort.each { |file| load file }
+
+RSpec.configure do |config|
+  config.order = "random"
+  DatabaseCleaner.strategy = :transaction
+
+  config.around(:each, db: true) do |example|
+    DatabaseCleaner.start
+    example.run
+    DatabaseCleaner.clean
+  end
+
+  unless defined?(silence_stream)
+    require "active_support/testing/stream"
+    config.include ActiveSupport::Testing::Stream
+  end
+end

data/spec/support/definition_helpers.rb

@@ -0,0 +1,37 @@
+module DefinitionHelpers
+  def with_function_definition(name:, sql_definition:, version: 1, &block)
+    definition = Fx::Definition.new(name: name, version: version)
+
+    with_definition(
+      definition: definition,
+      sql_definition: sql_definition,
+      block: block,
+    )
+  end
+
+  def with_trigger_definition(name:, sql_definition:, version: 1, &block)
+    definition = Fx::Definition.new(
+      name: name,
+      version: version,
+      type: "trigger",
+    )
+
+    with_definition(
+      definition: definition,
+      sql_definition: sql_definition,
+      block: block,
+    )
+  end
+
+  def with_definition(definition:, sql_definition:, block:)
+    FileUtils.mkdir_p(File.dirname(definition.full_path))
+    File.open(definition.full_path, "w") { |f| f.write(sql_definition) }
+    block.call
+  ensure
+    File.delete definition.full_path
+  end
+end
+
+RSpec.configure do |config|
+  config.include DefinitionHelpers
+end

data/spec/support/generator_setup.rb

@@ -0,0 +1,11 @@
+require "ammeter/init"
+
+RSpec.configure do |config|
+  config.before(:each, :generator) do
+    fake_rails_root = File.expand_path("../../../tmp", __FILE__)
+    allow(Rails).to receive(:root).and_return(Pathname.new(fake_rails_root))
+
+    destination fake_rails_root
+    prepare_destination
+  end
+end

data/spec/support/migration_helpers.rb

@@ -0,0 +1,25 @@
+module MigrationsHelper
+  def run_migration(migration, directions)
+    silence_stream(STDOUT) do
+      Array.wrap(directions).each do |direction|
+        migration.migrate(direction)
+      end
+    end
+  end
+
+  def migration_class
+    if Rails::VERSION::MAJOR >= 5
+      ::ActiveRecord::Migration[5.0]
+    else
+      ::ActiveRecord::Migration
+    end
+  end
+
+  def connection
+    @_connection ||= ActiveRecord::Base.connection
+  end
+end
+
+RSpec.configure do |config|
+  config.include MigrationsHelper
+end