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

class TestDeathMasterFileLoader < Test::Unit::TestCase

  def setup
    setup_death_master_file_table
  end

  def test_validations_for_load_file
    exception = assert_raise ArgumentError do
      DeathMasterFileLoader.new(nil,nil)
    end
    assert_equal "path_or_url not specified", exception.message

    exception = assert_raise ArgumentError do
      DeathMasterFileLoader.new('some/path.txt',nil)
    end
    assert_equal "as_of not specified", exception.message

    exception = assert_raise Errno::ENOENT do
      DeathMasterFileLoader.new('some/path.txt','2009-01-01')
    end
    assert_equal "No such file or directory - some/path.txt", exception.message

    exception = assert_raise URI::InvalidURIError do
      DeathMasterFileLoader.new('https://dmf.bad_url.gov/dmldata/monthly/bad_url','2009-01-01')
    end
    assert_equal "the scheme https does not accept registry part: dmf.bad_url.gov (or bad hostname?)", exception.message

    exception = assert_raise ArgumentError do
      DeathMasterFileLoader.new('https://dmf.ntis.gov/bad_url','2009-01-01')
    end
    assert_equal "Invalid URL: https://dmf.ntis.gov/bad_url", exception.message

    exception = assert_raise ArgumentError do
      DeathMasterFileLoader.new('https://dmf.ntis.gov/dmldata/monthly/bad_url','2009-01-01')
    end
    assert_equal "Authorization Required: Invalid username or password.  Set the variables SsnValidator::Ntis.user_name and SsnValidator::Ntis.password in your environment.rb file.", exception.message
  end

  def test_should_load_table_from_file
    assert_equal(0,DeathMasterFile.count)
    DeathMasterFileLoader.new(File.dirname(__FILE__) + '/files/test_dmf_initial_load.txt','2009-01-01').load_file
    assert DeathMasterFile.count == 5

    #load update file
    #update file adds 1 record, deletes 2 records, and changes 2 records
    DeathMasterFileLoader.new(File.dirname(__FILE__) + '/files/test_dmf_update_load.txt','2009-02-01').load_file
    assert DeathMasterFile.count == 4
    assert 1, DeathMasterFile.count(:conditions => {:last_name => 'NEW'})
    assert 2, DeathMasterFile.count(:conditions => {:last_name => 'CHANGED'})
  end

  def test_should_not_load_table_if_as_of_already_in_table_for_load_file
    DeathMasterFileLoader.new(File.dirname(__FILE__) + '/files/test_dmf_initial_load.txt','2009-01-01').load_file
    assert DeathMasterFile.count > 0
    record_count = DeathMasterFile.count
    
    exception = assert_raise ArgumentError do
      DeathMasterFileLoader.new(File.dirname(__FILE__) + '/files/test_dmf_update_load.txt','2008-01-01').load_file
    end
    assert_equal "A more recent file has already been processed.  DB as_of date 2009-01-01", exception.message

    assert_equal(record_count, DeathMasterFile.count)
  end

  def test_should_load_all_unloaded_updates
    #initial load was 3 months ago
    initial_run = Date.today.beginning_of_month - 3.months
    DeathMasterFileLoader.new(File.dirname(__FILE__) + '/files/test_dmf_initial_load.txt',initial_run.strftime('%Y-%m-%d')).load_file
    assert DeathMasterFile.count == 5

    #this will load the last 3 months of files
    #this has been mocked, not really hitting the web site for this
    DeathMasterFileLoader.load_update_files_from_web
    assert DeathMasterFile.count == 8
    assert DeathMasterFile.find_by_as_of(initial_run.to_s(:db))
    assert DeathMasterFile.find_by_as_of((initial_run += 1.month).to_s(:db))
    assert DeathMasterFile.find_by_as_of((initial_run += 1.month).to_s(:db))
    assert DeathMasterFile.find_by_as_of((initial_run += 1.month).to_s(:db))


  end

end