net_test: cover failure cases

2 files changed, 24 insertions, 0 deletions

diff --git a/test/net_test.rb b/test/net_test.rb
index 8d967e5..705ca73 100644
--- a/test/net_test.rb
+++ b/test/net_test.rb
@@ -8,6 +8,26 @@ class NetTest < HCl::TestCase
     assert_equal 'bobclock', http.subdomain
   end
 
+  def test_redirect_failure
+    register_status(:get, "/taco", 302)
+    assert_raises(HCl::HarvestMiddleware::Failure) { http.get('/taco') }
+  end
+
+  def test_auth_failure
+    register_status(:get, "/burrito", 403)
+    assert_raises(HCl::HarvestMiddleware::AuthFailure) { http.get('/burrito') }
+  end
+
+  def test_throttle_failure
+    register_status(:get, "/sushi", 503)
+    assert_raises(HCl::HarvestMiddleware::ThrottleFailure) { http.get('/sushi') }
+  end
+
+  def test_generic_failure
+    register_status(:get, "/kimchee", 500)
+    assert_raises(HCl::HarvestMiddleware::Failure) { http.get('/kimchee') }
+  end
+
   def test_http_deep_unescape
     register_uri(:get, "/foo", {
       status:'gotten &amp; got!',
diff --git a/test/test_helper.rb b/test/test_helper.rb
index addb134..c87c1db 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -43,6 +43,10 @@ class HCl::TestCase < MiniTest::Test
     @stubs.send(method, path) { [200, {}, Yajl::Encoder.encode(data)] }
   end
 
+  def register_status method, path, status_code
+    @stubs.send(method, path) { [status_code.to_i, {}, ''] }
+  end
+
   def teardown
     @stubs.verify_stubbed_calls
   end