require 'test_helper'
require 'net/http'
require 'net/https'
require 'uri'

class GravatarifySubdomainTest < Test::Unit::TestCase
  include Gravatarify::Base
  
  def setup; reset_gravatarify! end
    
  context "changing hosts through Gravatarify#subdomains" do
    should "override default subdomains (useful to e.g. switch back to 'www' only)" do
      Gravatarify.subdomains = ['0', '1']
      assert_equal "http://0.gravatar.com/avatar/4979dd9653e759c78a81d4997f56bae2.jpg", gravatar_url('info@initech.com')
      assert_equal "http://1.gravatar.com/avatar/d4489907918035d0bc6ff3f6c76e760d.jpg", gravatar_url('support@initech.com')
    end
    
    should "take in a string only argument, like www" do
      Gravatarify.subdomains = 'www'
      assert_equal "http://www.gravatar.com/avatar/4979dd9653e759c78a81d4997f56bae2.jpg", gravatar_url('info@initech.com')
      assert_equal "http://www.gravatar.com/avatar/d4489907918035d0bc6ff3f6c76e760d.jpg", gravatar_url('support@initech.com')      
    end
    
    should "still work as expected if passed in `nil` and return urls with default subdomain `www`" do
      Gravatarify.subdomains = nil
      assert_equal "http://www.gravatar.com/avatar/4979dd9653e759c78a81d4997f56bae2.jpg", gravatar_url('info@initech.com')
      assert_equal "http://www.gravatar.com/avatar/d4489907918035d0bc6ff3f6c76e760d.jpg", gravatar_url('support@initech.com')            
    end
  end
    
  context "with Net::HTTP the gravatar.com subdomains" do
    should "return an image of type image/jpeg" do
      %w{ 0 1 2 www }.each do |subdomain|
        response = Net::HTTP.get_response URI.parse("http://#{subdomain}.gravatar.com/avatar/4979dd9653e759c78a81d4997f56bae2.jpg")
        assert_equal 200, response.code.to_i
        assert_equal "image/jpeg", response.content_type
      end
    end
    
    should "not respond to 3.gravatar.com, if so add to subdomains dude!!!" do
      assert_raises(SocketError) { Net::HTTP.get_response URI.parse('http://3.gravatar.com/avatar/4979dd9653e759c78a81d4997f56bae2.jpg') }
    end
    
    should "respond to https://secure.gravatar.com/ urls as well" do
      http = Net::HTTP.new('secure.gravatar.com', 443)
      http.use_ssl = true
      
      # do not verify peer certificate (just get rid of that warning!)
      http.instance_variable_get('@ssl_context').verify_mode = OpenSSL::SSL::VERIFY_NONE
      
      response = http.get '/avatar/4979dd9653e759c78a81d4997f56bae2.jpg'
      assert_equal 200, response.code.to_i
      assert_equal "image/jpeg", response.content_type
    end
  end
end