oj 2.0.0 → 3.0.0

data/test/test_writer.rb

@@ -0,0 +1,308 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+$: << File.dirname(__FILE__)
+
+require 'helper'
+
+class OjWriter < Minitest::Test
+
+  def setup
+    @default_options = Oj.default_options
+  end
+
+  def teardown
+    Oj.default_options = @default_options
+  end
+
+  def test_string_writer_empty_array
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_array()
+    w.pop()
+    assert_equal("[]\n", w.to_s)
+  end
+
+  def test_string_writer_nested_array
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_array()
+    w.push_array()
+    w.pop()
+    w.push_array()
+    w.push_array()
+    w.pop()
+    w.pop()
+    w.push_array()
+    w.pop()
+    w.pop()
+    assert_equal("[[],[[]],[]]\n", w.to_s)
+  end
+
+  def test_string_writer_empty_object
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object()
+    w.pop()
+    assert_equal("{}\n", w.to_s)
+  end
+
+  def test_string_writer_push_null_value_with_key
+    w = Oj::StringWriter.new(:indent => 0, :mode => :compat)
+    w.push_object()
+    w.push_value(nil, 'nothing')
+    w.pop()
+    assert_equal(%|{"nothing":null}\n|, w.to_s)
+  end
+
+  def test_string_writer_nested_object
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object()
+    w.push_object("a1")
+    w.pop()
+    w.push_object("a2")
+    w.push_object("b")
+    w.pop()
+    w.pop()
+    w.push_object("a3")
+    w.pop()
+    w.pop()
+    assert_equal(%|{"a1":{},"a2":{"b":{}},"a3":{}}\n|, w.to_s)
+  end
+
+  def test_string_writer_nested_key
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object()
+    w.push_key('a1')
+    w.push_object()
+    w.pop()
+    w.push_key('a2')
+    w.push_object('x')
+    w.push_object('b')
+    w.pop()
+    w.pop()
+    w.push_key('a3')
+    w.push_object()
+    w.push_key('a4')
+    w.push_value(37)
+    w.pop()
+    w.pop()
+    assert_equal(%|{"a1":{},"a2":{"b":{}},"a3":{"a4":37}}\n|, w.to_s)
+  end
+
+  def test_string_writer_value_array
+    w = Oj::StringWriter.new(:indent => 2)
+    w.push_array()
+    w.push_value(7)
+    w.push_value(7.3)
+    w.push_value(true)
+    w.push_value(nil)
+    w.push_value("a string")
+    w.push_value({'a' => 65})
+    w.push_value([1,2])
+    w.pop()
+    assert_equal(%|[
+  7,
+  7.3,
+  true,
+  null,
+  "a string",
+  {
+    "a":65
+  },
+  [
+    1,
+    2
+  ]
+]
+|, w.to_s)
+  end
+
+  def test_string_writer_block
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object() {
+      w.push_object("a1") {
+        w.push_value(7, 'a7')
+      }
+      w.push_array("a2") {
+        w.push_value('x')
+        w.push_value(3)
+      }
+    }
+    assert_equal(%|{"a1":{"a7":7},"a2":["x",3]}\n|, w.to_s)
+  end
+
+  def test_string_writer_json
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_array()
+    w.push_json('7')
+    w.push_json('true')
+    w.push_json(%|"a string"|)
+    w.push_object()
+    w.push_json('{"a":65}', 'x')
+    w.pop()
+    w.pop()
+    assert_equal(%|[7,true,"a string",{"x":{"a":65}}]\n|, w.to_s)
+  end
+
+  def test_string_writer_pop_excess
+    w = Oj::StringWriter.new(:indent => 0)
+    begin
+      w.push_object() {
+        w.push_key('key')
+      }
+    rescue Exception
+      assert(true)
+      return
+    end
+    assert(false, "*** expected an exception")
+  end
+
+  def test_string_writer_array_key
+    w = Oj::StringWriter.new(:indent => 0)
+    begin
+      w.push_array() {
+        w.push_key('key')
+        w.push_value(7)
+      }
+    rescue Exception
+      assert(true)
+      return
+    end
+    assert(false, "*** expected an exception")
+  end
+
+  def test_string_writer_pop_with_key
+    w = Oj::StringWriter.new(:indent => 0)
+    begin
+      w.pop()
+    rescue Exception
+      assert(true)
+      return
+    end
+    assert(false, "*** expected an exception")
+  end
+
+  def test_string_writer_obj_no_key
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object()
+    begin
+      w.push_value(59)
+    rescue Exception
+      assert(true)
+      return
+    end
+    assert(false, "*** expected an exception")
+  end
+
+  def test_string_writer_deep
+    cnt = 1000
+    w = Oj::StringWriter.new(:indent => 0)
+    cnt.times do
+      w.push_array()
+    end
+    cnt.times do
+      w.pop()
+    end
+    # if no exception then passed
+    assert(true)
+  end
+
+  def test_string_writer_pop_all
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_object()
+    w.push_object("a1")
+    w.pop()
+    w.push_array("a2")
+    w.push_value(3)
+    w.push_array()
+    w.pop_all()
+    assert_equal(%|{"a1":{},"a2":[3,[]]}\n|, w.to_s)
+  end
+
+  def test_string_writer_reset
+    w = Oj::StringWriter.new(:indent => 0)
+    w.push_array()
+    w.pop()
+    w.reset()
+    assert_equal('', w.to_s)
+  end
+
+  # Stream Writer
+
+  def test_stream_writer_empty_array
+    output = StringIO.open("", "w+")
+    w = Oj::StreamWriter.new(output, :indent => 0)
+    w.push_array()
+    w.pop()
+    assert_equal("[]\n", output.string())
+  end
+
+  def test_stream_writer_mixed_stringio
+    output = StringIO.open("", "w+")
+    w = Oj::StreamWriter.new(output, :indent => 0)
+    w.push_object()
+    w.push_object("a1")
+    w.pop()
+    w.push_object("a2")
+    w.push_array("b")
+    w.push_value(7)
+    w.push_value(true)
+    w.push_value("string")
+    w.pop()
+    w.pop()
+    w.push_object("a3")
+    w.pop()
+    w.pop()
+    assert_equal(%|{"a1":{},"a2":{"b":[7,true,"string"]},"a3":{}}\n|, output.string())
+  end
+
+  def test_stream_writer_mixed_file
+    filename = File.join(File.dirname(__FILE__), 'open_file_test.json')
+    File.open(filename, "w") do |f|
+      w = Oj::StreamWriter.new(f, :indent => 0)
+      w.push_object()
+      w.push_object("a1")
+      w.pop()
+      w.push_object("a2")
+      w.push_array("b")
+      w.push_value(7)
+      w.push_value(true)
+      w.push_value("string")
+      w.pop()
+      w.pop()
+      w.push_object("a3")
+      w.pop()
+      w.pop()
+    end
+    content = File.read(filename)
+    assert_equal(%|{"a1":{},"a2":{"b":[7,true,"string"]},"a3":{}}\n|, content)
+  end
+
+  def test_stream_writer_nested_key_object
+    output = StringIO.open("", "w+")
+    w = Oj::StreamWriter.new(output, :indent => 0)
+    w.push_object()
+    w.push_key('a1')
+    w.push_object()
+    w.pop()
+    w.push_key('a2')
+    w.push_object('x')
+    w.push_object('b')
+    w.pop()
+    w.pop()
+    w.push_key('a3')
+    w.push_object()
+    w.push_key('a4')
+    w.push_value(37)
+    w.pop()
+    w.pop()
+    assert_equal(%|{"a1":{},"a2":{"b":{}},"a3":{"a4":37}}\n|, output.string())
+  end
+
+  def test_stream_writer_push_null_value_with_key
+    output = StringIO.open("", "w+")
+    w = Oj::StreamWriter.new(output, :indent => 0)
+    w.push_object()
+    w.push_value(nil, 'nothing')
+    w.pop()
+    assert_equal(%|{"nothing":null}\n|, output.string())
+  end
+end # OjWriter