require 'spec_helper'
require 'webmock/rspec'
require 'puppet/http'

describe Puppet::HTTP::Service::FileServer do
  include PuppetSpec::Files

  let(:ssl_context) { Puppet::SSL::SSLContext.new }
  let(:client) { Puppet::HTTP::Client.new(ssl_context: ssl_context) }
  let(:subject) { client.create_session.route_to(:fileserver) }
  let(:environment) { 'testing' }
  let(:report) { Puppet::Transaction::Report.new }

  before :each do
    Puppet[:server] = 'www.example.com'
    Puppet[:masterport] = 443
  end

  context 'when making requests' do
    let(:uri) {"https://www.example.com:443/puppet/v3/file_content/:mount/:path?environment=testing"}

    it 'includes default HTTP headers' do
      stub_request(:get, uri).with do |request|
        expect(request.headers).to include({'X-Puppet-Version' => /./, 'User-Agent' => /./})
        expect(request.headers).to_not include('X-Puppet-Profiling')
      end

      subject.get_file_content(path: '/:mount/:path', environment: environment) { |data| }
    end
  end

  context 'when routing to the file service' do
    it 'defaults the server and port based on settings' do
      Puppet[:server] = 'file.example.com'
      Puppet[:masterport] = 8141

      stub_request(:get, "https://file.example.com:8141/puppet/v3/file_content/:mount/:path?environment=testing")

      subject.get_file_content(path: '/:mount/:path', environment: environment) { |data| }
    end
  end

  context 'retrieving file metadata' do
    let(:path) { tmpfile('get_file_metadata') }
    let(:url) { "https://www.example.com/puppet/v3/file_metadata/:mount/#{path}?checksum_type=md5&environment=testing&links=manage&source_permissions=ignore" }
    let(:filemetadata) { Puppet::FileServing::Metadata.new(path) }
    let(:request_path) { "/:mount/#{path}"}

    it 'includes puppet headers set via the :http_extra_headers and :profile settings' do
      stub_request(:get, url).with(headers: {'Example-Header' => 'real-thing', 'another' => 'thing', 'X-Puppet-Profiling' => 'true'}).
        to_return(status: 200, body: filemetadata.render, headers: { 'Content-Type' => 'application/json' })

      Puppet[:http_extra_headers] = 'Example-Header:real-thing,another:thing'
      Puppet[:profile] = true

      subject.get_file_metadata(path: request_path, environment: environment)
    end

    it 'submits a request for file metadata to the server' do
      stub_request(:get, url).with(
        headers: {'Accept'=>'application/json, application/x-msgpack, text/pson',}
      ).to_return(
        status: 200, body: filemetadata.render, headers: { 'Content-Type' => 'application/json' }
      )

      _, metadata = subject.get_file_metadata(path: request_path, environment: environment)
      expect(metadata.path).to eq(path)
    end

    it 'returns the request response' do
      stub_request(:get, url).to_return(
        status: 200, body: filemetadata.render, headers: { 'Content-Type' => 'application/json' }
      )

      resp, _ = subject.get_file_metadata(path: request_path, environment: environment)
      expect(resp).to be_a(Puppet::HTTP::Response)
    end

    it 'raises a protocol error if the Content-Type header is missing from the response' do
      stub_request(:get, url).to_return(status: 200, body: '', headers: {})

      expect {
        subject.get_file_metadata(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::ProtocolError, "No content type in http response; cannot parse")
    end

    it 'raises an error if the Content-Type is unsupported' do
      stub_request(:get, url).to_return(status: 200, body: '', headers: { 'Content-Type' => 'text/yaml' })

      expect {
        subject.get_file_metadata(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::ProtocolError, "Content-Type is unsupported")
    end

    it 'raises response error if unsuccessful' do
      stub_request(:get, url).to_return(status: [400, 'Bad Request'])

      expect {
        subject.get_file_metadata(path: request_path, environment: environment)
      }.to raise_error do |err|
        expect(err).to be_an_instance_of(Puppet::HTTP::ResponseError)
        expect(err.message).to eq('Bad Request')
        expect(err.response.code).to eq(400)
      end
    end

    it 'raises a serialization error if serialization fails' do
      stub_request(:get, url).to_return(
        status: 200, body: '', headers: { 'Content-Type' => 'application/json' }
      )

      expect {
        subject.get_file_metadata(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::SerializationError, /Failed to deserialize Puppet::FileServing::Metadata from json/)
    end

    it 'raises response error if path is relative' do
      expect {
        subject.get_file_metadata(path: 'relative_path', environment: environment)
      }.to raise_error(ArgumentError, 'Path must start with a slash')
    end
  end

  context 'retrieving multiple file metadatas' do
    let(:path) { tmpfile('get_file_metadatas') }
    let(:url) { "https://www.example.com/puppet/v3/file_metadatas/:mount/#{path}?checksum_type=md5&links=manage&recurse=false&source_permissions=ignore&environment=testing" }
    let(:filemetadatas) { [Puppet::FileServing::Metadata.new(path)] }
    let(:formatter) { Puppet::Network::FormatHandler.format(:json) }
    let(:request_path) { "/:mount/#{path}"}

    it 'includes puppet headers set via the :http_extra_headers and :profile settings' do
      stub_request(:get, url).with(headers: {'Example-Header' => 'real-thing', 'another' => 'thing', 'X-Puppet-Profiling' => 'true'}).
        to_return(status: 200, body: formatter.render_multiple(filemetadatas), headers: { 'Content-Type' => 'application/json' })

      Puppet[:http_extra_headers] = 'Example-Header:real-thing,another:thing'
      Puppet[:profile] = true

      subject.get_file_metadatas(path: request_path, environment: environment)
    end

    it 'submits a request for file metadata to the server' do
      stub_request(:get, url).with(
        headers: {'Accept'=>'application/json, application/x-msgpack, text/pson',}
      ).to_return(
        status: 200, body: formatter.render_multiple(filemetadatas), headers: { 'Content-Type' => 'application/json' }
      )

      _, metadatas = subject.get_file_metadatas(path: request_path, environment: environment)
      expect(metadatas.first.path).to eq(path)
    end

    it 'returns the request response' do
      stub_request(:get, url).to_return(
        status: 200, body: formatter.render_multiple(filemetadatas), headers: { 'Content-Type' => 'application/json' }
      )

      resp, _ = subject.get_file_metadatas(path: request_path, environment: environment)
      expect(resp).to be_a(Puppet::HTTP::Response)
    end

    it 'automatically converts an array of parameters to the stringified query' do
      url = "https://www.example.com/puppet/v3/file_metadatas/:mount/#{path}?checksum_type=md5&environment=testing&ignore=CVS&ignore=.git&ignore=.hg&links=manage&recurse=false&source_permissions=ignore"
      stub_request(:get, url).with(
        headers: {'Accept'=>'application/json, application/x-msgpack, text/pson',}
      ).to_return(
        status: 200, body: formatter.render_multiple(filemetadatas), headers: { 'Content-Type' => 'application/json' }
      )

      _, metadatas = subject.get_file_metadatas(path: request_path, environment: environment, ignore: ['CVS', '.git', '.hg'])
      expect(metadatas.first.path).to eq(path)
    end

    it 'raises a protocol error if the Content-Type header is missing from the response' do
      stub_request(:get, url).to_return(status: 200, body: '', headers: {})

      expect {
        subject.get_file_metadatas(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::ProtocolError, "No content type in http response; cannot parse")
    end

    it 'raises an error if the Content-Type is unsupported' do
      stub_request(:get, url).to_return(status: 200, body: '', headers: { 'Content-Type' => 'text/yaml' })

      expect {
        subject.get_file_metadatas(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::ProtocolError, "Content-Type is unsupported")
    end

    it 'raises response error if unsuccessful' do
      stub_request(:get, url).to_return(status: [400, 'Bad Request'])

      expect {
        subject.get_file_metadatas(path: request_path, environment: environment)
      }.to raise_error do |err|
        expect(err).to be_an_instance_of(Puppet::HTTP::ResponseError)
        expect(err.message).to eq('Bad Request')
        expect(err.response.code).to eq(400)
      end
    end

    it 'raises a serialization error if serialization fails' do
      stub_request(:get, url).to_return(
        status: 200, body: '', headers: { 'Content-Type' => 'application/json' }
      )

      expect {
        subject.get_file_metadatas(path: request_path, environment: environment)
      }.to raise_error(Puppet::HTTP::SerializationError, /Failed to deserialize multiple Puppet::FileServing::Metadata from json/)
    end

    it 'raises response error if path is relative' do
      expect {
        subject.get_file_metadatas(path: 'relative_path', environment: environment)
      }.to raise_error(ArgumentError, 'Path must start with a slash')
    end
  end

  context 'getting file content' do
    let(:uri) {"https://www.example.com:443/puppet/v3/file_content/:mount/:path?environment=testing"}

    it 'includes puppet headers set via the :http_extra_headers and :profile settings' do
      stub_request(:get, uri).with(headers: {'Example-Header' => 'real-thing', 'another' => 'thing', 'X-Puppet-Profiling' => 'true'}).
        to_return(status: 200, body: "and beyond")

      Puppet[:http_extra_headers] = 'Example-Header:real-thing,another:thing'
      Puppet[:profile] = true

      expect { |b|
        subject.get_file_content(path: '/:mount/:path', environment: environment, &b)
      }.to yield_with_args("and beyond")
    end

    it 'yields file content' do
      stub_request(:get, uri).with do |request|
        expect(request.headers).to include({'Accept' => 'application/octet-stream'})
      end.to_return(status: 200, body: "and beyond")

      expect { |b|
        subject.get_file_content(path: '/:mount/:path', environment: environment, &b)
      }.to yield_with_args("and beyond")
    end

    it 'returns the request response' do
      stub_request(:get, uri)

      resp = subject.get_file_content(path: '/:mount/:path', environment: environment) { |b| b }
      expect(resp).to be_a(Puppet::HTTP::Response)
    end

    it 'raises response error if unsuccessful' do
      stub_request(:get, uri).to_return(status: [400, 'Bad Request'])

      expect {
        subject.get_file_content(path: '/:mount/:path', environment: environment) { |data| }
      }.to raise_error do |err|
        expect(err).to be_an_instance_of(Puppet::HTTP::ResponseError)
        expect(err.message).to eq('Bad Request')
        expect(err.response.code).to eq(400)
      end
    end

    it 'raises response error if path is relative' do
      expect {
        subject.get_file_content(path: 'relative_path', environment: environment) { |data| }
      }.to raise_error(ArgumentError, 'Path must start with a slash')
    end
  end

  context 'getting static file content' do
    let(:code_id) { "0fc72115-adc6-4b1a-aa50-8f31b3ece440" }
    let(:uri) { "https://www.example.com:443/puppet/v3/static_file_content/:mount/:path?environment=testing&code_id=#{code_id}"}

    it 'yields file content' do
      stub_request(:get, uri).with do |request|
        expect(request.headers).to include({'Accept' => 'application/octet-stream'})
      end.to_return(status: 200, body: "and beyond")

      expect { |b|
        subject.get_static_file_content(path: '/:mount/:path', environment: environment, code_id: code_id, &b)
      }.to yield_with_args("and beyond")
    end

    it 'returns the request response' do
      stub_request(:get, uri)

      resp = subject.get_static_file_content(path: '/:mount/:path', environment: environment, code_id: code_id) { |b| b }
      expect(resp).to be_a(Puppet::HTTP::Response)
    end

    it 'raises response error if unsuccessful' do
      stub_request(:get, uri).to_return(status: [400, 'Bad Request'])

      expect {
        subject.get_static_file_content(path: '/:mount/:path', environment: environment, code_id: code_id) { |data| }
      }.to raise_error do |err|
        expect(err).to be_an_instance_of(Puppet::HTTP::ResponseError)
        expect(err.message).to eq('Bad Request')
        expect(err.response.code).to eq(400)
      end
    end

    it 'raises response error if path is relative' do
      expect {
        subject.get_static_file_content(path: 'relative_path', environment: environment, code_id: code_id) { |data| }
      }.to raise_error(ArgumentError, 'Path must start with a slash')
    end
  end
end