curb 0.1.4 → 0.7.15

data/tests/tc_curl_easy.rb

@@ -1,8 +1,11 @@
-require File.join(File.dirname(__FILE__), 'helper')
+require File.expand_path(File.join(File.dirname(__FILE__), 'helper'))
+class FooNoToS 
+  undef to_s
+end
 
 class TestCurbCurlEasy < Test::Unit::TestCase
   def test_class_perform_01   
-    assert_instance_of Curl::Easy, c = Curl::Easy.perform($TEST_URL)    
+    assert_instance_of Curl::Easy, c = Curl::Easy.perform($TEST_URL)
     assert_match(/^# DO NOT REMOVE THIS COMMENT/, c.body_str)
     assert_equal "", c.header_str
   end    
@@ -22,6 +25,7 @@ class TestCurbCurlEasy < Test::Unit::TestCase
   
   def test_new_01
     c = Curl::Easy.new
+    assert_equal Curl::Easy, c.class
     assert_nil c.url
     assert_nil c.body_str
     assert_nil c.header_str
@@ -54,8 +58,26 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     assert_equal $TEST_URL, c.url
     assert_equal blk, c.on_body   # sets handler nil, returns old handler
     assert_equal nil, c.on_body
-  end    
-  
+  end
+
+  class Foo < Curl::Easy ; end
+  def test_new_05
+    # can use Curl::Easy as a base class
+    c = Foo.new
+    assert_equal Foo, c.class
+    c.url = $TEST_URL
+    c.perform
+    assert_match(/^# DO NOT REMOVE THIS COMMENT/, c.body_str)
+  end
+
+  # test invalid use of new
+  def test_new_06
+    Curl::Easy.new(TestServlet.url) do|curl|
+      curl.http_post
+      assert_equal "POST\n", curl.body_str
+    end
+  end
+
   def test_escape
     c = Curl::Easy.new
     
@@ -86,7 +108,7 @@ class TestCurbCurlEasy < Test::Unit::TestCase
 
   def test_get_01   
     c = Curl::Easy.new($TEST_URL)    
-    assert_equal true, c.http_get    
+    assert_equal true, c.http_get
     assert_match(/^# DO NOT REMOVE THIS COMMENT/, c.body_str)
     assert_equal "", c.header_str
   end    
@@ -110,7 +132,8 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     assert_equal "", c.body_str
     assert_equal "", c.header_str
   end    
-  
+
+
   def test_last_effective_url_01
     c = Curl::Easy.new($TEST_URL)
     
@@ -124,6 +147,15 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     
     assert_not_equal c.last_effective_url, c.url
   end
+
+  def test_http_get_block
+    curl = Curl::Easy.http_get(TestServlet.url) do|c|
+      c.follow_location = true
+      c.max_redirects = 3
+    end
+    assert_equal curl.url, curl.last_effective_url
+    assert_equal 'GET', curl.body_str
+  end
   
   def test_local_port_01
     c = Curl::Easy.new($TEST_URL)
@@ -192,7 +224,7 @@ class TestCurbCurlEasy < Test::Unit::TestCase
 
     assert_equal $TEST_URL, c.url
     assert_equal "http://some.proxy", c.proxy_url
-    
+ 
     c.proxy_url = nil
     assert_equal $TEST_URL, c.url
     assert_nil c.proxy_url
@@ -322,6 +354,30 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     assert_equal 30, c.dns_cache_timeout
   end
   
+  def test_low_speed_limit_01
+    c = Curl::Easy.new($TEST_URL)
+    
+    assert_nil c.low_speed_limit
+    
+    c.low_speed_limit = 3
+    assert_equal 3, c.low_speed_limit
+    
+    c.low_speed_limit = nil
+    assert_nil c.low_speed_limit
+  end
+  
+  def test_low_speed_time_01
+    c = Curl::Easy.new($TEST_URL)
+    
+    assert_nil c.low_speed_time
+    
+    c.low_speed_time = 3
+    assert_equal 3, c.low_speed_time
+    
+    c.low_speed_time = nil
+    assert_nil c.low_speed_time
+  end
+  
   def test_on_body
     blk = lambda { |i| i.length }
     
@@ -331,6 +387,25 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     assert_equal blk, c.on_body   # sets handler nil, returns old handler
     assert_equal nil, c.on_body
   end
+
+  def test_inspect_with_no_url
+    c = Curl::Easy.new
+    assert_equal '#<Curl::Easy>', c.inspect
+  end
+  
+  def test_inspect_with_short_url
+    c = Curl::Easy.new('http://www.google.com/')
+    assert_equal "#<Curl::Easy http://www.google.com/>", c.inspect
+  end
+  
+  def test_inspect_truncates_to_64_chars
+    base_url      = 'http://www.google.com/'
+    truncated_url = base_url + 'x' * (64 - '#<Curl::Easy >'.size - base_url.size)
+    long_url      = truncated_url + 'yyyy'
+    c = Curl::Easy.new(long_url)
+    assert_equal 64, c.inspect.size
+    assert_equal "#<Curl::Easy #{truncated_url}>", c.inspect
+  end
   
   def test_on_header
     blk = lambda { |i| i.length }
@@ -425,18 +500,398 @@ class TestCurbCurlEasy < Test::Unit::TestCase
     assert c.multipart_form_post?
   end
 
+  def test_ignore_content_length
+    c = Curl::Easy.new
+    assert !c.ignore_content_length?
+    assert c.ignore_content_length = true
+    assert c.ignore_content_length?
+  end
+
   def test_enable_cookies
     c = Curl::Easy.new
     assert !c.enable_cookies?
     assert c.enable_cookies = true
     assert c.enable_cookies?
   end
-  
+
+  def test_cookies_option
+    c = Curl::Easy.new
+    assert_nil c.cookies
+    assert_equal "name1=content1; name2=content2;", c.cookies = "name1=content1; name2=content2;"
+    assert_equal "name1=content1; name2=content2;", c.cookies
+  end
+
+  def test_cookiefile
+    c = Curl::Easy.new
+    assert_nil c.cookiefile
+    assert_equal "some.file", c.cookiefile = "some.file"
+    assert_equal "some.file", c.cookiefile       
+  end
+
   def test_cookiejar
     c = Curl::Easy.new
     assert_nil c.cookiejar
     assert_equal "some.file", c.cookiejar = "some.file"
     assert_equal "some.file", c.cookiejar        
   end
+
+  def test_on_success
+    curl = Curl::Easy.new($TEST_URL)    
+    on_success_called = false
+    curl.on_success {|c|
+      on_success_called = true
+      assert_not_nil c.body_str
+      assert_equal "", c.header_str
+      assert_match(/^# DO NOT REMOVE THIS COMMENT/, c.body_str)
+    }
+    curl.perform
+    assert on_success_called, "Success handler not called" 
+  end
+
+  def test_on_success_with_on_failure
+    curl = Curl::Easy.new("#{$TEST_URL.gsub(/file:\/\//,'')}/not_here")
+    on_failure_called = false
+    curl.on_success {|c| } # make sure we get the failure call even though this handler is defined
+    curl.on_failure {|c,code| on_failure_called = true }
+    curl.perform
+    assert on_failure_called, "Failure handler not called" 
+  end
+  
+  def test_get_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_get
+    assert_equal 'GET', curl.body_str
+  end
+  
+  def test_post_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_post([Curl::PostField.content('document_id', 5)])
+    assert_equal "POST\ndocument_id=5", curl.unescape(curl.body_str)
+  end
+
+  def test_post_remote_is_easy_handle
+    # see: http://pastie.org/560852 and
+    # http://groups.google.com/group/curb---ruby-libcurl-bindings/browse_thread/thread/216bb2d9b037f347?hl=en
+    [:post, :get,:head,:delete].each do |method|
+      count = 0
+      curl = Curl::Easy.send("http_#{method}", TestServlet.url) do|c|
+        count += 1
+        assert_equal Curl::Easy, c.class
+      end
+      assert_equal 1, count, "For request method: #{method.to_s.upcase}"
+    end
+  end
+  
+  def test_post_with_body_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.post_body = 'foo=bar&encoded%20string=val'
+    
+    curl.perform
+    
+    assert_equal "POST\nfoo=bar&encoded%20string=val", curl.body_str
+    assert_equal 'foo=bar&encoded%20string=val', curl.post_body
+  end
+  
+  def test_form_post_body_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_post('foo=bar', 'encoded%20string=val')
+    
+    assert_equal "POST\nfoo=bar&encoded%20string=val", curl.body_str
+    assert_equal 'foo=bar&encoded%20string=val', curl.post_body
+  end
+
+  def test_post_multipart_file_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.multipart_form_post = true
+    pf = Curl::PostField.file('readme', File.expand_path(File.join(File.dirname(__FILE__),'..','README')))
+    curl.http_post(pf)
+    assert_match /HTTP POST file upload/, curl.body_str
+    assert_match /Content-Disposition: form-data/, curl.body_str
+  end
+
+  def test_delete_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_delete
+    assert_equal 'DELETE', curl.body_str
+  end
+
+  def test_arbitrary_http_verb
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http('PURGE')
+    assert_equal 'PURGE', curl.body_str
+  end
+
+  def test_head_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_head
+
+    redirect = curl.header_str.match(/Location: (.*)/)
+
+    assert_equal '', curl.body_str
+    assert_match '/nonexistent', redirect[1]
+  end
+
+  def test_head_accessor
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.head = true
+    curl.perform
+
+    redirect = curl.header_str.match(/Location: (.*)/)
+
+    assert_equal '', curl.body_str
+    assert_match '/nonexistent', redirect[1]
+    curl.head = false
+    curl.perform
+    assert_equal 'GET', curl.body_str
+  end
+
+  def test_put_remote
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.headers['Content-Type'] = 'application/json'
+    assert curl.http_put("message")
+    assert_match /^PUT/, curl.body_str
+    assert_match /message$/, curl.body_str
+    assert_match /application\/json/, curl.header_str
+  end 
   
-end
+  def test_put_data
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.put_data = 'message'
+    
+    curl.perform
+    
+    assert_match /^PUT/, curl.body_str
+    assert_match /message$/, curl.body_str
+  end
+
+  def test_put_nil_data_no_crash
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.put_data = nil
+    
+    curl.perform
+  end
+
+  def test_put_remote_file
+    curl = Curl::Easy.new(TestServlet.url)
+    File.open(__FILE__,'r') do|f|
+      assert curl.http_put(f)
+    end
+    assert_equal "PUT\n#{File.read(__FILE__)}", curl.body_str
+  end
+  
+  def test_put_class_method
+    count = 0
+    curl = Curl::Easy.http_put(TestServlet.url,File.open(__FILE__,'rb')) do|c|
+      count += 1
+      assert_equal Curl::Easy, c.class
+    end
+    assert_equal 1, count
+    assert_equal "PUT\n#{File.read(__FILE__)}", curl.body_str
+  end
+
+  # Generate a self-signed cert with
+  # openssl req -new -newkey rsa:1024 -days 365 -nodes -x509 \
+  #   -keyout tests/cert.pem  -out tests/cert.pem
+  def test_cert
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.cert= File.join(File.dirname(__FILE__),"cert.pem")
+    assert_match /cert.pem$/,curl.cert
+  end
+
+  def test_cert_with_password
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.cert= File.join(File.dirname(__FILE__),"cert.pem:password")
+    assert_match /cert.pem$/,curl.cert
+  end
+
+  def test_cert_type
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.certtype= "DER"
+    assert_equal "DER", curl.certtype
+  end
+
+  def test_default_certtype
+    curl = Curl::Easy.new(TestServlet.url)
+    assert_nil curl.certtype
+    curl.certtype = "PEM"
+    assert_equal "PEM", curl.certtype
+  end
+
+  # Generate a CA cert with instructions at
+  # http://technocage.com/~caskey/openssl/
+  def test_ca_cert
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.cacert= File.join(File.dirname(__FILE__),"cacert.pem")
+    assert_match /cacert.pem$/, curl.cacert
+  end
+
+  def test_user_agent
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.useragent= "Curb-Easy/Ruby"
+    assert_equal "Curb-Easy/Ruby",curl.useragent
+  end
+
+  def test_username_password
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.username = "foo"
+    curl.password = "bar"
+    if !curl.username.nil?
+      assert_equal "foo", curl.username
+      assert_equal "bar", curl.password
+    else
+      curl.userpwd = "foo:bar"
+    end
+    curl.http_auth_types = :basic
+    #curl.verbose = true
+    curl.perform
+    assert_equal 'Basic Zm9vOmJhcg==', $auth_header
+    $auth_header = nil
+    # curl checks the auth type supported by the server, so we have to create a 
+    # new easy handle if we're going to change the auth type...
+
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.username = "foo"
+    curl.password = "bar"
+    if curl.username.nil?
+      curl.userpwd = "foo:bar"
+    end
+    curl.http_auth_types = :ntlm
+    curl.perform
+    assert_equal 'NTLM TlRMTVNTUAABAAAABoIIAAAAAAAAAAAAAAAAAAAAAAA=', $auth_header
+  end
+
+  def test_primary_ip
+    curl = Curl::Easy.new(TestServlet.url)
+    if curl.respond_to?(:primary_ip)
+      curl.perform
+      assert_equal '127.0.0.1', curl.primary_ip
+    end
+  end
+
+  def test_post_streaming
+    readme = File.expand_path(File.join(File.dirname(__FILE__),'..','README'))
+    
+    pf = Curl::PostField.file("filename", readme)
+
+    easy = Curl::Easy.new
+
+    easy.url = TestServlet.url
+    easy.multipart_form_post = true
+    easy.http_post(pf)
+
+    assert_not_equal(0,easy.body_str.size)
+    assert_equal(easy.body_str,File.read(readme))
+  end
+
+
+  def test_easy_close
+    easy = Curl::Easy.new
+    easy.close
+    easy.url = TestServlet.url
+    easy.http_get
+  end
+
+  def test_easy_reset
+    easy = Curl::Easy.new
+    easy.url = TestServlet.url + "?query=foo"
+    easy.http_get
+    settings = easy.reset
+    assert settings.key?(:url)
+    assert settings.key?(:body_data)
+    assert settings.key?(:header_data)
+    easy.url = TestServlet.url
+    easy.http_get
+  end
+
+  def test_easy_use_http_versions
+    easy = Curl::Easy.new
+    easy.url = TestServlet.url + "?query=foo"
+    #puts "http none: #{Curl::HTTP_NONE.inspect}"
+    #puts "http1.0: #{Curl::HTTP_1_0.inspect}"
+    #puts "http1.1: #{Curl::HTTP_1_1.inspect}"
+    easy.version = Curl::HTTP_1_1
+    #easy.verbose = true
+    easy.http_get
+  end
+
+  def test_easy_http_verbs
+    curl = Curl::Easy.new(TestServlet.url)
+    curl.http_delete
+    assert_equal 'DELETE', curl.body_str
+    curl.http_get
+    assert_equal 'GET', curl.body_str
+    curl.http_post
+    assert_equal "POST\n", curl.body_str
+    curl.http('PURGE')
+    assert_equal 'PURGE', curl.body_str
+    curl.http_put('hello')
+    assert_equal "PUT\nhello", curl.body_str
+    curl.http('COPY')
+    assert_equal 'COPY', curl.body_str
+  end
+
+  def test_easy_http_verbs_must_respond_to_str
+    # issue http://github.com/taf2/curb/issues#issue/45
+    assert_nothing_raised do
+      c = Curl::Easy.new ; c.url = 'http://example.com' ; c.http(:get)
+    end
+
+    assert_raise RuntimeError do
+      c = Curl::Easy.new ; c.url = 'http://example.com' ; c.http(FooNoToS.new)
+    end
+
+  end
+
+  # http://github.com/taf2/curb/issues/#issue/33
+  def test_easy_http_verbs_with_errors
+    curl = Curl::Easy.new("http://127.0.0.1:9012/") # test will fail if http server on port 9012
+    assert_raise Curl::Err::ConnectionFailedError do
+      curl.http_delete
+    end
+    curl.url = TestServlet.url
+    curl.http_get
+    assert_equal 'GET', curl.body_str
+  end
+
+  def test_easy_can_put_with_content_length
+    curl = Curl::Easy.new(TestServlet.url)
+    rd, wr = IO.pipe
+    buf = (("hello")* (1000 / 5))
+
+    producer = Thread.new do
+      5.times do
+        wr << buf
+        sleep 0.1 # act as a slow producer
+      end
+    end
+
+    consumer = Thread.new do
+
+      #curl.verbose = true
+      curl.headers['Content-Length'] = buf.size * 5
+      curl.headers['User-Agent']     = "Something else"
+      curl.headers['Content-Type']   = "text/javascript"
+      curl.headers['Date']           = Time.now.httpdate
+      curl.headers['Host']           = 's3.amazonaws.com'
+      curl.headers['Accept']         = '*/*'
+      curl.headers['Authorization']  = 'Foo Bar Biz Baz'
+      curl.http_put(rd)
+      assert_match /^PUT/, curl.body_str
+      assert_match /hello$/, curl.body_str
+      curl.header_str
+      curl.body_str
+    end
+
+    producer.join
+    wr.close
+    consumer.join
+
+  end
+
+  include TestServerMethods 
+
+  def setup
+    server_setup
+  end
+
+end