statsd-client 0.0.3 → 0.0.4

data/Gemfile

@@ -0,0 +1,2 @@
+source :rubygems
+gemspec

data/Gemfile.lock

@@ -0,0 +1,18 @@
+PATH
+  remote: .
+  specs:
+    statsd-client (0.0.4)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    mocha (0.9.12)
+    shoulda-context (1.0.0.beta1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  mocha
+  shoulda-context
+  statsd-client!

data/Rakefile

@@ -0,0 +1,13 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+$LOAD_PATH.unshift 'lib'
+
+task :default => :test
+
+desc "Run tests"
+task :test do
+  Dir['test/*_test.rb'].each do |f|
+    require File.expand_path(f)
+  end
+end

data/lib/statsd.rb

@@ -1,13 +1,23 @@
 require 'socket'
+require 'resolv'
 
 class Statsd
   
-  Version = '0.0.3'
+  Version = '0.0.4'
   
   class << self
     
     attr_accessor :host, :port
     
+    def host_ip_addr
+      @host_ip_addr ||= Resolv.getaddress(host)
+    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)
@@ -53,7 +63,7 @@ class Statsd
       begin
         sock = UDPSocket.new
         data.each do |d|
-          sock.send(d, 0, host, port)
+          sock.send(d, 0, host_ip_addr, port)
         end
       rescue # silent but deadly
       ensure
@@ -61,9 +71,7 @@ class Statsd
       end
       true
     end
-      
     
   end
   
-  
 end

data/statsd-client.gemspec

@@ -16,5 +16,9 @@ Gem::Specification.new do |s|
   s.files        = `git ls-files`.split("\n")
   s.executables  = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
   s.require_path = 'lib'
+  
+  s.add_development_dependency("shoulda-context")
+  s.add_development_dependency("mocha")  
+  
 end
 

data/test/statsd_test.rb

@@ -0,0 +1,124 @@
+require File.expand_path('../test_helper', __FILE__)
+
+class StatsdTest < Test::Unit::TestCase
+  
+  def setup
+    Statsd.host = 'localhost'
+    Statsd.port = 8125
+    super
+  end
+  
+  context "timing" do
+    
+    should "send with ms" do
+      expected_send('test.stat:23|ms')
+      Statsd.timing('test.stat', 23)
+    end
+    
+    should "log when sampled" do
+      fake_rand(0.09)
+      expected_send('test.stat:23|ms@0.1')
+      Statsd.timing('test.stat', 23, 0.1)
+    end
+    
+    should "not log when not sampled" do
+      fake_rand(0.11)
+      expect_nothing
+      Statsd.timing('test.stat', 23, 0.1)      
+    end
+    
+  end
+  
+  context "increment" do
+    
+    should "send number" do
+      expected_send('test.stat:1|c')
+      Statsd.increment('test.stat')
+    end
+
+    should "log when sampled" do
+      fake_rand(0.09)
+      expected_send('test.stat:1|c@0.1')
+      Statsd.increment('test.stat', 0.1)
+    end
+
+    should "not log when not sampled" do
+      fake_rand(0.11)
+      expect_nothing
+      Statsd.increment('test.stat', 0.1)      
+    end
+        
+  end
+  
+  context "decrement" do
+    
+    should "send number" do
+      expected_send('test.stat:-1|c')
+      Statsd.decrement('test.stat')
+    end
+
+    should "log when sampled" do
+      fake_rand(0.09)
+      expected_send('test.stat:-1|c@0.1')
+      Statsd.decrement('test.stat', 0.1)
+    end
+
+    should "not log when not sampled" do
+      fake_rand(0.11)
+      expect_nothing
+      Statsd.decrement('test.stat', 0.1)      
+    end
+    
+  end
+  
+  context "host" do
+    
+    should "be gettable and settable" do
+      Statsd.host = 'statsd-01'
+      assert_equal 'statsd-01', Statsd.host
+    end
+    
+  end
+  
+  context "port" do
+
+    should "be gettable and settable" do
+      Statsd.port = 1234
+      assert_equal 1234, Statsd.port
+    end
+    
+  end
+  
+  context "host_ip_addr" do
+    
+    should "resolve dns" do
+      Statsd.host = 'localhost'
+      assert loopback?(Statsd.host_ip_addr)
+    end
+    
+    should "be cleared when host is set" do
+      Statsd.host = 'statsd-01'
+      assert_nil Statsd.instance_variable_get(:@host_ip_addr)
+    end
+    
+  end
+  
+  private
+  
+  def fake_rand(v)
+    Statsd.stubs(:rand).returns(v)
+  end
+  
+  def expected_send(buf)
+    UDPSocket.any_instance.expects(:send).with(buf, 0, Statsd.host_ip_addr, Statsd.port).once
+  end
+  
+  def expect_nothing
+    UDPSocket.any_instance.expects(:send).never
+  end
+  
+  def loopback?(ip)
+    ['::1', '127.0.0.1'].include?(ip)
+  end
+  
+end

data/test/test_helper.rb

@@ -0,0 +1,7 @@
+
+require 'shoulda-context'
+require 'mocha'
+
+# initialize to something
+Statsd.host = 'localhost'
+Statsd.port = 8125

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - Ben VandenBos
@@ -14,10 +14,35 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-23 00:00:00 -08:00
+date: 2011-04-20 00:00:00 -07:00
 default_executable: 
-dependencies: []
-
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: shoulda-context
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
 description: Ruby client for statsd.
 email: 
 - bvandenbos@gmail.com
@@ -29,9 +54,14 @@ extra_rdoc_files: []
 
 files: 
 - .gitignore
+- Gemfile
+- Gemfile.lock
 - README.md
+- Rakefile
 - lib/statsd.rb
 - statsd-client.gemspec
+- test/statsd_test.rb
+- test/test_helper.rb
 has_rdoc: true
 homepage: http://github.com/bvandenbos/statsd-client
 licenses: []