require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')

class Tokamak::Builder::JsonTest < Test::Unit::TestCase

  def test_media_type_should_be_json
    assert_equal ["application/json"], Tokamak::Builder::Json.media_types
  end
  
  def test_the_most_simple_json
    obj = [{ :foo => "bar" }]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.members do |member, some_foos|
        member.values do |values|
          values.id some_foos[:foo]
        end        
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "an_id", hash.id
    assert_equal "bar"  , hash.members.first.id
    assert hash.members.kind_of?(Array)
  end

  def test_root_set_on_builder
    obj = [{ :foo => "bar" }, { :foo => "zue" }]
    json = Tokamak::Builder::Json.build(obj, :root => "foos") do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.members do |member, some_foos|
        member.values do |values|
          values.id some_foos[:foo]
        end        
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert hash.has_key?("foos")
    assert_equal "an_id", hash.foos.id
    assert_equal "bar"  , hash.foos.members.first.id
  end
  
  def test_collection_set_on_members
    obj = { :foo => "bar" }
    a_collection = [1,2,3,4]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.members(:collection => a_collection) do |member, number|
        member.values do |values|
          values.id number
        end        
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "an_id", hash.id
    assert_equal 1      , hash.members.first.id
    assert_equal 4      , hash.members.size
  end
  
  def test_collection_set_on_members_only_one
    obj = { :foo => "bar" }
    a_collection = [1]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.members(:collection => a_collection) do |member, number|
        member.values do |values|
          values.id number
        end        
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "an_id", hash.id
    assert_equal 1      , hash.members.first.id
    assert_equal 1      , hash.members.size
    assert hash.members.kind_of?(Array)
  end
  
  def test_raise_exception_for_not_passing_a_collection_as_parameter_to_members
    obj = 42
    
    assert_raise Tokamak::BuilderError do
      json = Tokamak::Builder::Json.build(obj) do |collection, number|
        collection.values do |values|
          values.id number
        end
      
        collection.members do |member, item|
          member.values do |values|
            values.id item
          end        
        end
      end
    end
  end

  def test_root_set_on_members
    obj = [{ :foo => "bar" }, { :foo => "zue" }]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.members(:root => "foos") do |member, some_foos|
        member.values do |values|
          values.id some_foos[:foo]
        end        
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "an_id", hash.id
    assert_equal "bar"  , hash.foos.first.id
    assert_equal 2      , hash.foos.size
  end
  
  def test_nested_crazy_values
    obj = [{ :foo => "bar" }, { :foo => "zue" }]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.body {
          values.face {
            values.eyes  "blue"
            values.mouth "large"
          }
          values.legs [
            { :right => { :fingers_count => 5 } }, { :left => { :fingers_count => 4 } }
          ]
        }
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "blue" , hash.body.face.eyes
    assert_equal "large", hash.body.face.mouth
    assert_equal 2      , hash.body.legs.count
    assert_equal 4      , hash.body.legs.last.left.fingers_count
  end
  
  def test_build_single_link_in_collection_and_member
    obj = [{ :foo => "bar" }]
    json = Tokamak::Builder::Json.build(obj) do |collection|
      collection.values do |values|
        values.id "an_id"
      end
      
      collection.link 'self', "http://example.com/an_id"
      
      collection.members do |member, some_foos|
        member.values do |values|
          values.id some_foos[:foo]
        end
        
        member.link 'self', "http://example.com/an_id/#{some_foos[:foo]}"
      end
    end
    
    hash = JSON.parse(json).extend(Methodize)
    
    assert_equal "an_id", hash.id
    assert_equal "bar"  , hash.members.first.id
    assert hash.members.kind_of?(Array)
    assert hash.link.kind_of?(Array)
    assert hash.members.first.link.kind_of?(Array)
  end

  def test_build_full_collection
    time = Time.now
    some_articles = [
      {:id => 1, :title => "a great article", :updated => time},
      {:id => 2, :title => "another great article", :updated => time}
    ]
    
    json = Tokamak::Builder::Json.build(some_articles) do |collection|
      collection.values do |values|
        values.id      "http://example.com/json"
        values.title   "Feed"
        values.updated time

        values.author { 
          values.name  "John Doe"
          values.email "joedoe@example.com"
        }
        
        values.author { 
          values.name  "Foo Bar"
          values.email "foobar@example.com"
        }
      end
      
      collection.link("next"    , "http://a.link.com/next")
      collection.link("previous", "http://a.link.com/previous")
      
      collection.members(:root => "articles") do |member, article|
        member.values do |values|
          values.id      "uri:#{article[:id]}"                   
          values.title   article[:title]
          values.updated article[:updated]              
        end
        
        member.link("image", "http://example.com/image/1")
        member.link("image", "http://example.com/image/2", :type => "application/json")
      end
    end

    hash = JSON.parse(json)
    hash.extend(Methodize)
    
    assert_equal "John Doe"               , hash.author.first.name
    assert_equal "foobar@example.com"     , hash.author.last.email
    assert_equal "http://example.com/json", hash.id
    
    assert_equal "http://a.link.com/next" , hash.link.first.href
    assert_equal "next"                   , hash.link.first.rel
    assert_equal "application/json"       , hash.link.last.type
    
    assert_equal "uri:1"                      , hash.articles.first.id
    assert_equal "a great article"            , hash.articles.first.title
    assert_equal "http://example.com/image/1" , hash.articles.last.link.first.href
    assert_equal "image"                      , hash.articles.last.link.first.rel
    assert_equal "application/json"           , hash.articles.last.link.last.type
  end

  def test_build_full_member
    time = Time.now
    an_article = {:id => 1, :title => "a great article", :updated => time}
    
    json = Tokamak::Builder::Json.build(an_article, :root => "article") do |member, article|
      member.values do |values|
        values.id      "uri:#{article[:id]}"           
        values.title   article[:title]
        values.updated article[:updated]
        
        values.domain("xmlns" => "http://a.namespace.com") {
          member.link("image", "http://example.com/image/1")
          member.link("image", "http://example.com/image/2", :type => "application/atom+xml")
        }
      end
      
      member.link("image", "http://example.com/image/1")
      member.link("image", "http://example.com/image/2", :type => "application/json")                                
    end
    
    hash = JSON.parse(json)
    hash.extend(Methodize)
        
    assert_equal "uri:1"                      , hash.article.id
    assert_equal "a great article"            , hash.article.title
    assert_equal "http://example.com/image/1" , hash.article.link.first.href
    assert_equal "image"                      , hash.article.link.first.rel
    assert_equal "application/json"           , hash.article.link.first.type
    
    assert_equal "http://example.com/image/1" , hash.article.domain.link.first.href
    assert_equal "image"                      , hash.article.domain.link.first.rel
    assert_equal "application/json"           , hash.article.domain.link.first.type
    assert_equal "http://a.namespace.com"     , hash.article.domain.xmlns
  end
end