require 'rhoconnect'
require File.join(File.dirname(__FILE__),'..','spec_helper')

STATS_RECORD_RESOLUTION = 2 unless defined? STATS_RECORD_RESOLUTION
STATS_RECORD_SIZE = 8 unless defined? STATS_RECORD_SIZE

include Rhoconnect
include Rhoconnect::Stats

describe "Record" do
  
  before(:each) do
    @now = 9
    Store.db.flushdb
    Store.stub!(:lock).and_yield
  end
  
  it "should add metric to the record and trim record size" do
    Time.stub!(:now).and_return { @now += 1; @now }
    10.times { Record.add('foo') }
    Store.db.zrange('stat:foo', 0, -1).should == ["2:12", "2:14", "2:16", "2:18"]
  end
  
  it "should add single record" do
    Time.stub!(:now).and_return { @now += 1; @now }
    Record.add('foo')
    Store.db.zrange('stat:foo', 0, -1).should == ["1:10"]
  end
  
  it "should return type of metric" do
    Record.add('foo')
    Record.rtype('foo').should == 'zset'
  end
  
  it "should set string metric" do
    Record.set_value('foo', 'bar')
    Store.db.get('stat:foo').should == 'bar'
  end
  
  it "should get string metric" do
    Store.db.set('stat:foo', 'bar')
    Record.get_value('foo').should == 'bar'
  end
  
  it "should get keys" do
    Record.add('foo')
    Record.add('bar')
    Record.keys.sort.should == ['bar','foo']
  end
  
  it "should add absolute metric value" do
    Time.stub!(:now).and_return { @now += 1; @now }
    time = 0
    4.times do 
      Record.save_average('foo',time)
      time += 1
    end
    Store.db.zrange('stat:foo', 0, -1).should == ["2.0,1.0:10", "2.0,5.0:12"]
  end
  
  it "should update metric" do
    Rhoconnect.stats = true
    Time.stub!(:now).and_return { @now += 1; @now }
    4.times do
      Record.update('foo') do
        # something interesting
      end
    end
    Store.db.zrange('stat:foo', 0, -1).should == ["1,1.0:14", "1,1.0:18", "1,1.0:20"]
    Rhoconnect.stats = false
  end
  
  it "should get range of metric values" do
    Time.stub!(:now).and_return { @now += 1; @now }
    10.times { Record.add('foo') }
    Record.range('foo', 0, 1).should == ["2:12", "2:14"]
  end
  
  it "should reset metric" do
    Time.stub!(:now).and_return { @now += 1; @now }
    10.times { Record.add('foo') }
    Store.db.zrange('stat:foo', 0, -1).should == ["2:12", "2:14", "2:16", "2:18"]
    Record.reset('foo')
    Store.db.zrange('stat:foo', 0, -1).should == []
  end
  
  it "should reset all metrics" do
    Record.add('foo')
    Record.add('bar')
    Record.reset_all
    Store.db.keys('stat:*').should == []
  end
end