require "stringio"
require "json"

class StreamerTest < MiniTest::Unit::TestCase
  attr_reader :output, :json

  def setup
    @output = StringIO.new
    @json = Elster::Streamer.new(output)
  end

  def teardown
    @parsed = nil
  end

  def parsed
    @parsed ||= JSON.parse(output.string)
  end

  def test_object_single_value
    json.key(:name, "Bobert")
    json.close

    assert_equal "Bobert", parsed["name"]
  end

  def test_object_multiple_values
    json.key(:first_name, "Bobert")
    json.key(:last_name, "Price")
    json.close

    assert_equal "Bobert", parsed["first_name"]
    assert_equal "Price", parsed["last_name"]
  end

  def test_object_numbers
    json.key(:age, 40)
    json.close

    assert_equal 40, parsed["age"]
  end

  def test_array_muliple
    json.add(1)
    json.add(2)
    json.close

    assert_equal [1,2], parsed
  end

  def test_array_single
    json.add("Hooper")
    json.close

    assert_equal ["Hooper"], parsed
  end

  def test_nested_object
    json.key(:person) do
      json.key(:name, "Bobert")
      json.key(:age, 55)
    end
    json.close

    assert_equal "Bobert", parsed["person"]["name"]
  end

  def test_deep_nested_object
    json.key(:person) do
      json.key(:name) do
        json.key(:first, "Bobert")
      end
    end
    json.close

    assert_equal "Bobert", parsed["person"]["name"]["first"]
  end

  def test_nested_array_in_object
    json.key(:children) do
      json.add(1)
      json.add(2)
    end
    json.close

    assert_equal [1,2], parsed["children"]
  end

  def test_nested_object_in_array
    json.add do
      json.key(:child) do
        json.key(:name, "Bobertson")
      end
    end
    json.close

    assert_equal "Bobertson", parsed[0]["child"]["name"]
  end

  def test_array_nested_object
    json.add do
      json.key(:name, "Bobert")
    end
    json.add do
      json.key(:age, 44)
    end
    json.close

    assert_equal "Bobert", parsed[0]["name"]
    assert_equal 44, parsed[1]["age"]
  end

  def test_ruby_hash
    json.key("name", {:first => "Bobert"})
    json.close

    assert_equal "Bobert", parsed["name"]["first"]
  end

  def test_ruby_array
    json.key("an_array", [1, 2])
    json.close

    assert_equal [1, 2], parsed["an_array"]
  end

  def test_string_with_cr
    json.key("description", "I am \nmulti\nlined")
    json.close

    assert_equal "I am \nmulti\nlined", parsed["description"]
  end
end