#
# Scalyr Output Plugin for Fluentd
#
# Copyright (C) 2015 Scalyr, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


require 'helper'
require 'flexmock/test_unit'

class HandleResponseTest < Scalyr::ScalyrOutTest

  def test_handle_response_missing_status
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "message":"An invalid message" }'  )
    exception = assert_raise( Scalyr::ServerError, "Server error not raised for missing status" ) {
      d.instance.handle_response( response )
    }

    assert_equal( "JSON response does not contain status message", exception.message )
  end

  def test_handle_response_discard_buffer
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "message":"An invalid message", "status":"error/server/discardBuffer" }'  )
    logger = flexmock( $log )
    logger.should_receive( :warn ).with( /buffer dropped/i )
    assert_nothing_raised( Scalyr::ServerError, Scalyr::ClientError, "Nothing should be raised when discarding the buffer" ) {
      d.instance.handle_response( response )
    }

  end

  def test_handle_response_unknown_error
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "message":"An invalid message", "status":"error/other" }'  )
    exception = assert_raise( Scalyr::ServerError, "Server error not raised for error status" ) {
      d.instance.handle_response( response )
    }
    assert_equal( "error/other", exception.message )
  end

  def test_handle_response_client_error
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "message":"An invalid message", "status":"error/client/test" }'  )
    exception = assert_raise( Scalyr::ClientError, "Client  error not raised for error status" ) {
      d.instance.handle_response( response )
    }
    assert_equal( "error/client/test", exception.message )
  end

  def test_handle_response_server_error
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "message":"An invalid message", "status":"error/server/test" }'  )
    exception = assert_raise( Scalyr::ServerError, "Server error not raised for error status" ) {
      d.instance.handle_response( response )
    }
    assert_equal( "error/server/test", exception.message )
  end

  def test_handle_response_code_4xx
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '404', :body =>'{ "status":"error/server/fileNotFound" }'  )
    exception = assert_raise( Scalyr::Client4xxError, "No 4xx exception raised" ) {
      d.instance.handle_response( response )
    }
    assert_equal( "error/server/fileNotFound", exception.message )
  end

  def test_handle_response_code_200
    d = create_driver
    response = flexmock( Net::HTTPResponse, :code => '200', :body =>'{ "status":"success" }'  )
    exception = assert_nothing_raised( Scalyr::ServerError, Scalyr::ClientError, "Error raised on success" ) {
      d.instance.handle_response( response )
    }
  end

end