require 'spec_helper'

describe ApiAuth::RequestDrivers::RestClientRequest do

  let(:timestamp){ Time.now.utc.httpdate }

  let(:request_path){ "/resource.xml?foo=bar&bar=foo" }

  let(:request_headers){
    {
      'Authorization'  => 'APIAuth 1044:12345',
      'Content-MD5' => '1B2M2Y8AsgTpgAmY7PhCfg==',
      'Content-Type' => 'text/plain',
      'Date' => timestamp
    }
  }

  let(:request) do
    RestClient::Request.new(
      :url => "/resource.xml?foo=bar&bar=foo",
      :headers => request_headers,
      :method => :put,
      :payload => "hello\nworld"
    )
  end

  subject(:driven_request){ ApiAuth::RequestDrivers::RestClientRequest.new(request) }

  describe "getting headers correctly" do
    it "gets the content_type" do
      expect(driven_request.content_type).to eq('text/plain')
    end

    it "gets the content_md5" do
      expect(driven_request.content_md5).to eq('1B2M2Y8AsgTpgAmY7PhCfg==')
    end

    it "gets the request_uri" do
      expect(driven_request.request_uri).to eq('/resource.xml?foo=bar&bar=foo')
    end

    it "gets the timestamp" do
      expect(driven_request.timestamp).to eq(timestamp)
    end

    it "gets the authorization_header" do
      expect(driven_request.authorization_header).to eq('APIAuth 1044:12345')
    end

    describe "#calculated_md5" do
      it "calculates md5 from the body" do
        expect(driven_request.calculated_md5).to eq('kZXQvrKoieG+Be1rsZVINw==')
      end

      it "treats no body as empty string" do
        request = RestClient::Request.new(
          :url => "/resource.xml?foo=bar&bar=foo",
          :headers => request_headers,
          :method => :put
        )
        driven_request = ApiAuth::RequestDrivers::RestClientRequest.new(request)
        expect(driven_request.calculated_md5).to eq('1B2M2Y8AsgTpgAmY7PhCfg==')
      end
    end

    describe "http_method" do
      context "when put request" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :put
          )
        end

        it "returns upcased put" do
          expect(driven_request.http_method).to eq('PUT')
        end
      end

      context "when get request" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :get
          )
        end

        it "returns upcased get" do
          expect(driven_request.http_method).to eq('GET')
        end
      end
    end
  end

  describe "setting headers correctly" do
    let(:request_headers){
      {
        'Content-Type' => 'text/plain'
      }
    }

    describe "#populate_content_md5" do
      context "when getting" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :get
          )
        end

        it "doesn't populate content-md5" do
          driven_request.populate_content_md5
          expect(request.headers["Content-MD5"]).to be_nil
        end
      end

      context "when posting" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :post,
            :payload => "hello\nworld"
          )
        end

        it "populates content-md5" do
          driven_request.populate_content_md5
          expect(request.headers["Content-MD5"]).to eq('kZXQvrKoieG+Be1rsZVINw==')
        end

        it "refreshes the cached headers" do
          driven_request.populate_content_md5
          expect(driven_request.content_md5).to eq('kZXQvrKoieG+Be1rsZVINw==')
        end
      end

      context "when putting" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :put,
            :payload => "hello\nworld"
          )
        end

        it "populates content-md5" do
          driven_request.populate_content_md5
          expect(request.headers["Content-MD5"]).to eq('kZXQvrKoieG+Be1rsZVINw==')
        end

        it "refreshes the cached headers" do
          driven_request.populate_content_md5
          expect(driven_request.content_md5).to eq('kZXQvrKoieG+Be1rsZVINw==')
        end
      end

      context "when deleting" do
        let(:request) do
          RestClient::Request.new(
            :url => "/resource.xml?foo=bar&bar=foo",
            :headers => request_headers,
            :method => :delete
          )
        end

        it "doesn't populate content-md5" do
          driven_request.populate_content_md5
          expect(request.headers["Content-MD5"]).to be_nil
        end
      end

    end

    describe "#set_date" do
      before do
        allow(Time).to receive_message_chain(:now, :utc, :httpdate).and_return(timestamp)
      end

      it "sets the date header of the request" do
        allow(Time).to receive_message_chain(:now, :utc, :httpdate).and_return(timestamp)
        driven_request.set_date
        expect(request.headers['DATE']).to eq(timestamp)
      end

      it "refreshes the cached headers" do
        driven_request.set_date
        expect(driven_request.timestamp).to eq(timestamp)
      end
    end

    describe "#set_auth_header" do
      it "sets the auth header" do
        driven_request.set_auth_header('APIAuth 1044:54321')
        expect(request.headers['Authorization']).to eq('APIAuth 1044:54321')
      end
    end
  end

  describe "md5_mismatch?" do

    context "when getting" do
      let(:request) do
        RestClient::Request.new(
          :url => "/resource.xml?foo=bar&bar=foo",
          :headers => request_headers,
          :method => :get
        )
      end

      it "is false" do
        expect(driven_request.md5_mismatch?).to be false
      end
    end

    context "when posting" do
      let(:request) do
        RestClient::Request.new(
          :url => "/resource.xml?foo=bar&bar=foo",
          :headers => request_headers,
          :method => :post,
          :payload => "hello\nworld"
        )
      end

      context "when calculated matches sent" do
        let(:request_headers){
          {
            'Authorization'  => 'APIAuth 1044:12345',
            'Content-MD5' => 'kZXQvrKoieG+Be1rsZVINw==',
            'Content-Type' => 'text/plain',
            'Date' => timestamp
          }
        }

        it "is false" do
          expect(driven_request.md5_mismatch?).to be false
        end
      end

      context "when calculated doesn't match sent" do
        let(:request_headers){
          {
            'Authorization'  => 'APIAuth 1044:12345',
            'Content-MD5' => '3',
            'Content-Type' => 'text/plain',
            'Date' => timestamp
          }
        }

        it "is true" do
          expect(driven_request.md5_mismatch?).to be true
        end
      end
    end

    context "when putting" do
      let(:request) do
        RestClient::Request.new(
          :url => "/resource.xml?foo=bar&bar=foo",
          :headers => request_headers,
          :method => :put,
          :payload => "hello\nworld"
        )
      end

      context "when calculated matches sent" do
        let(:request_headers){
          {
            'Authorization'  => 'APIAuth 1044:12345',
            'Content-MD5' => 'kZXQvrKoieG+Be1rsZVINw==',
            'Content-Type' => 'text/plain',
            'Date' => timestamp
          }
        }

        it "is false" do
          expect(driven_request.md5_mismatch?).to be false
        end
      end

      context "when calculated doesn't match sent" do
        let(:request_headers){
          {
            'Authorization'  => 'APIAuth 1044:12345',
            'Content-MD5' => '3',
            'Content-Type' => 'text/plain',
            'Date' => timestamp
          }
        }

        it "is true" do
          expect(driven_request.md5_mismatch?).to be true
        end
      end
    end

    context "when deleting" do
      let(:request) do
        RestClient::Request.new(
          :url => "/resource.xml?foo=bar&bar=foo",
          :headers => request_headers,
          :method => :delete
        )
      end

      it "is false" do
        expect(driven_request.md5_mismatch?).to be false
      end
    end
  end

  describe "edge cases" do
    it "doesn't mess up symbol based headers" do
      headers = { 'Content-MD5' => "e59ff97941044f85df5297e1c302d260",
                  :content_type => "text/plain",
                  'Date' => "Mon, 23 Jan 1984 03:29:56 GMT" }
      request = RestClient::Request.new(:url => "/resource.xml?foo=bar&bar=foo",
        :headers => headers,
        :method => :put)
      headers = ApiAuth::Headers.new(request)
      ApiAuth.sign!(request, "some access id", "some secret key")
      expect(request.processed_headers).to have_key('Content-Type')
    end
  end
end