Add unit tests for Pry::Output

1 files changed, 197 insertions, 0 deletions

diff --git a/spec/output_spec.rb b/spec/output_spec.rb
new file mode 100644
index 00000000..40ccddc9
--- /dev/null
+++ b/spec/output_spec.rb
@@ -0,0 +1,197 @@
+# frozen_string_literal: true
+
+RSpec.describe Pry::Output do
+  let(:output) { StringIO.new }
+  let(:pry_instance) { Pry.new(output: output) }
+
+  subject { described_class.new(pry_instance) }
+
+  describe "#puts" do
+    it "returns nil" do
+      expect(subject.puts).to be_nil
+    end
+
+    context "when given an empty array" do
+      it "prints a newline" do
+        subject.puts([])
+        expect(output.string).to eq("\n")
+      end
+    end
+
+    context "when given multiple empty arrays" do
+      it "prints multiple newline" do
+        subject.puts([], [], [])
+        expect(output.string).to eq("\n\n\n")
+      end
+    end
+
+    context "when given convertible to array objects" do
+      let(:obj) do
+        Object.new.tap { |o| o.define_singleton_method(:to_ary) { [1] } }
+      end
+
+      let(:other_obj) do
+        Object.new.tap { |o| o.define_singleton_method(:to_ary) { [2] } }
+      end
+
+      it "prints the converted objects" do
+        subject.puts([obj, other_obj])
+        expect(output.string).to eq("1\n2\n")
+      end
+    end
+
+    context "when given non-convertible to array objects" do
+      let(:obj) { Object.new }
+      let(:other_obj) { Object.new }
+
+      it "prints the non-converted objects in its string form" do
+        subject.puts([obj, other_obj])
+        expect(output.string).to match(/\A#<Object:.+>\n#<Object:.+>\n\z/)
+      end
+
+      context "and when the object's #to_s has a newline" do
+        let(:obj) do
+          Object.new.tap { |o| o.define_singleton_method(:to_s) { "abc\n" } }
+        end
+
+        it "doesn't print a double newline" do
+          subject.puts(obj)
+          expect(output.string).to eq("abc\n")
+        end
+      end
+    end
+
+    context "when the given pry instance has 'color' enabled" do
+      let(:pry_instance) { Pry.new(output: output, color: true) }
+
+      it "doesn't decolorize output" do
+        subject.puts("\e[30mhi\e[0m")
+        expect(output.string).to eq("\e[30mhi\e[0m\n")
+      end
+    end
+
+    context "when the given pry instance has 'color' disabled" do
+      let(:pry_instance) { Pry.new(output: output, color: false) }
+
+      it "decolorizes output" do
+        subject.puts("\e[30mhi\e[0m")
+        expect(output.string).to eq("hi\n")
+      end
+    end
+  end
+
+  describe "#print" do
+    it "returns nil" do
+      expect(subject.print).to be_nil
+    end
+
+    context "when the given pry instance has 'color' enabled" do
+      let(:pry_instance) { Pry.new(output: output, color: true) }
+
+      it "doesn't decolorize output" do
+        subject.print("\e[30mhi\e[0m")
+        expect(output.string).to eq("\e[30mhi\e[0m")
+      end
+    end
+
+    context "when the given pry instance has 'color' disabled" do
+      let(:pry_instance) { Pry.new(output: output, color: false) }
+
+      it "decolorizes output" do
+        subject.print("\e[30mhi\e[0m")
+        expect(output.string).to eq('hi')
+      end
+    end
+  end
+
+  describe "#<<" do
+    specify { expect(subject.method(:<<)).to eq(subject.method(:print)) }
+  end
+
+  describe "#write" do
+    specify { expect(subject.method(:write)).to eq(subject.method(:print)) }
+  end
+
+  describe "#tty?" do
+    context "when the output responds to #tty? and is a TTY" do
+      before { expect(output).to receive(:tty?).and_return(true) }
+
+      it "returns true" do
+        expect(subject).to be_tty
+      end
+    end
+
+    context "when the output responds to #tty? and is not a TTY" do
+      before do
+        expect(output).to receive(:respond_to?).with(:tty?).and_return(true)
+        expect(output).to receive(:tty?).and_return(false)
+      end
+
+      it "returns false" do
+        expect(subject).not_to be_tty
+      end
+    end
+
+    context "when the output doesn't respond to #tty?" do
+      before do
+        expect(output).to receive(:respond_to?).with(:tty?).and_return(false)
+      end
+
+      it "returns false" do
+        expect(subject).not_to be_tty
+      end
+    end
+  end
+
+  describe "#method_missing" do
+    context "when the output responds to the given method name" do
+      it "forwards the method to the output" do
+        expect(output).to receive(:abcd)
+        subject.abcd
+      end
+    end
+
+    context "when the output doesn't respond to the given method name" do
+      it "raises NoMethodError" do
+        expect { subject.abcd }.to raise_error(NoMethodError)
+      end
+    end
+  end
+
+  describe "#respond_to_missing?" do
+    context "when the output responds to the given method name" do
+      before { output.define_singleton_method(:test_method) {} }
+
+      it "finds the method that is not defined on self" do
+        expect(subject).to respond_to(:test_method)
+        expect(subject.method(:test_method)).to be_a(Method)
+      end
+    end
+
+    context "when the output doesn't respond to the given method name" do
+      it "doesn't find the method" do
+        expect(subject).not_to respond_to(:test_method)
+        expect { subject.method(:test_method) }.to raise_error(NameError)
+      end
+    end
+  end
+
+  describe "#decolorize_maybe" do
+    context "when the given pry instance has 'color' enabled" do
+      let(:pry_instance) { Pry.new(output: output, color: true) }
+
+      it "returns the given string without modifications" do
+        str = "\e[30mhi\e[0m"
+        expect(subject.decolorize_maybe(str)).to eql(str)
+      end
+    end
+
+    context "when the given pry instance has 'color' disabled" do
+      let(:pry_instance) { Pry.new(output: output, color: false) }
+
+      it "returns decolorized string" do
+        expect(subject.decolorize_maybe("\e[30mhi\e[0m")).to eq('hi')
+      end
+    end
+  end
+end