require "spec_helper"

RSpec.describe Fx::Definition do
  describe "#to_sql" do
    context "representing a function definition" do
      it "returns the content of a function definition" do
        sql_definition = <<-EOS
          CREATE OR REPLACE FUNCTION test()
          RETURNS text AS $$
          BEGIN
              RETURN 'test';
          END;
          $$ LANGUAGE plpgsql;
        EOS
        allow(File).to receive(:read).and_return(sql_definition)

        definition = Fx::Definition.function(name: "test", version: 1)

        expect(definition.to_sql).to eq(sql_definition)
      end

      it "raises an error if the file is empty" do
        allow(File).to receive(:read).and_return("")
        definition = Fx::Definition.function(name: "test", version: 1)

        expect do
          definition.to_sql
        end.to raise_error(
          RuntimeError,
          %r{Define function in db/functions/test_v01.sql before migrating}
        )
      end

      context "when definition is at Rails engine" do
        it "returns the content of a function definition" do
          sql_definition = <<~EOS
            CREATE OR REPLACE FUNCTION test()
            RETURNS text AS $$
            BEGIN
                RETURN 'test';
            END;
            $$ LANGUAGE plpgsql;
          EOS
          engine_path = Rails.root.join("tmp", "engine")
          FileUtils.mkdir_p(engine_path.join("db", "functions"))
          File.write(engine_path.join("db", "functions", "custom_test_v01.sql"), sql_definition)
          Rails.application.config.paths["db/migrate"].push(engine_path.join("db", "migrate"))

          definition = Fx::Definition.function(name: "custom_test", version: 1)

          expect(definition.to_sql).to eq(sql_definition)

          FileUtils.rm_rf(engine_path)
        end
      end
    end

    context "representing a trigger definition" do
      it "returns the content of a trigger definition" do
        sql_definition = <<~EOS
          CREATE TRIGGER check_update
          BEFORE UPDATE ON accounts
          FOR EACH ROW
          EXECUTE FUNCTION check_account_update();
        EOS
        allow(File).to receive(:read).and_return(sql_definition)

        definition = Fx::Definition.trigger(name: "test", version: 1)

        expect(definition.to_sql).to eq(sql_definition)
      end

      it "raises an error if the file is empty" do
        allow(File).to receive(:read).and_return("")
        definition = Fx::Definition.trigger(name: "test", version: 1)

        expect do
          definition.to_sql
        end.to raise_error(
          RuntimeError,
          %r{Define trigger in db/triggers/test_v01.sql before migrating}
        )
      end
    end
  end

  describe "#path" do
    context "representing a function definition" do
      it "returns a sql file with padded version and function name" do
        definition = Fx::Definition.function(name: "test", version: 1)

        expect(definition.path).to eq("db/functions/test_v01.sql")
      end
    end

    context "representing a trigger definition" do
      it "returns a sql file with padded version and trigger name" do
        definition = Fx::Definition.trigger(name: "test", version: 1)

        expect(definition.path).to eq("db/triggers/test_v01.sql")
      end
    end
  end

  describe "#full_path" do
    it "joins the path with Rails.root" do
      definition = Fx::Definition.function(name: "test", version: 15)

      expect(definition.full_path).to eq(Rails.root.join(definition.path))
    end
  end

  describe "#version" do
    it "pads the version number with 0" do
      definition = Fx::Definition.function(name: :_, version: 1)

      expect(definition.version).to eq("01")
    end

    it "does not pad more than 2 characters" do
      definition = Fx::Definition.function(name: :_, version: 15)

      expect(definition.version).to eq("15")
    end
  end
end