require_relative "test_helper"
require "splunk-sdk-ruby"

require 'json'

include Splunk

# URI's classes compare by object identity, which is exactly what we
# *don't* want to do. Instead we use simple textual identity.
module URI
  class Generic
    def ==(other)
      return self.to_s == other.to_s
    end
  end

  class HTTPS
    def ==(other)
      return self.to_s == other.to_s
    end
  end

  class HTTP
    def ==(other)
      return self.to_s == other.to_s
    end
  end
end

class TestAtomFeed < Test::Unit::TestCase
  # If Nokogiri is available, we'll test AtomFeed against both it
  # and REXML. Otherwise, we'll print a warning and test only against
  # REXML. REXML is part of the standard library in Ruby 1.9, so it will
  # always be present.
  if nokogiri_available?
    xml_libraries = [:nokogiri, :rexml]
  else
    xml_libraries = [:rexml]
    puts "Nokogiri not installed. Skipping."
  end

  data_path = File.dirname(File.expand_path(__FILE__))
  test_cases = JSON::parse(open(data_path + "/data/atom_test_data.json").read())

  xml_libraries.each do |xml_library|
    test_cases.each_entry do |filename, expected|
      define_method("test_#{xml_library}_#{filename}".intern()) do

        file = File.open(File.dirname(File.expand_path(__FILE__)) + "/data/atom/#{filename}.xml")
        Splunk::require_xml_library(xml_library)
        feed = Splunk::AtomFeed.new(file)

        # In the assert statements below, the output of the code is first,
        # the expected data second, which is breaking convention, but has
        # to be this way since we need to match URLs which are URI objects
        # in the output of the code to URLs which are strings in the expected
        # data from the JSON file. URI has been patched to make this
        # work...but only if the == method is called on the URI object,
        # not the string. == is not commutative.

        # To make debugging easy, test the metadata a key at
        # a time, since Test::Unit doesn't display diffs.
        # Then test the whole thing at the end to make sure it all matches.
        expected["metadata"].each_entry do |key, value|
          assert_equal([filename, key, feed.metadata[key]],
                       [filename, key, value])
        end
        assert_equal(feed.metadata, expected["metadata"])

        # To make debugging easy, test each key of each entry
        # separately, since Test::Unit doesn't display diffs.
        # Then test the whole thing at the end to make sure it all matches.
        expected["entries"].each_with_index do |entry, index|
          entry.each_entry do |key, value|
            assert_equal([filename, index, key, feed.entries[index][key]],
                         [filename, index, key, value])
          end
        end
        assert_equal(feed.entries, expected["entries"])
      end
    end
  end
end