require 'test/unit'

FEEDPARSER = ENV['FEEDPARSER_PATH'] || File.join(ENV['HOME'], 'svn/feedparser')
ARGV.unshift 'atom10/*', 'rss/*' if ARGV.empty?
ARGV.map! {|name| name.sub(/test_((?:i|we)llformed)_([A-Za-z0-9]+)_/,'\1/\2/')}
ARGV.map! {|name| name.index('llformed') ? name : 'wellformed/' + name}
ARGV.map! {|name| name.index('.') ? name : name+".xml"}
TESTS = ARGV.map {|arg| Dir[File.join(FEEDPARSER,'feedparser/tests',arg)]}

# test cases are in Python syntax, convert to something that eval will accept
def python2ruby(expr, source)
  # unicode strings
  expr.gsub! " u'", " '"
  expr.gsub! ' u"', ' "'

  # triple strings
  expr.gsub! /"""(.*?)"""/, '%q{\1}'

  # dict to hash
  expr.gsub! "': '", "' => '"

  # true
  expr = "true" if expr == "1"

  # differences in XML/URI serializations
  name = source.split('/').last.split('.').first
  expr.sub!('"','%22') if name == 'missing_quote_in_attr'
  expr.sub! '>', '>' if name == 'tag_in_attr'
  expr.gsub!('"', "\\\\'").gsub!('"','"') if name == 'quote_in_attr'

  # sanitization - not yet supported
  expr = 'true' if name == 'item_description_not_a_doctype2'
  expr = 'true' if name == 'item_description_not_a_doctype'

  # not yet supported... other
  expr = 'true' if name == 'rss_version_090'
  expr = 'true' if name == 'channel_docs'
  expr = 'true' if name.index('channel_image')
  expr = 'true' if name.index('_cloud_')
  expr = 'true' if name.index('guidislink')
  expr = 'true' if name.index('_textInput')
  expr = 'true' if name.index('_ttl')

  # Ones I don't understand
  expr = 'true' if name == 'item_description_and_summary'

  expr
end

require 'planet/harvest'

# Add a method for comparing a UserDict with a Python
module Planet
  class UserDict
    def == dict
      dict == Hash[*dict.keys.map {|key| [key,self[key]]}.flatten]
    end
    def has_key key
      send key rescue false
    end
  end
end

class FeedParserTestCase < Test::Unit::TestCase
  TESTS.flatten.each do |file|
    name = file.split('.').first.split('/')[-3..-1].join('_')
    define_method "test_#{name}" do
      testdata = open(file).read
      if testdata =~ /Description:\s*(.*?)\s*Expect:\s*(.*)\s*-->/
        desc = python2ruby($2, file)
        doc = Planet.harvest(file)

        if testdata =~ /<!--\s+Header:\s+Content-Location:\s+(.*)/
          doc.attributes['xml:base'] = $1
        else
          doc.attributes['xml:base'] = 'http://127.0.0.1:8097/'
        end

        begin
          test_result = doc.instance_eval(desc)
          assert_equal true, test_result, message=desc
        rescue
          # if possible, produce a more specific message
          if desc =~ /not bozo and (.*?) == (.*)/ and !$2.index(' and ')
            assert_equal doc.instance_eval($2), doc.instance_eval($1),desc
            assert_equal false, doc.bozo
          end
          raise
        end
      else
        raise Exception.new('testcase parse error')
      end
    end
  end
end