From 3955e7e13b980ad4f8d38ffc3c0253d86204902e Mon Sep 17 00:00:00 2001
From: Bo Anderson <mail@boanderson.me>
Date: Fri, 27 Mar 2020 19:18:05 +0000
Subject: [PATCH] language/test: add shebang tests

---
 .../test/language/perl/shebang_spec.rb        | 52 +++++++++++++++++++
 Library/Homebrew/test/language/python_spec.rb | 43 +++++++++++++++
 2 files changed, 95 insertions(+)
 create mode 100644 Library/Homebrew/test/language/perl/shebang_spec.rb

diff --git a/Library/Homebrew/test/language/perl/shebang_spec.rb b/Library/Homebrew/test/language/perl/shebang_spec.rb
new file mode 100644
index 0000000000..5a59f162b7
--- /dev/null
+++ b/Library/Homebrew/test/language/perl/shebang_spec.rb
@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require "language/perl"
+require "utils/shebang"
+
+describe Language::Perl::Shebang do
+  let(:file) { Tempfile.new("perl-shebang") }
+  let(:perl_f) do
+    formula "perl" do
+      url "https://brew.sh/perl-1.0.tgz"
+    end
+  end
+  let(:f) do
+    formula "foo" do
+      url "https://brew.sh/foo-1.0.tgz"
+
+      uses_from_macos "perl"
+    end
+  end
+
+  before do
+    file.write <<~EOS
+      #!/usr/bin/env perl
+      a
+      b
+      c
+    EOS
+    file.flush
+  end
+
+  after { file.unlink }
+
+  describe "#detected_perl_shebang" do
+    it "can be used to replace Perl shebangs" do
+      allow(Formulary).to receive(:factory).with(perl_f.name).and_return(perl_f)
+      Utils::Shebang.rewrite_shebang described_class.detected_perl_shebang(f), file
+
+      expected_shebang = if OS.mac?
+        "/usr/bin/perl"
+      else
+        HOMEBREW_PREFIX/"opt/perl/bin/perl"
+      end
+
+      expect(File.read(file)).to eq <<~EOS
+        #!#{expected_shebang}
+        a
+        b
+        c
+      EOS
+    end
+  end
+end
diff --git a/Library/Homebrew/test/language/python_spec.rb b/Library/Homebrew/test/language/python_spec.rb
index a474e04da6..174373d999 100644
--- a/Library/Homebrew/test/language/python_spec.rb
+++ b/Library/Homebrew/test/language/python_spec.rb
@@ -2,6 +2,7 @@
 
 require "language/python"
 require "resource"
+require "utils/shebang"
 
 describe Language::Python, :needs_python do
   describe "#major_minor_version" do
@@ -32,6 +33,48 @@ describe Language::Python, :needs_python do
   end
 end
 
+describe Language::Python::Shebang do
+  let(:file) { Tempfile.new("python-shebang") }
+  let(:python_f) do
+    formula "python" do
+      url "https://brew.sh/python-1.0.tgz"
+    end
+  end
+  let(:f) do
+    formula "foo" do
+      url "https://brew.sh/foo-1.0.tgz"
+
+      depends_on "python"
+    end
+  end
+
+  before do
+    file.write <<~EOS
+      #!/usr/bin/env python3
+      a
+      b
+      c
+    EOS
+    file.flush
+  end
+
+  after { file.unlink }
+
+  describe "#detected_python_shebang" do
+    it "can be used to replace Python shebangs" do
+      expect(Formulary).to receive(:factory).with(python_f.name).and_return(python_f)
+      Utils::Shebang.rewrite_shebang described_class.detected_python_shebang(f), file
+
+      expect(File.read(file)).to eq <<~EOS
+        #!#{HOMEBREW_PREFIX}/opt/python/bin/python3
+        a
+        b
+        c
+      EOS
+    end
+  end
+end
+
 describe Language::Python::Virtualenv::Virtualenv do
   subject { described_class.new(formula, dir, "python") }