1 files changed, 21 insertions, 3 deletions
diff --git a/spec/unit/zero/request/client/address_spec.rb b/spec/unit/zero/request/client/address_spec.rb
index 5e0b65d..09c6da9 100644
--- a/spec/unit/zero/request/client/address_spec.rb
+++ b/spec/unit/zero/request/client/address_spec.rb
@@ -2,8 +2,26 @@ require 'spec_helper'
 
 describe Zero::Request::Client, '#address' do
   subject { Zero::Request::Client.new(env) }
-  let(:address) { '127.0.0.1' }
-  let(:env) { {'REMOTE_ADDR' => address} }
 
-  its(:address) { should == address }
+  context 'without a proxy' do
+    let(:address) { '127.0.0.1' }
+    let(:env) { {'REMOTE_ADDR' => address} }
+
+    its(:address) { should == address }
+  end
+
+  context 'with a proxy' do
+    let(:proxy)   { '127.0.0.1' }
+    let(:address) { '192.168.42.3' }
+    let(:env) do
+      {
+        'REMOTE_ADDR' => proxy,
+        'X_FORWARDED_FOR' => address
+      }
+    end
+
+    its(:address) { should == address }
+    its(:remote_address) { should == proxy }
+    its(:forwarded_for)  { should == address }
+  end
 end