brew/Library/Homebrew/test/patching_spec.rb

require "formula"

describe "patching" do
  TESTBALL_URL = "file://#{TEST_FIXTURE_DIR}/tarballs/testball-0.1.tbz".freeze
  TESTBALL_PATCHES_URL = "file://#{TEST_FIXTURE_DIR}/tarballs/testball-0.1-patches.tgz".freeze
  PATCH_URL_A = "file://#{TEST_FIXTURE_DIR}/patches/noop-a.diff".freeze
  PATCH_URL_B = "file://#{TEST_FIXTURE_DIR}/patches/noop-b.diff".freeze
  PATCH_A_CONTENTS = File.read "#{TEST_FIXTURE_DIR}/patches/noop-a.diff"
  PATCH_B_CONTENTS = File.read "#{TEST_FIXTURE_DIR}/patches/noop-b.diff"
  APPLY_A = "noop-a.diff".freeze
  APPLY_B = "noop-b.diff".freeze
  APPLY_C = "noop-c.diff".freeze

  def formula(name = "formula_name", path: Formulary.core_path(name), spec: :stable, alias_path: nil, &block)
    Class.new(Formula) {
      url TESTBALL_URL
      sha256 TESTBALL_SHA256
      class_eval(&block)
    }.new(name, path, spec, alias_path: alias_path)
  end

  matcher :be_patched do
    match do |formula|
      formula.brew do
        formula.patch
        s = File.read("libexec/NOOP")
        expect(s).not_to include("NOOP"), "libexec/NOOP was not patched as expected"
        expect(s).to include("ABCD"), "libexec/NOOP was not patched as expected"
      end
    end
  end

  matcher :be_sequentially_patched do
    match do |formula|
      formula.brew do
        formula.patch
        s = File.read("libexec/NOOP")
        expect(s).not_to include("NOOP"), "libexec/NOOP was not patched as expected"
        expect(s).not_to include("ABCD"), "libexec/NOOP was not patched as expected"
        expect(s).to include("1234"), "libexec/NOOP was not patched as expected"
      end
    end
  end

  matcher :miss_apply do
    match do |formula|
      expect {
        formula.brew do
          formula.patch
        end
      }.to raise_error(MissingApplyError)
    end
  end

  specify "single_patch" do
    expect(
      formula do
        def patches
          PATCH_URL_A
        end
      end,
    ).to be_patched
  end

  specify "single_patch_dsl" do
    expect(
      formula do
        patch do
          url PATCH_URL_A
          sha256 PATCH_A_SHA256
        end
      end,
    ).to be_patched
  end

  specify "single_patch_dsl_with_apply" do
    expect(
      formula do
        patch do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply APPLY_A
        end
      end,
    ).to be_patched
  end

  specify "single_patch_dsl_with_sequential_apply" do
    expect(
      formula do
        patch do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply APPLY_A, APPLY_C
        end
      end,
    ).to be_sequentially_patched
  end

  specify "single_patch_dsl_with_strip" do
    expect(
      formula do
        patch :p1 do
          url PATCH_URL_A
          sha256 PATCH_A_SHA256
        end
      end,
    ).to be_patched
  end

  specify "single_patch_dsl_with_strip_with_apply" do
    expect(
      formula do
        patch :p1 do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply APPLY_A
        end
      end,
    ).to be_patched
  end

  specify "single_patch_dsl_with_incorrect_strip" do
    expect {
      f = formula do
        patch :p0 do
          url PATCH_URL_A
          sha256 PATCH_A_SHA256
        end
      end

      f.brew { |formula, _staging| formula.patch }
    }.to raise_error(ErrorDuringExecution)
  end

  specify "single_patch_dsl_with_incorrect_strip_with_apply" do
    expect {
      f = formula do
        patch :p0 do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply APPLY_A
        end
      end

      f.brew { |formula, _staging| formula.patch }
    }.to raise_error(ErrorDuringExecution)
  end

  specify "patch_p0_dsl" do
    expect(
      formula do
        patch :p0 do
          url PATCH_URL_B
          sha256 PATCH_B_SHA256
        end
      end,
    ).to be_patched
  end

  specify "patch_p0_dsl_with_apply" do
    expect(
      formula do
        patch :p0 do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply APPLY_B
        end
      end,
    ).to be_patched
  end

  specify "patch_p0" do
    expect(
      formula do
        def patches
          { p0: PATCH_URL_B }
        end
      end,
    ).to be_patched
  end

  specify "patch_array" do
    expect(
      formula do
        def patches
          [PATCH_URL_A]
        end
      end,
    ).to be_patched
  end

  specify "patch_hash" do
    expect(
      formula do
        def patches
          { p1: PATCH_URL_A }
        end
      end,
    ).to be_patched
  end

  specify "patch_hash_array" do
    expect(
      formula do
        def patches
          { p1: [PATCH_URL_A] }
        end
      end,
    ).to be_patched
  end

  specify "patch_string" do
    expect(formula { patch PATCH_A_CONTENTS }).to be_patched
  end

  specify "patch_string_with_strip" do
    expect(formula { patch :p0, PATCH_B_CONTENTS }).to be_patched
  end

  specify "patch_data_constant" do
    expect(
      formula("test", path: Pathname.new(__FILE__).expand_path) do
        def patches
          :DATA
        end
      end,
    ).to be_patched
  end

  specify "single_patch_missing_apply_fail" do
    expect(
      formula do
        def patches
          TESTBALL_PATCHES_URL
        end
      end,
    ).to miss_apply
  end

  specify "single_patch_dsl_missing_apply_fail" do
    expect(
      formula do
        patch do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
        end
      end,
    ).to miss_apply
  end

  specify "single_patch_dsl_with_apply_enoent_fail" do
    expect {
      f = formula do
        patch do
          url TESTBALL_PATCHES_URL
          sha256 TESTBALL_PATCHES_SHA256
          apply "patches/#{APPLY_A}"
        end
      end

      f.brew { |formula, _staging| formula.patch }
    }.to raise_error(ErrorDuringExecution)
  end
end

__END__
diff --git a/libexec/NOOP b/libexec/NOOP
index bfdda4c..e08d8f4 100755
--- a/libexec/NOOP
+++ b/libexec/NOOP
@@ -1,2 +1,2 @@
 #!/bin/bash
-echo NOOP
\ No newline at end of file
+echo ABCD
\ No newline at end of file