vanity 1.1.1 → 1.2.0

data/test/test_helper.rb

@@ -5,11 +5,20 @@ require "test/unit"
 require "mocha"
 require "action_controller"
 require "action_controller/test_case"
+require "active_record"
 require "initializer"
-require "lib/vanity/rails"
+Rails.configuration = Rails::Configuration.new
+require "lib/vanity"
 require "timecop"
 
 
+if $VERBOSE
+  $logger = Logger.new(STDOUT)
+  $logger.level = Logger::DEBUG
+else
+  $logger = Logger.new("/dev/null")
+end
+
 class Test::Unit::TestCase
 
   def setup
@@ -27,8 +36,7 @@ class Test::Unit::TestCase
   # Call this if you need a new playground, e.g. to re-define the same experiment,
   # or reload an experiment (saved by the previous playground).
   def new_playground
-    logger = Logger.new("/dev/null") unless $VERBOSE
-    Vanity.playground = Vanity::Playground.new(:logger=>logger, :load_path=>"tmp/experiments", :db=>15)
+    Vanity.playground = Vanity::Playground.new(:logger=>$logger, :load_path=>"tmp/experiments", :db=>15)
     Vanity.playground.mock! unless ENV["REDIS"]
   end
 
@@ -41,6 +49,11 @@ class Test::Unit::TestCase
     end
     names.size == 1 ? metrics.first : metrics
   end
+
+  # Returns named experiment.
+  def experiment(name)
+    Vanity.playground.experiment(name)
+  end
   
   def teardown
     Vanity.context = nil
@@ -53,7 +66,37 @@ end
 ActionController::Routing::Routes.draw do |map|
   map.connect ':controller/:action/:id'
 end
-Rails.configuration = Rails::Configuration.new
+
+
+ActiveRecord::Base.logger = $logger
+ActiveRecord::Base.establish_connection :adapter=>"sqlite3", :database=>File.expand_path("database.sqlite")
+# Call this to define aggregate functions not available in SQlite.
+class ActiveRecord::Base
+  def self.aggregates
+    connection.raw_connection.create_aggregate("minimum", 1) do
+      step do |func, value|
+        func[:minimum] = value.to_i unless func[:minimum] && func[:minimum].to_i < value.to_i
+      end
+      finalize { |func| func.result = func[:minimum] }
+    end
+
+    connection.raw_connection.create_aggregate("maximum", 1) do
+      step do |func, value|
+        func[:maximum] = value.to_i unless func[:maximum] && func[:maximum].to_i > value.to_i
+      end
+      finalize { |func| func.result = func[:maximum] }
+    end
+
+    connection.raw_connection.create_aggregate("average", 1) do
+      step do |func, value|
+        func[:total] = func[:total].to_i + value.to_i
+        func[:count] = func[:count].to_i + 1
+      end
+      finalize { |func| func.result = func[:total].to_i / func[:count].to_i }
+    end
+  end
+end
+
 
 class Array
   # Not in Ruby 1.8.6.
@@ -64,3 +107,20 @@ class Array
     end
   end
 end
+
+
+# Source: http://gist.github.com/25455
+def context(*args, &block)
+  return super unless (name = args.first) && block
+  parent = Class === self ? self : (defined?(ActiveSupport::TestCase) ? ActiveSupport::TestCase : Test::Unit::TestCase)
+  klass = Class.new(parent) do
+    def self.test(name, &block) 
+      define_method("test_#{name.gsub(/\W/,'_')}", &block) if block
+    end
+    def self.xtest(*args) end
+    def self.setup(&block) define_method(:setup) { super() ; block.call } end
+    def self.teardown(&block) define_method(:teardown) { super() ; block.call } end
+  end
+  parent.const_set name.split(/\W+/).map(&:capitalize).join, klass
+  klass.class_eval &block
+end

data/vanity.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name           = "vanity"
-  spec.version        = "1.1.1"
+  spec.version        = "1.2.0"
   spec.author         = "Assaf Arkin"
   spec.email          = "assaf@labnotes.org"
   spec.homepage       = "http://vanity.labnotes.org"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: vanity
 version: !ruby/object:Gem::Version 
-  version: 1.1.1
+  version: 1.2.0
 platform: ruby
 authors: 
 - Assaf Arkin
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-12-04 00:00:00 -08:00
+date: 2009-12-14 00:00:00 -08:00
 default_executable: 
 dependencies: []
 
@@ -29,6 +29,7 @@ files:
 - lib/vanity/commands.rb
 - lib/vanity/experiment/ab_test.rb
 - lib/vanity/experiment/base.rb
+- lib/vanity/helpers.rb
 - lib/vanity/metric.rb
 - lib/vanity/mock_redis.rb
 - lib/vanity/playground.rb
@@ -91,7 +92,7 @@ licenses: []
 post_install_message: To get started run vanity --help
 rdoc_options: 
 - --title
-- Vanity 1.1.1
+- Vanity 1.2.0
 - --main
 - README.rdoc
 - --webcvs