setting host should clear out the ip addr

jwarchol · Apr 21, 2011 · d1ac4ca · d1ac4ca
1 parent 8a415e8
commit d1ac4ca
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/statsd.rb b/lib/statsd.rb
@@ -15,6 +15,11 @@ def host_ip_addr
       end
     end
 
+    def host=(h)
+      @host_ip_addr = nil
+      @host = h
+    end
+
     # +stat+ to log timing for
     # +time+ is the time to log in ms
     def timing(stat, time, sample_rate = 1)

diff --git a/test/statsd_test.rb b/test/statsd_test.rb
@@ -96,6 +96,12 @@ def setup
       assert_equal '127.0.0.1', Statsd.host_ip_addr
     end
 
+    should "be cleared when host is set" do
+      assert_equal '127.0.0.1', Statsd.host_ip_addr
+      Statsd.host = 'statsd-01'
+      assert_nil Statsd.instance_variable_get(:@host_ip_addr)
+    end
+
   end
 
   private