require File.join(File.dirname(__FILE__), "test_helper")

class TestFakeWeb < Test::Unit::TestCase

  def test_register_uri
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => "example")
    assert FakeWeb.registered_uri?(:get, 'http://mock/test_example.txt')
  end

  def test_register_uri_with_wrong_number_of_arguments
    assert_raises ArgumentError do
      FakeWeb.register_uri("http://example.com")
    end
    assert_raises ArgumentError do
      FakeWeb.register_uri(:get, "http://example.com", "/example", :body => "example")
    end
  end

  def test_registered_uri_with_wrong_number_of_arguments
    assert_raises ArgumentError do
      FakeWeb.registered_uri?
    end
    assert_raises ArgumentError do
      FakeWeb.registered_uri?(:get, "http://example.com", "/example")
    end
  end

  def test_response_for_with_wrong_number_of_arguments
    assert_raises ArgumentError do
      FakeWeb.response_for
    end
    assert_raises ArgumentError do
      FakeWeb.response_for(:get, "http://example.com", "/example")
    end
  end

  def test_register_uri_without_domain_name
    assert_raises URI::InvalidURIError do
      FakeWeb.register_uri(:get, 'test_example2.txt', File.dirname(__FILE__) + '/fixtures/test_example.txt')
    end
  end

  def test_register_uri_with_port_and_check_with_port
    FakeWeb.register_uri(:get, 'http://example.com:3000/', :body => 'foo')
    assert FakeWeb.registered_uri?(:get, 'http://example.com:3000/')
  end

  def test_register_uri_with_port_and_check_without_port
    FakeWeb.register_uri(:get, 'http://example.com:3000/', :body => 'foo')
    assert !FakeWeb.registered_uri?(:get, 'http://example.com/')
  end

  def test_register_uri_with_default_port_for_http_and_check_without_port
    FakeWeb.register_uri(:get, 'http://example.com:80/', :body => 'foo')
    assert FakeWeb.registered_uri?(:get, 'http://example.com/')
  end

  def test_register_uri_with_default_port_for_https_and_check_without_port
    FakeWeb.register_uri(:get, 'https://example.com:443/', :body => 'foo')
    assert FakeWeb.registered_uri?(:get, 'https://example.com/')
  end

  def test_register_uri_with_no_port_for_http_and_check_with_default_port
    FakeWeb.register_uri(:get, 'http://example.com/', :body => 'foo')
    assert FakeWeb.registered_uri?(:get, 'http://example.com:80/')
  end

  def test_register_uri_with_no_port_for_https_and_check_with_default_port
    FakeWeb.register_uri(:get, 'https://example.com/', :body => 'foo')
    assert FakeWeb.registered_uri?(:get, 'https://example.com:443/')
  end

  def test_register_uri_with_no_port_for_https_and_check_with_443_on_http
    FakeWeb.register_uri(:get, 'https://example.com/', :body => 'foo')
    assert !FakeWeb.registered_uri?(:get, 'http://example.com:443/')
  end

  def test_register_uri_with_no_port_for_http_and_check_with_80_on_https
    FakeWeb.register_uri(:get, 'http://example.com/', :body => 'foo')
    assert !FakeWeb.registered_uri?(:get, 'https://example.com:80/')
  end

  def test_register_uri_for_any_method_explicitly
    FakeWeb.register_uri(:any, "http://example.com/rpc_endpoint", :body => "OK")
    assert FakeWeb.registered_uri?(:get, "http://example.com/rpc_endpoint")
    assert FakeWeb.registered_uri?(:post, "http://example.com/rpc_endpoint")
    assert FakeWeb.registered_uri?(:put, "http://example.com/rpc_endpoint")
    assert FakeWeb.registered_uri?(:delete, "http://example.com/rpc_endpoint")
    assert FakeWeb.registered_uri?(:any, "http://example.com/rpc_endpoint")
    capture_stderr do  # silence deprecation warning
      assert FakeWeb.registered_uri?("http://example.com/rpc_endpoint")
    end
  end

  def test_register_uri_for_get_method_only
    FakeWeb.register_uri(:get, "http://example.com/users", :body => "User list")
    assert FakeWeb.registered_uri?(:get, "http://example.com/users")
    assert !FakeWeb.registered_uri?(:post, "http://example.com/users")
    assert !FakeWeb.registered_uri?(:put, "http://example.com/users")
    assert !FakeWeb.registered_uri?(:delete, "http://example.com/users")
    assert !FakeWeb.registered_uri?(:any, "http://example.com/users")
    capture_stderr do  # silence deprecation warning
      assert !FakeWeb.registered_uri?("http://example.com/users")
    end
  end

  def test_response_for_with_registered_uri
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    assert_equal 'test example content', FakeWeb.response_for(:get, 'http://mock/test_example.txt').body
  end

  def test_response_for_with_unknown_uri
    assert_nil FakeWeb.response_for(:get, 'http://example.com/')
  end

  def test_response_for_with_put_method
    FakeWeb.register_uri(:put, "http://example.com", :body => "response")
    assert_equal 'response', FakeWeb.response_for(:put, "http://example.com").body
  end

  def test_response_for_with_any_method_explicitly
    FakeWeb.register_uri(:any, "http://example.com", :body => "response")
    assert_equal 'response', FakeWeb.response_for(:get, "http://example.com").body
    assert_equal 'response', FakeWeb.response_for(:any, "http://example.com").body
  end

  def test_content_for_registered_uri_with_port_and_request_with_port
    FakeWeb.register_uri(:get, 'http://example.com:3000/', :body => 'test example content')
    response = Net::HTTP.start('example.com', 3000) { |http| http.get('/') }
    assert_equal 'test example content', response.body
  end

  def test_content_for_registered_uri_with_default_port_for_http_and_request_without_port
    FakeWeb.register_uri(:get, 'http://example.com:80/', :body => 'test example content')
    response = Net::HTTP.start('example.com') { |http| http.get('/') }
    assert_equal 'test example content', response.body
  end

  def test_content_for_registered_uri_with_no_port_for_http_and_request_with_default_port
    FakeWeb.register_uri(:get, 'http://example.com/', :body => 'test example content')
    response = Net::HTTP.start('example.com', 80) { |http| http.get('/') }
    assert_equal 'test example content', response.body
  end

  def test_content_for_registered_uri_with_default_port_for_https_and_request_with_default_port
    FakeWeb.register_uri(:get, 'https://example.com:443/', :body => 'test example content')
    http = Net::HTTP.new('example.com', 443)
    http.use_ssl = true
    response = http.get('/')
    assert_equal 'test example content', response.body
  end

  def test_content_for_registered_uri_with_no_port_for_https_and_request_with_default_port
    FakeWeb.register_uri(:get, 'https://example.com/', :body => 'test example content')
    http = Net::HTTP.new('example.com', 443)
    http.use_ssl = true
    response = http.get('/')
    assert_equal 'test example content', response.body
  end

  def test_content_for_registered_uris_with_ports_on_same_domain_and_request_without_port
    FakeWeb.register_uri(:get, 'http://example.com:3000/', :body => 'port 3000')
    FakeWeb.register_uri(:get, 'http://example.com/', :body => 'port 80')
    response = Net::HTTP.start('example.com') { |http| http.get('/') }
    assert_equal 'port 80', response.body
  end

  def test_content_for_registered_uris_with_ports_on_same_domain_and_request_with_port
    FakeWeb.register_uri(:get, 'http://example.com:3000/', :body => 'port 3000')
    FakeWeb.register_uri(:get, 'http://example.com/', :body => 'port 80')
    response = Net::HTTP.start('example.com', 3000) { |http| http.get('/') }
    assert_equal 'port 3000', response.body
  end

  def test_content_for_registered_uri_with_get_method_only
    FakeWeb.allow_net_connect = false
    FakeWeb.register_uri(:get, "http://example.com/", :body => "test example content")
    http = Net::HTTP.new('example.com')
    assert_equal 'test example content', http.get('/').body
    assert_raises(FakeWeb::NetConnectNotAllowedError) { http.post('/', nil) }
    assert_raises(FakeWeb::NetConnectNotAllowedError) { http.put('/', nil) }
    assert_raises(FakeWeb::NetConnectNotAllowedError) { http.delete('/') }
  end

  def test_content_for_registered_uri_with_any_method_explicitly
    FakeWeb.allow_net_connect = false
    FakeWeb.register_uri(:any, "http://example.com/", :body => "test example content")
    http = Net::HTTP.new('example.com')
    assert_equal 'test example content', http.get('/').body
    assert_equal 'test example content', http.post('/', nil).body
    assert_equal 'test example content', http.put('/', nil).body
    assert_equal 'test example content', http.delete('/').body
  end

  def test_content_for_registered_uri_with_any_method_implicitly
    FakeWeb.allow_net_connect = false
    capture_stderr do  # silence deprecation warning
      FakeWeb.register_uri("http://example.com/", :body => "test example content")
    end

    http = Net::HTTP.new('example.com')
    assert_equal 'test example content', http.get('/').body
    assert_equal 'test example content', http.post('/', nil).body
    assert_equal 'test example content', http.put('/', nil).body
    assert_equal 'test example content', http.delete('/').body
  end

  def test_mock_request_with_block
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    response = Net::HTTP.start('mock') { |http| http.get('/test_example.txt') }
    assert_equal 'test example content', response.body
  end

  def test_request_with_registered_body_yields_the_response_body_to_a_request_block
    FakeWeb.register_uri(:get, "http://example.com", :body => "content")
    body = nil
    Net::HTTP.start("example.com") do |http|
      http.get("/") do |response_body|
        body = response_body
      end
    end
    assert_equal "content", body
  end

  def test_request_with_registered_response_yields_the_response_body_to_a_request_block
    fake_response = Net::HTTPOK.new('1.1', '200', 'OK')
    fake_response.instance_variable_set(:@body, "content")
    FakeWeb.register_uri(:get, 'http://example.com', :response => fake_response)
    body = nil
    Net::HTTP.start("example.com") do |http|
      http.get("/") do |response_body|
        body = response_body
      end
    end
    assert_equal "content", body
  end

  def test_mock_request_with_undocumented_full_uri_argument_style
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    response = Net::HTTP.start('mock') { |query| query.get('http://mock/test_example.txt') }
    assert_equal 'test example content', response.body
  end

  def test_mock_request_with_undocumented_full_uri_argument_style_and_query
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt?a=b', :body => 'test query content')
    response = Net::HTTP.start('mock') { |query| query.get('http://mock/test_example.txt?a=b') }
    assert_equal 'test query content', response.body
  end

  def test_mock_post
    FakeWeb.register_uri(:post, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    response = Net::HTTP.start('mock') { |query| query.post('/test_example.txt', '') }
    assert_equal 'test example content', response.body
  end

  def test_mock_post_with_string_as_registered_uri
    FakeWeb.register_uri(:post, 'http://mock/test_string.txt', :body => 'foo')
    response = Net::HTTP.start('mock') { |query| query.post('/test_string.txt', '') }
    assert_equal 'foo', response.body
  end

  def test_mock_get_with_request_as_registered_uri
    fake_response = Net::HTTPOK.new('1.1', '200', 'OK')
    FakeWeb.register_uri(:get, 'http://mock/test_response', :response => fake_response)
    response = Net::HTTP.start('mock') { |query| query.get('/test_response') }
    assert_equal fake_response, response
  end

  def test_mock_get_with_request_from_file_as_registered_uri
    FakeWeb.register_uri(:get, 'http://www.google.com/', :response => File.dirname(__FILE__) + '/fixtures/google_response_without_transfer_encoding')
    response = Net::HTTP.start('www.google.com') { |query| query.get('/') }
    assert_equal '200', response.code
    assert response.body.include?('<title>Google</title>')
  end

  def test_mock_post_with_request_from_file_as_registered_uri
    FakeWeb.register_uri(:post, 'http://www.google.com/', :response => File.dirname(__FILE__) + '/fixtures/google_response_without_transfer_encoding')
    response = Net::HTTP.start('www.google.com') { |query| query.post('/', '') }
    assert_equal "200", response.code
    assert response.body.include?('<title>Google</title>')
  end

  def test_proxy_request
    FakeWeb.register_uri(:get, 'http://www.example.com/', :body => "hello world")
    FakeWeb.register_uri(:get, 'http://your.proxy.host/', :body => "lala")

    response = nil
    Net::HTTP::Proxy('your.proxy.host', 8080).start('www.example.com') do |http|
      response = http.get('/')
    end
    assert_equal "hello world", response.body
  end

  def test_https_request
    FakeWeb.register_uri(:get, 'https://www.example.com/', :body => "Hello World")
    http = Net::HTTP.new('www.example.com', 443)
    http.use_ssl = true
    response = http.get('/')
    assert_equal "Hello World", response.body
  end

  def test_register_unimplemented_response
    FakeWeb.register_uri(:get, 'http://mock/unimplemented', :response => 1)
    assert_raises StandardError do
      Net::HTTP.start('mock') { |q| q.get('/unimplemented') }
    end
  end

  def test_real_http_request
    FakeWeb.allow_net_connect = true
    setup_expectations_for_real_apple_hot_news_request

    resp = nil
    Net::HTTP.start('images.apple.com') do |query|
      resp = query.get('/main/rss/hotnews/hotnews.rss')
    end
    assert resp.body.include?('Apple')
    assert resp.body.include?('News')
  end

  def test_real_http_request_with_undocumented_full_uri_argument_style
    FakeWeb.allow_net_connect = true
    setup_expectations_for_real_apple_hot_news_request(:path => 'http://images.apple.com/main/rss/hotnews/hotnews.rss')

    resp = nil
    Net::HTTP.start('images.apple.com') do |query|
      resp = query.get('http://images.apple.com/main/rss/hotnews/hotnews.rss')
    end
    assert resp.body.include?('Apple')
    assert resp.body.include?('News')
  end

  def test_real_https_request
    FakeWeb.allow_net_connect = true
    setup_expectations_for_real_apple_hot_news_request(:port => 443)

    http = Net::HTTP.new('images.apple.com', 443)
    http.use_ssl = true
    http.verify_mode = OpenSSL::SSL::VERIFY_NONE  # silence certificate warning
    response = http.get('/main/rss/hotnews/hotnews.rss')
    assert response.body.include?('Apple')
    assert response.body.include?('News')
  end

  def test_real_request_on_same_domain_as_mock
    FakeWeb.allow_net_connect = true
    setup_expectations_for_real_apple_hot_news_request

    FakeWeb.register_uri(:get, 'http://images.apple.com/test_string.txt', :body => 'foo')

    resp = nil
    Net::HTTP.start('images.apple.com') do |query|
      resp = query.get('/main/rss/hotnews/hotnews.rss')
    end
    assert resp.body.include?('Apple')
    assert resp.body.include?('News')
  end

  def test_mock_request_on_real_domain
    FakeWeb.register_uri(:get, 'http://images.apple.com/test_string.txt', :body => 'foo')
    resp = nil
    Net::HTTP.start('images.apple.com') do |query|
      resp = query.get('/test_string.txt')
    end
    assert_equal 'foo', resp.body
  end

  def test_mock_post_that_raises_exception
    FakeWeb.register_uri(:post, 'http://mock/raising_exception.txt', :exception => StandardError)
    assert_raises(StandardError) do
      Net::HTTP.start('mock') do |query|
        query.post('/raising_exception.txt', 'some data')
      end
    end
  end

  def test_mock_post_that_raises_an_http_error
    FakeWeb.register_uri(:post, 'http://mock/raising_exception.txt', :exception => Net::HTTPError)
    assert_raises(Net::HTTPError) do
      Net::HTTP.start('mock') do |query|
        query.post('/raising_exception.txt', '')
      end
    end
  end

  def test_raising_an_exception_that_requires_an_argument_to_instantiate
    FakeWeb.register_uri(:get, "http://example.com/timeout.txt", :exception => Timeout::Error)
    assert_raises(Timeout::Error) do
      Net::HTTP.get(URI.parse("http://example.com/timeout.txt"))
    end
  end

  def test_mock_instance_syntax
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    response = nil
    uri = URI.parse('http://mock/test_example.txt')
    http = Net::HTTP.new(uri.host, uri.port)
    response = http.start do
      http.get(uri.path)
    end

    assert_equal 'test example content', response.body
  end

  def test_mock_via_nil_proxy
    response = nil
    proxy_address = nil
    proxy_port = nil
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    uri = URI.parse('http://mock/test_example.txt')
    http = Net::HTTP::Proxy(proxy_address, proxy_port).new(
              uri.host, (uri.port or 80))
    response = http.start do
      http.get(uri.path)
    end

    assert_equal 'test example content', response.body
  end

  def test_response_type
    FakeWeb.register_uri(:get, 'http://mock/test_example.txt', :body => "test")
    response = Net::HTTP.start('mock') { |http| http.get('/test_example.txt') }
    assert_kind_of Net::HTTPSuccess, response
  end

  def test_mock_request_that_raises_an_http_error_with_a_specific_status
    FakeWeb.register_uri(:get, 'http://mock/raising_exception.txt', :exception => Net::HTTPError, :status => ['404', 'Not Found'])
    exception = assert_raises(Net::HTTPError) do
      Net::HTTP.start('mock') { |http| http.get('/raising_exception.txt') }
    end
    assert_equal '404', exception.response.code
    assert_equal 'Not Found', exception.response.msg
  end

  def test_mock_rotate_responses
    FakeWeb.register_uri(:get, 'http://mock/multiple_test_example.txt',
                         [ {:body => File.dirname(__FILE__) + '/fixtures/test_example.txt', :times => 2},
                           {:body => "thrice", :times => 3},
                           {:body => "ever_more"} ])

    uri = URI.parse('http://mock/multiple_test_example.txt')
    2.times { assert_equal 'test example content', Net::HTTP.get(uri) }
    3.times { assert_equal 'thrice',               Net::HTTP.get(uri) }
    4.times { assert_equal 'ever_more',            Net::HTTP.get(uri) }
  end

  def test_mock_request_using_response_with_transfer_encoding_header_has_valid_transfer_encoding_header
    FakeWeb.register_uri(:get, 'http://www.google.com/', :response => File.dirname(__FILE__) + '/fixtures/google_response_with_transfer_encoding')
    response = Net::HTTP.start('www.google.com') { |query| query.get('/') }
    assert_not_nil response['transfer-encoding']
    assert response['transfer-encoding'] == 'chunked'
  end

  def test_mock_request_using_response_without_transfer_encoding_header_does_not_have_a_transfer_encoding_header
    FakeWeb.register_uri(:get, 'http://www.google.com/', :response => File.dirname(__FILE__) + '/fixtures/google_response_without_transfer_encoding')
    response = nil
    response = Net::HTTP.start('www.google.com') { |query| query.get('/') }
    assert !response.key?('transfer-encoding')
  end

  def test_mock_request_using_response_from_curl_has_original_transfer_encoding_header
    FakeWeb.register_uri(:get, 'http://www.google.com/', :response => File.dirname(__FILE__) + '/fixtures/google_response_from_curl')
    response = Net::HTTP.start('www.google.com') { |query| query.get('/') }
    assert_not_nil response['transfer-encoding']
    assert response['transfer-encoding'] == 'chunked'
  end

  def test_txt_file_should_have_three_lines
    FakeWeb.register_uri(:get, 'http://www.google.com/', :body => File.dirname(__FILE__) + '/fixtures/test_txt_file')
    response = Net::HTTP.start('www.google.com') { |query| query.get('/') }
    assert response.body.split(/\n/).size == 3, "response has #{response.body.split(/\n/).size} lines should have 3"
  end

  def test_requiring_fakeweb_instead_of_fake_web
    require "fakeweb"
  end

  def test_registering_with_string_containing_null_byte
    # Regression test for File.exists? raising an ArgumentError ("string
    # contains null byte") since :response first tries to find by filename.
    # The string should be treated as a response body, instead, and an
    # EOFError is raised when the byte is encountered.
    FakeWeb.register_uri(:get, "http://example.com", :response => "test\0test")
    assert_raise EOFError do
      Net::HTTP.get(URI.parse("http://example.com"))
    end

    FakeWeb.register_uri(:get, "http://example.com", :body => "test\0test")
    body = Net::HTTP.get(URI.parse("http://example.com"))
    assert_equal "test\0test", body
  end

  def test_registering_with_string_that_is_a_directory_name
    # Similar to above, but for Errno::EISDIR being raised since File.exists?
    # returns true for directories
    FakeWeb.register_uri(:get, "http://example.com", :response => File.dirname(__FILE__))
    assert_raise EOFError do
      body = Net::HTTP.get(URI.parse("http://example.com"))
    end

    FakeWeb.register_uri(:get, "http://example.com", :body => File.dirname(__FILE__))
    body = Net::HTTP.get(URI.parse("http://example.com"))
    assert_equal File.dirname(__FILE__), body
  end

  def test_http_version_from_string_response
    FakeWeb.register_uri(:get, "http://example.com", :body => "example")
    response = Net::HTTP.start("example.com") { |http| http.get("/") }
    assert_equal "1.0", response.http_version
  end

  def test_http_version_from_file_response
    FakeWeb.register_uri(:get, "http://example.com", :body => File.dirname(__FILE__) + '/fixtures/test_example.txt')
    response = Net::HTTP.start("example.com") { |http| http.get("/") }
    assert_equal "1.0", response.http_version
  end

end