require 'minitest/autorun'
require 'mocha/setup'
require_relative '../../lib/inspec-compliance/api.rb'

describe InspecPlugins::Compliance::API do
  let(:profiles_response) do
    [{ 'name'=>'apache-baseline',
      'title'=>'DevSec Apache Baseline',
      'maintainer'=>'DevSec Hardening Framework Team',
      'copyright'=>'DevSec Hardening Framework Team',
      'copyright_email'=>'hello@dev-sec.io',
      'license'=>'Apache 2 license',
      'summary'=>'Test-suite for best-practice apache hardening',
      'version'=>'2.0.2',
      'supports'=>[{ 'os-family'=>'unix' }],
      'depends'=>nil,
      'owner_id'=>'admin' },
     { 'name'=>'apache-baseline',
      'title'=>'DevSec Apache Baseline',
      'maintainer'=>'Hardening Framework Team',
      'copyright'=>'Hardening Framework Team',
      'copyright_email'=>'hello@dev-sec.io',
      'license'=>'Apache 2 license',
      'summary'=>'Test-suite for best-practice apache hardening',
      'version'=>'2.0.1',
      'supports'=>[{ 'os-family'=>'unix' }],
      'depends'=>nil,
      'latest_version'=>'2.0.2',
      'owner_id'=>'admin' },
     { 'name'=>'cis-aix-5.3-6.1-level1',
      'title'=>'CIS AIX 5.3 and AIX 6.1 Benchmark Level 1',
      'maintainer'=>'Chef Software, Inc.',
      'copyright'=>'Chef Software, Inc.',
      'copyright_email'=>'support@chef.io',
      'license'=>'Proprietary, All rights reserved',
      'summary'=>'CIS AIX 5.3 and AIX 6.1 Benchmark Level 1 translated from SCAP',
      'version'=>'1.1.0',
      'supports'=>nil,
      'depends'=>nil,
      'latest_version'=>'1.1.0-3',
      'owner_id'=>'admin' }]
  end

  describe '.version' do
    let(:headers) { 'test-headers' }
    let(:config) do
      {
        'server' => 'myserver',
        'insecure' => true,
      }
    end

    before do
      InspecPlugins::Compliance::API.expects(:get_headers).returns(headers)
    end

    describe 'when a 404 is received' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('404')
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({})
      end
    end

    describe 'when the returned body is nil' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('200')
        response.stubs(:body).returns(nil)
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({})
      end
    end

    describe 'when the returned body is an empty string' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('200')
        response.stubs(:body).returns('')
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({})
      end
    end

    describe 'when the returned body has no version key' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('200')
        response.stubs(:body).returns('{"api":"compliance"}')
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({})
      end
    end

    describe 'when the returned body has an empty version key' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('200')
        response.stubs(:body).returns('{"api":"compliance","version":""}')
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({})
      end
    end

    describe 'when the returned body has a proper version' do
      it 'should return an empty hash' do
        response = mock
        response.stubs(:code).returns('200')
        response.stubs(:body).returns('{"api":"compliance","version":"1.2.3"}')
        InspecPlugins::Compliance::HTTP.expects(:get).with('myserver/version', 'test-headers', true).returns(response)
        InspecPlugins::Compliance::API.version(config).must_equal({ 'version' => '1.2.3', 'api' => 'compliance' })
      end
    end
  end

  describe 'automate/compliance is? checks' do
    describe 'when the config has a compliance server_type' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'compliance'
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate2_server?(config).must_equal false
      end
    end

    describe 'when the config has a automate2 server_type' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'automate2'
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate2_server?(config).must_equal true
      end
    end

    describe 'when the config has an automate server_type and no version key' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'automate'
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate2_server?(config).must_equal false
      end
    end

    describe 'when the config has an automate server_type and a version key that is not a hash' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'automate'
        config['version'] = '1.2.3'
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate2_server?(config).must_equal false
      end
    end

    describe 'when the config has an automate server_type and a version hash with no version' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'automate'
        config['version'] = {}
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal false
      end
    end

    describe 'when the config has an automate server_type and a version hash with a version' do
      it 'automate/compliance server is? methods return correctly' do
        config = InspecPlugins::Compliance::Configuration.new
        config.clean
        config['server_type'] = 'automate'
        config['version'] = { 'version' => '0.8.1' }
        InspecPlugins::Compliance::API.is_compliance_server?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server?(config).must_equal true
        InspecPlugins::Compliance::API.is_automate_server_pre_080?(config).must_equal false
        InspecPlugins::Compliance::API.is_automate_server_080_and_later?(config).must_equal true
      end
    end
  end

  describe '.server_version_from_config' do
    it 'returns nil when the config has no version key' do
      config = {}
      InspecPlugins::Compliance::API.server_version_from_config(config).must_be_nil
    end

    it 'returns nil when the version value is not a hash' do
      config = { 'version' => '123' }
      InspecPlugins::Compliance::API.server_version_from_config(config).must_be_nil
    end

    it 'returns nil when the version value is a hash but has no version key inside' do
      config = { 'version' => {} }
      InspecPlugins::Compliance::API.server_version_from_config(config).must_be_nil
    end

    it 'returns the version if the version value is a hash containing a version' do
      config = { 'version' => { 'version' => '1.2.3' } }
      InspecPlugins::Compliance::API.server_version_from_config(config).must_equal '1.2.3'
    end
  end

  describe 'profile_split' do
    it 'handles a profile without version' do
      InspecPlugins::Compliance::API.profile_split('admin/apache-baseline').must_equal ['admin', 'apache-baseline', nil]
    end

    it 'handles a profile with a version' do
      InspecPlugins::Compliance::API.profile_split('admin/apache-baseline#2.0.1').must_equal ['admin', 'apache-baseline', '2.0.1']
    end
  end

  describe 'target_url' do
    it 'handles a automate profile with and without version' do
      config = InspecPlugins::Compliance::Configuration.new
      config.clean
      config['server_type'] = 'automate'
      config['server'] = 'https://myautomate'
      config['version'] = '1.6.99'
      InspecPlugins::Compliance::API.target_url(config, 'admin/apache-baseline').must_equal       'https://myautomate/profiles/admin/apache-baseline/tar'
      InspecPlugins::Compliance::API.target_url(config, 'admin/apache-baseline#2.0.2').must_equal 'https://myautomate/profiles/admin/apache-baseline/version/2.0.2/tar'
    end

    it 'handles a chef-compliance profile with and without version' do
      config = InspecPlugins::Compliance::Configuration.new
      config.clean
      config['server_type'] = 'compliance'
      config['server'] = 'https://mychefcompliance'
      config['version'] = '1.1.2'
      InspecPlugins::Compliance::API.target_url(config, 'admin/apache-baseline').must_equal       'https://mychefcompliance/owners/admin/compliance/apache-baseline/tar'
      InspecPlugins::Compliance::API.target_url(config, 'admin/apache-baseline#2.0.2').must_equal 'https://mychefcompliance/owners/admin/compliance/apache-baseline/tar'
    end
  end

  describe 'exist?' do
    it 'works with profiles returned by Automate' do
      # ruby 2.3.3 has issues running stub_requests properly
      # skipping for that specific version
      return if RUBY_VERSION = '2.3.3'

      config = InspecPlugins::Compliance::Configuration.new
      config.clean
      config['owner'] = 'admin'
      config['server_type'] = 'automate'
      config['server'] = 'https://myautomate'
      config['version'] = '1.6.99'
      config['automate'] = { 'ent'=>'automate', 'token_type'=>'dctoken' }
      config['version'] = { 'api'=> 'compliance', 'version'=>'0.8.24' }
      
      stub_request(:get, 'https://myautomate/profiles/admin')
        .with(headers: { 'Accept'=>'*/*', 'Accept-Encoding'=>'gzip;q=1.0,deflate;q=0.6,identity;q=0.3', 'Chef-Delivery-Enterprise'=>'automate', 'User-Agent'=>'Ruby', 'X-Data-Collector-Token'=>'' })
        .to_return(status: 200, body: profiles_response.to_json, headers: {})

      InspecPlugins::Compliance::API.exist?(config, 'admin/apache-baseline').must_equal true
      InspecPlugins::Compliance::API.exist?(config, 'admin/apache-baseline#2.0.1').must_equal true
      InspecPlugins::Compliance::API.exist?(config, 'admin/apache-baseline#2.0.999').must_equal false
      InspecPlugins::Compliance::API.exist?(config, 'admin/missing-in-action').must_equal false
    end
  end

  describe '.determine_server_type' do
    let(:url) { 'https://someserver.onthe.net/' }

    let(:compliance_endpoint) { '/api/version' }
    let(:automate_endpoint) { '/compliance/version' }
    let(:automate2_endpoint) { '/dex/auth' }
    let(:headers) { nil }
    let(:insecure) { true }

    let(:good_response) { mock }
    let(:bad_response) { mock }

    it 'returns `:automate2` when a 400 is received from `https://URL/dex/auth`' do
      good_response.stubs(:code).returns('400')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(good_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_equal(:automate2)
    end

    it 'returns `:automate` when a 401 is received from `https://URL/compliance/version`' do
      good_response.stubs(:code).returns('401')
      bad_response.stubs(:code).returns('404')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate_endpoint, headers, insecure)
                      .returns(good_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_equal(:automate)
    end

    # Chef Automate currently returns 401 for `/compliance/version` but some
    # versions of OpsWorks Chef Automate return 200 and a Chef Manage page when
    # unauthenticated requests are received.
    it 'returns `:automate` when a 200 is received from `https://URL/compliance/version`' do
      bad_response.stubs(:code).returns('404')
      good_response.stubs(:code).returns('200')
      good_response.stubs(:body).returns('Are You Looking For the Chef Server?')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate_endpoint, headers, insecure)
                      .returns(good_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_equal(:automate)
    end

    it 'returns `nil` if a 200 is received from `https://URL/compliance/version` but not redirected to Chef Manage' do
      bad_response.stubs(:code).returns('200')
      bad_response.stubs(:body).returns('No Chef Manage here')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(bad_response)

      mock_compliance_response = mock
      mock_compliance_response.stubs(:code).returns('404')
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + compliance_endpoint, headers, insecure)
                      .returns(mock_compliance_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_be_nil
    end

    it 'returns `:compliance` when a 200 is received from `https://URL/api/version`' do
      good_response.stubs(:code).returns('200')
      bad_response.stubs(:code).returns('404')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + compliance_endpoint, headers, insecure)
                      .returns(good_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_equal(:compliance)
    end

    it 'returns `nil` if it cannot determine the server type' do
      bad_response.stubs(:code).returns('404')

      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate2_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + automate_endpoint, headers, insecure)
                      .returns(bad_response)
      InspecPlugins::Compliance::HTTP.expects(:get)
                      .with(url + compliance_endpoint, headers, insecure)
                      .returns(bad_response)

      InspecPlugins::Compliance::API.determine_server_type(url, insecure).must_be_nil
    end
  end
end