require 'spec_helper'

describe Grape::Middleware::Versioner::AcceptVersionHeader do
  let(:app) { lambda { |env| [200, env, env] } }
  subject { Grape::Middleware::Versioner::AcceptVersionHeader.new(app, @options || {}) }

  before do
    @options = {
      version_options: {
        using: :accept_version_header
      },
    }
  end

  context 'api.version' do
    before do
      @options[:versions] = ['v1']
    end

    it 'is set' do
      status, _, env = subject.call('HTTP_ACCEPT_VERSION' => 'v1')
      env['api.version'].should eql 'v1'
      status.should == 200
    end

    it 'is set if format provided' do
      status, _, env = subject.call('HTTP_ACCEPT_VERSION' => 'v1')
      env['api.version'].should eql 'v1'
      status.should == 200
    end

    it 'fails with 406 Not Acceptable if version is not supported' do
      expect {
        subject.call('HTTP_ACCEPT_VERSION' => 'v2').last
      }.to throw_symbol(
        :error,
        status: 406,
        headers: { 'X-Cascade' => 'pass' },
        message: 'The requested version is not supported.'
      )
    end
  end

  it 'succeeds if :strict is not set' do
    subject.call('HTTP_ACCEPT_VERSION' => '').first.should == 200
    subject.call({}).first.should == 200
  end

  it 'succeeds if :strict is set to false' do
    @options[:version_options][:strict] = false
    subject.call('HTTP_ACCEPT_VERSION' => '').first.should == 200
    subject.call({}).first.should == 200
  end

  context 'when :strict is set' do
    before do
      @options[:versions] = ['v1']
      @options[:version_options][:strict] = true
    end

    it 'fails with 406 Not Acceptable if header is not set' do
      expect {
        subject.call({}).last
      }.to throw_symbol(
        :error,
        status: 406,
        headers: { 'X-Cascade' => 'pass' },
        message: 'Accept-Version header must be set.'
      )
    end

    it 'fails with 406 Not Acceptable if header is empty' do
      expect {
        subject.call('HTTP_ACCEPT_VERSION' => '').last
      }.to throw_symbol(
        :error,
        status: 406,
        headers: { 'X-Cascade' => 'pass' },
        message: 'Accept-Version header must be set.'
      )
    end

    it 'succeeds if proper header is set' do
      subject.call('HTTP_ACCEPT_VERSION' => 'v1').first.should == 200
    end
  end

  context 'when :strict and :cascade=>false are set' do
    before do
      @options[:versions] = ['v1']
      @options[:version_options][:strict] = true
      @options[:version_options][:cascade] = false
    end

    it 'fails with 406 Not Acceptable if header is not set' do
      expect {
        subject.call({}).last
      }.to throw_symbol(
        :error,
        status: 406,
        headers: {},
        message: 'Accept-Version header must be set.'
      )
    end

    it 'fails with 406 Not Acceptable if header is empty' do
      expect {
        subject.call('HTTP_ACCEPT_VERSION' => '').last
      }.to throw_symbol(
        :error,
        status: 406,
        headers: {},
        message: 'Accept-Version header must be set.'
      )
    end

    it 'succeeds if proper header is set' do
      subject.call('HTTP_ACCEPT_VERSION' => 'v1').first.should == 200
    end
  end
end