brew/Library/Homebrew/test/utils/shell_spec.rb

require "utils/shell"

describe Utils::Shell do
  describe "::profile" do
    it "returns ~/.bash_profile by default" do
      ENV["SHELL"] = "/bin/another_shell"
      expect(subject.profile).to eq("~/.bash_profile")
    end

    it "returns ~/.bash_profile for Bash" do
      ENV["SHELL"] = "/bin/bash"
      expect(subject.profile).to eq("~/.bash_profile")
    end

    it "returns ~/.zshrc for Zsh" do
      ENV["SHELL"] = "/bin/zsh"
      expect(subject.profile).to eq("~/.zshrc")
    end

    it "returns ~/.kshrc for Ksh" do
      ENV["SHELL"] = "/bin/ksh"
      expect(subject.profile).to eq("~/.kshrc")
    end
  end

  describe "::from_path" do
    it "supports a raw command name" do
      expect(subject.from_path("bash")).to eq(:bash)
    end

    it "supports full paths" do
      expect(subject.from_path("/bin/bash")).to eq(:bash)
    end

    it "supports versions" do
      expect(subject.from_path("zsh-5.2")).to eq(:zsh)
    end

    it "strips newlines" do
      expect(subject.from_path("zsh-5.2\n")).to eq(:zsh)
    end

    it "returns nil when input is invalid" do
      expect(subject.from_path("")).to be nil
      expect(subject.from_path("@@@@@@")).to be nil
      expect(subject.from_path("invalid_shell-4.2")).to be nil
    end
  end

  specify "::sh_quote" do
    expect(subject.send(:sh_quote, "")).to eq("''")
    expect(subject.send(:sh_quote, "\\")).to eq("\\\\")
    expect(subject.send(:sh_quote, "\n")).to eq("'\n'")
    expect(subject.send(:sh_quote, "$")).to eq("\\$")
    expect(subject.send(:sh_quote, "word")).to eq("word")
  end

  specify "::csh_quote" do
    expect(subject.send(:csh_quote, "")).to eq("''")
    expect(subject.send(:csh_quote, "\\")).to eq("\\\\")
    # note this test is different than for sh
    expect(subject.send(:csh_quote, "\n")).to eq("'\\\n'")
    expect(subject.send(:csh_quote, "$")).to eq("\\$")
    expect(subject.send(:csh_quote, "word")).to eq("word")
  end

  describe "::prepend_path_in_profile" do
    let(:path) { "/my/path" }

    it "supports Tcsh" do
      ENV["SHELL"] = "/bin/tcsh"
      expect(subject.prepend_path_in_profile(path))
        .to eq("echo 'setenv PATH #{path}:$PATH' >> #{shell_profile}")
    end

    it "supports Bash" do
      ENV["SHELL"] = "/bin/bash"
      expect(subject.prepend_path_in_profile(path))
        .to eq("echo 'export PATH=\"#{path}:$PATH\"' >> #{shell_profile}")
    end

    it "supports Fish" do
      ENV["SHELL"] = "/usr/local/bin/fish"
      ENV["fish_user_paths"] = "/some/path"
      expect(subject.prepend_path_in_profile(path))
        .to eq("echo 'set -g fish_user_paths \"#{path}\" $fish_user_paths' >> #{shell_profile}")
    end
  end
end