refinery 0.12.2 → 1.0

data/test/unit/statistics_test.rb

@@ -1,44 +0,0 @@
-require 'test_helper'
-
-if defined?(Sequel)
-  class StatisticsTest < Test::Unit::TestCase
-    context "a statistics instance" do
-      should "record a done message" do
-      
-        dataset = stub('dataset')
-        dataset.expects(:<<) # TODO: improve this expectation
-      
-        db = stub('db')
-        db.stubs(:table_exists?).returns(true)
-        db.stubs(:[]).with(:completed_jobs).returns(dataset)
-      
-        Sequel.expects(:connect).with('sqlite://stats.db').returns(db)
-      
-        message = {
-          'host_info' => {'hostname' => 'test', 'pid' => 1},
-          'run_time' => 1,
-          'original' => ''
-        }
-        Refinery::Statistics.new.record_done(message)
-      end
-    
-      should "record an error message" do
-        dataset = stub('dataset')
-        dataset.expects(:<<) # TODO: improve this expectation
-      
-        db = stub('db')
-        db.stubs(:table_exists?).returns(true)
-        db.stubs(:[]).with(:errors).returns(dataset)
-      
-        Sequel.expects(:connect).with('sqlite://stats.db').returns(db)
-      
-        message = {
-          'host_info' => {'hostname' => 'test', 'pid' => 1},
-          'error' => {'class' => 'Error', 'message' => 'An error occurred.'},
-          'original' => ''
-        }
-        Refinery::Statistics.new.record_error(message)
-      end
-    end
-  end
-end

data/test/unit/utilities_test.rb

@@ -1,25 +0,0 @@
-require 'test_helper'
-
-class UtilitiesIncluded
-  include Refinery::Utilities
-end
-
-class UtilitiesTest < Test::Unit::TestCase
-  context "a class with utilities included" do
-    setup do
-      @o = UtilitiesIncluded.new
-    end
-    should "camelize a word" do
-      
-      assert_equal 'ClassName', @o.camelize('class_name')
-    end
-    should "encode and decode message" do
-      message = {'some' => 'message'}
-      assert_equal message, @o.decode_message(@o.encode_message(message))
-    end
-    should "provide host info" do
-      assert_not_nil @o.host_info['pid']
-      assert_not_nil @o.host_info['hostname']
-    end
-  end
-end

data/test/unit/validations_test.rb

@@ -1,37 +0,0 @@
-require 'test_helper'
-
-class ValidatedWorker < Refinery::Worker
-  include Refinery::Validations
-  validate_with do |message|
-    raise Refinery::InvalidMessageError, "A message is required" unless message
-  end
-  validates_key_exists :test
-  
-  def execute(message)
-  end
-end
-class ValidationsTest < Test::Unit::TestCase
-  context "a validated worker" do
-    setup do
-      daemon = stub('daemon')
-      @worker = ValidatedWorker.new(daemon)
-    end
-    should "raise a validation error if the key does not exist" do
-      message = {}
-      assert_raise Refinery::InvalidMessageError do
-        @worker.run(message)
-      end
-    end
-    should "not raise a validation error if the key does exist" do
-      message = {:test => 'ding!'}
-      assert_nothing_raised do
-        @worker.run(message)
-      end
-    end
-    should "raise a validation error if the message is nil" do
-      assert_raise Refinery::InvalidMessageError do
-        @worker.run(nil)
-      end
-    end
-  end
-end

data/test/unit/worker_test.rb

@@ -1,44 +0,0 @@
-require 'test_helper'
-
-class SampleWorker < Refinery::Worker
-  attr_reader :message
-  def execute(message)
-    @message = message
-  end
-end
-
-class WorkerTest < Test::Unit::TestCase
-  context "a worker" do
-    setup do
-      daemon = stub('daemon')
-      @worker = SampleWorker.new(daemon)
-      @message = {'test' => 'value'}
-    end
-    should "run" do
-      @worker.run(@message)
-      assert_equal @message, @worker.message
-    end
-    should "provide a data store" do
-      options = {:bucket => 'bucket'}
-      
-      data_store = stub('data store')
-      
-      Moneta::S3.expects(:new).with(
-        :access_key_id => 'aki',  
-        :secret_access_key => 'sak',
-        :bucket => options[:bucket],
-        :multi_thread => true
-      ).returns(data_store)
-      
-      setup_default_config
-      assert_not_nil @worker.data_store(options)
-    end
-    should "provide a queue" do
-      queue = stub('queue')
-      queue_provider = stub('queue provider')
-      queue_provider.expects(:queue).with('a_queue').returns(queue)
-      @worker.expects(:queue_provider).returns(queue_provider)      
-      assert_not_nil @worker.queue('a_queue')
-    end
-  end
-end

data/workers/error.rb

@@ -1,8 +0,0 @@
-# This is an example worker that raises an error.
-class ErrorWorker < Refinery::Worker
-  # Execute the work once.
-  def execute(message)
-    logger.info "received message: #{message.inspect}"
-    raise RuntimeError, "processing failure example"
-  end
-end

data/workers/sample.rb

@@ -1,8 +0,0 @@
-# This is a sample worker.
-class SampleWorker < Refinery::Worker
-  # Execute the work once.
-  def execute(message)
-    logger.info "received message: #{message.inspect}"
-    return true
-  end
-end

data/workers/sleep.rb

@@ -1,7 +0,0 @@
-class SleepWorker < Refinery::Worker
-  def execute(message)
-    logger.info "received message: #{message.inspect}"
-    sleep(message['seconds'])
-    return true
-  end
-end