require 'test_plugin_helper'
module ForemanVirtWhoConfigure
  class OutputGeneratorTest < ActiveSupport::TestCase
    let(:config) { FactoryBot.create(:virt_who_config) }
    let(:generator) { OutputGenerator.new(config) }
    let(:output) { generator.virt_who_output }
    let(:bash_script_output) { generator.virt_who_output(:bash_script) }

    describe 'cr_password for libvirt' do
      test 'empty password is replaced by fake value so that virt-who runs in an non-interactive mode' do
        config.hypervisor_password = nil
        config.hypervisor_type = 'libvirt'
        # (echo bGlidmlydF9mYWtlX3Bhc3N3b3Jk|base64 -d) = OutputGenerator::LIBVIRT_FAKE_PASSWORD
	assert_includes output, 'virt-who-password --password $(echo bGlidmlydF9mYWtlX3Bhc3N3b3Jk|base64 -d)'
      end
    end

    describe 'filtering' do
      test 'it should not filter anything for unlimited configs' do
        assert_equal ForemanVirtWhoConfigure::Config::UNLIMITED, config.listing_mode.to_i
        assert_not_includes output, 'filter_hosts'
        assert_not_includes output, 'exclude_hosts'
      end

      test 'whitelisting should set filter_hosts directive' do
        config.listing_mode = ForemanVirtWhoConfigure::Config::WHITELIST
        config.whitelist = 'a.host'
        assert_includes output, 'filter_hosts=a.host'
      end

      test 'whitelisting should set exclude_hosts directive' do
        config.listing_mode = ForemanVirtWhoConfigure::Config::BLACKLIST
        config.blacklist = 'a.host'
        assert_includes output, 'exclude_hosts=a.host'
      end

      test 'filtering output is stripped' do
        config.listing_mode = ForemanVirtWhoConfigure::Config::WHITELIST
        config.whitelist = ' name '
        assert_includes output, 'filter_hosts=name'
      end

      test 'filtering output is stripped for multiple values as well' do
        config.listing_mode = ForemanVirtWhoConfigure::Config::WHITELIST
        config.whitelist = ' host1,host2  '
        assert_includes output, 'filter_hosts=host1,host2'
      end

      test 'filtering output new line characters are removed' do
        config.listing_mode = ForemanVirtWhoConfigure::Config::WHITELIST
        config.whitelist = "aaaa,bbb,\r\nccc,\r\n\r\nddd\r\n\r\neee"
        assert_includes output, 'filter_hosts=aaaa,bbb,ccc,dddeee'

        config.whitelist = "aaaa,bbb,\rccc,\n\r\nddd\neee"
        assert_includes output, 'filter_hosts=aaaa,bbb,ccc,dddeee'
      end
    end

    describe 'proxy attributes' do
      test 'it does not set any proxy attributes by default and set no_proxy to * to bypass sub-man proxy' do
        assert_nil config.http_proxy
        assert_nil config.no_proxy
        assert_not_includes output, 'http_proxy'
        assert_includes output, 'no_proxy=*'
      end

      test 'it configures proxy when set' do
        http_proxy = HttpProxy.create!(name: "test", url: "http://test.com")
        config.http_proxy = http_proxy
        assert_includes output, 'http_proxy=http://test.com'
      end

      test 'it configures no_proxy when set' do
        config.no_proxy = '*'
        assert_includes output, 'no_proxy=*'
      end

      test 'proxy_strings prints both proxy and no proxy if present' do
        http_proxy = HttpProxy.create!(name: "test", url: "http://test.com")
        config.http_proxy = http_proxy
        config.no_proxy = 'b'
        assert_includes generator.proxy_strings, "\nhttp_proxy=http://test.com"
        assert_includes generator.proxy_strings, "\nno_proxy=b"
      end

      test 'proxy_strings ignores empty string values' do
        config.http_proxy = nil
        config.no_proxy = ''
        assert_not_includes generator.proxy_strings, 'http_proxy'
        assert_includes generator.proxy_strings, 'no_proxy=*'
      end

      test 'proxy_strings removes any new line chars' do
        config.no_proxy = "\nx\ny\nz"
        assert_includes generator.proxy_strings, "\nno_proxy=xyz"
      end

      test 'it configures https proxy when https proxy is set' do
        http_proxy = HttpProxy.create!(name: "test", url: "https://test.com")
        config.http_proxy = http_proxy
        assert_includes output, 'https_proxy=https://test.com'
      end
    end

    describe 'error codes' do
      test 'returns code number for known error name specified as symbol' do
        assert_equal 1, generator.error_code(:virt_who_too_old)
      end

      test 'returns code number for known error name specified as string' do
        assert_equal 0, generator.error_code('success')
      end

      test 'returns nil for unknown error name' do
        assert_nil generator.error_code('unknown')
      end
    end

    describe 'output format' do
      test 'it returns the inline content of script if no format is specified' do
        assert_not_includes output, '#!/bin/bash'
        assert_not_includes output, 'exit $result_code'
        assert_includes output, 'step 1 "Installing virt-who"'
      end

      test 'it returns the bash script with shebang and exit code for :bash_script format' do
        assert_includes bash_script_output, "#!/bin/bash\n"
        assert_includes bash_script_output, 'exit $result_code'
        assert_includes bash_script_output, 'step 1 "Installing virt-who"'
      end
    end

    describe 'nutanix ahv' do
      test 'options specific to ahv are all set' do
        config.hypervisor_type = 'ahv'
        config.prism_flavor = 'central'
        config.ahv_internal_debug = true

        assert_includes output, 'type=ahv'
        assert_includes output, 'prism_central=true'
        assert_includes output, 'ahv_internal_debug=true'
      end

      test 'options specific to ahv are not all set' do
        config.hypervisor_type = 'ahv'
        config.prism_flavor = 'central'

        assert_includes output, 'type=ahv'
        assert_includes output, 'prism_central=true'
      end
    end
  end
end