require './spec/WindowsInstaller.rb'

require './admin.rb'
require './lib/command.rb'

def get_product_name(msi_file, arg2)
  product_name = File.basename(msi_file, File.extname(msi_file))
  product_name = arg2[:product_name] if(arg2.has_key?(:product_name))
  
  return product_name
end

def test_msi(msi_file, arg2)
  product_name = get_product_name(msi_file, arg2)
	
  msi_info = WindowsInstaller.msi_records(msi_file)
  #puts msi_info.to_s
  
  if(arg2.has_key?(:product_name))
    raise "ProductName is #{msi_info['ProductName']} expected #{product_name}" unless(product_name == msi_info['ProductName'])
  end

  if(arg2.has_key?(:product_code))
    expected = arg2[:product_code].upcase
	raise "ProductCode is #{msi_info['ProductCode']} expected #{expected}" unless(expected == msi_info['ProductCode'])
  end

  if(arg2.has_key?(:upgrade_code))
    expected = arg2[:upgrade_code].upcase
	raise "UpgradeCode is #{msi_info['UpgradeCode']} expected #{expected}" unless(expected == msi_info['UpgradeCode'])
  end
  
  expected_product_version = '1.0.0.0'
  expected_product_version = arg2[:version] if(arg2.has_key?(:version))
  raise "Invalid product version #{msi_info['ProductVersion']}" if(msi_info['ProductVersion'] != expected_product_version)

  expected_manufacturer = 'Not Set'
  expected_manufacturer = arg2[:manufacturer] if(arg2.has_key?(:manufacturer))
  raise "Invalid Manufacturer #{msi_info['Manufacturer']}" if(msi_info['Manufacturer'] != expected_manufacturer)
end

def test_install(name, msi_file, arg2, callback=nil)
  arg2 = { files: arg2} unless(arg2.kind_of?(Hash))
  msi_file = msi_file.gsub(/\//) { |s| s = '\\' }
  test_msi(msi_file, arg2)
  
  msi_info = WindowsInstaller.msi_records(msi_file)
  product_name = msi_info['ProductName']
	
  if(admin?)
	while(WindowsInstaller.installed?(product_name))
	  WindowsInstaller.uninstall(product_name)
	end
    raise "#{name}: Uninstall #{product_name} before running tests" if(WindowsInstaller.installed?(product_name))
    
	begin
	  WindowsInstaller.install(msi_file)			
	  raise "#{name}: Product name #{product_name} is not installed" unless(WindowsInstaller.installed?(product_name))

	  eval callback unless(callback == nil)
	ensure
	  WindowsInstaller.uninstall(msi_file) if(WindowsInstaller.installed?(product_name))
	  raise "Failed to uninstall product #{product_name}" if(WindowsInstaller.installed?(product_name))
	end
  end
end