require File.dirname(__FILE__) + '/spec_helper.rb'

class Post
  include HappyMapper
  
  attribute :href, String
  attribute :hash, String
  attribute :description, String
  attribute :tag, String
  attribute :time, Time
  attribute :others, Integer
  attribute :extended, String
end

class User  
  include HappyMapper
  
  element :id, Integer
  element :name, String
  element :screen_name, String
  element :location, String
  element :description, String
  element :profile_image_url, String
  element :url, String
  element :protected, Boolean
  element :followers_count, Integer
end

class Status
  include HappyMapper
  
  element :id, Integer
  element :text, String
	element :created_at, Time
	element :source, String
	element :truncated, Boolean
	element :in_reply_to_status_id, Integer
	element :in_reply_to_user_id, Integer
	element :favorited, Boolean
	has_one :user, User
end

module PITA
  class Item
    include HappyMapper
    
    tag 'Item' # if you put class in module you need tag
    element :asin, String, :tag => 'ASIN'
    element :detail_page_url, String, :tag => 'DetailPageURL'
    element :manufacturer, String, :tag => 'Manufacturer', :deep => true
  end

  class Items
    include HappyMapper
    
    tag 'Items' # if you put class in module you need tag
    element :total_results, Integer, :tag => 'TotalResults'
    element :total_pages, Integer, :tag => 'TotalPages'
    has_many :items, Item
  end
end

describe HappyMapper do
  
  describe "being included into another class" do
    before do
      Foo.instance_variable_set("@attributes", {})
      Foo.instance_variable_set("@elements", {})
    end
    class Foo; include HappyMapper end
    
    it "should set attributes to an array" do
      Foo.attributes.should == []
    end
    
    it "should set @elements to a hash" do
      Foo.elements.should == []
    end
    
    it "should allow adding an attribute" do
      lambda {
        Foo.attribute :name, String
      }.should change(Foo, :attributes)
    end
    
    it "should be able to get all attributes in array" do
      Foo.attribute :name, String
      Foo.attributes.size.should == 1
    end
    
    it "should allow adding an element" do
      lambda {
        Foo.element :name, String
      }.should change(Foo, :elements)
    end
    
    it "should be able to get all elements in array" do
      Foo.element(:name, String)
      Foo.elements.size.should == 1
    end
    
    it "should allow has one association" do
      Foo.has_one(:user, User)
      element = Foo.elements.first
      element.name.should == 'user'
      element.type.should == User
      element.options[:single] = true
    end
    
    it "should allow has many association" do
      Foo.has_many(:users, User)
      element = Foo.elements.first
      element.name.should == 'users'
      element.type.should == User
      element.options[:single] = false
    end
    
    it "should default tag name to class" do
      Foo.get_tag_name.should == 'foo'
    end
    
    it "should allow setting tag name" do
      Foo.tag('FooBar')
      Foo.get_tag_name.should == 'FooBar'
    end
    
    it "should provide #parse" do
      Foo.should respond_to(:parse)
    end
  end
  
  describe "#attributes" do
    it "should only return attributes for the current class" do
      Post.attributes.size.should == 7
      Status.attributes.size.should == 0
    end
  end
  
  describe "#elements" do
    it "should only return elements for the current class" do
      Post.elements.size.should == 0
      Status.elements.size.should == 9
    end
  end
  
  describe "#parse (with xml attributes mapping to ruby attributes)" do
    before do
      @posts = Post.parse(File.read(File.dirname(__FILE__) + '/fixtures/posts.xml'))
    end
    
    it "should get the correct number of elements" do
      @posts.size.should == 20
    end
    
    it "should properly create objects" do
      first = @posts.first
      first.href.should == 'http://roxml.rubyforge.org/'
      first.hash.should == '19bba2ab667be03a19f67fb67dc56917'
      first.description.should == 'ROXML - Ruby Object to XML Mapping Library'
      first.tag.should == 'ruby xml gems mapping'
      first.time.should == Time.utc(2008, 8, 9, 5, 24, 20)
      first.others.should == 56
      first.extended.should == 'ROXML is a Ruby library designed to make it easier for Ruby developers to work with XML. Using simple annotations, it enables Ruby classes to be custom-mapped to XML. ROXML takes care of the marshalling and unmarshalling of mapped attributes so that developers can focus on building first-class Ruby classes.'
    end
  end
  
  describe "#parse (with xml elements mapping to ruby attributes)" do
    before do
      @statuses = Status.parse(File.read(File.dirname(__FILE__) + '/fixtures/statuses.xml'))
    end
    
    it "should get the correct number of elements" do
      @statuses.size.should == 20
    end
    
    it "should properly create objects" do
      first = @statuses.first
      first.id.should == 882281424
      first.created_at.should == Time.mktime(2008, 8, 9, 1, 38, 12)
      first.source.should == 'web'
      first.truncated.should be_false
      first.in_reply_to_status_id.should == 1234
      first.in_reply_to_user_id.should == 12345
      first.favorited.should be_false
      first.user.id.should == 4243
      first.user.name.should == 'John Nunemaker'
      first.user.screen_name.should == 'jnunemaker'
      first.user.location.should == 'Mishawaka, IN, US'
      first.user.description.should == 'Loves his wife, ruby, notre dame football and iu basketball'
      first.user.profile_image_url.should == 'http://s3.amazonaws.com/twitter_production/profile_images/53781608/Photo_75_normal.jpg'
      first.user.url.should == 'http://addictedtonew.com'
      first.user.protected.should be_false
      first.user.followers_count.should == 486
    end
  end
  
  # TODO: someone please get xml with namespaces working, kthxbai
  describe "#parse (with xml that has namespace)" do
    before do
      file_contents = File.read(File.dirname(__FILE__) + '/fixtures/pita.xml')
      @items = PITA::Items.parse(file_contents, :single => true, :use_default_namespace => true)
    end
    
    it "should properly create objects" do
      @items.total_results.should == 22
      @items.total_pages.should == 3
      first = @items.items.first
      first.asin.should == '0321480791'
      first.detail_page_url.should == 'http://www.amazon.com/gp/redirect.html%3FASIN=0321480791%26tag=ws%26lcode=xm2%26cID=2025%26ccmID=165953%26location=/o/ASIN/0321480791%253FSubscriptionId=dontbeaswoosh'
      first.manufacturer.should == 'Addison-Wesley Professional'
    end
  end
end