array_table_spec.rb in tablesmith-0.5.0

- old
+ new

@@ -1,5 +1,7 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 
 describe 'Array Source' do
   it 'just works in a console' do
     expected = <<~TABLE
@@ -7,8 +9,65 @@
       | a | b | c |
       +---+---+---+
       | d | e | f |
       +---+---+---+
     TABLE
-    [%w(a b c), %w(d e f)].to_table.text_table.to_s.should == expected
+    [%w[a b c], %w[d e f]].to_table.to_s.should == expected
   end
-end
\ No newline at end of file
+
+  def do_just_works_test(_expected, meth_id)
+    orig_stdout = $stdout
+    sio = StringIO.new
+    $stdout = sio
+    send(meth_id, [%w[a b c], %w[d e f]].to_table)
+    sio.string
+  ensure
+    $stdout = orig_stdout
+  end
+
+  def just_works(meth_id)
+    expected = <<~TABLE
+      +---+---+---+
+      | a | b | c |
+      +---+---+---+
+      | d | e | f |
+      +---+---+---+
+    TABLE
+
+    actual = do_just_works_test(expected, meth_id)
+    actual.should == expected
+  end
+
+  it 'just works with print' do
+    just_works(:print)
+  end
+
+  it 'just works with puts' do
+    pending
+
+    # Kernel.puts has special behavior for puts with Array,
+    # which Table subclasses, so this isn't going to work,
+    # unless we can stop subclassing Array.
+    just_works(:puts)
+  end
+
+  it 'just works with inspect' do
+    expected = <<~TABLE
+      +---+---+---+
+      | a | b | c |
+      +---+---+---+
+      | d | e | f |
+      +---+---+---+
+    TABLE
+    [%w[a b c], %w[d e f]].to_table.inspect.should == expected
+  end
+
+  it 'just works with inspect two element integer array' do
+    expected = <<~TABLE
+      +---+
+      | 1 |
+      | 3 |
+      +---+
+    TABLE
+    [1, 3].to_table.inspect.should == expected
+  end
+end