autoscale 0.9.1

data/spec/autoscaler/sidekiq/monitor_middleware_adapter_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+require 'autoscaler/sidekiq/monitor_middleware_adapter'
+
+describe Autoscaler::Sidekiq::MonitorMiddlewareAdapter do
+  before do
+    @redis = Sidekiq.redis = REDIS
+    Sidekiq.redis {|c| c.flushdb }
+  end
+
+  let(:cut) {Autoscaler::Sidekiq::MonitorMiddlewareAdapter}
+  let(:scaler) {TestScaler.new(1)}
+  let(:server) {cut.new(scaler, 0, ['queue'])}
+
+  it('yields') {server.call(Object.new, {}, 'queue') {:foo}.should == :foo}
+  it('yields with a redis pool') {server.call(Object.new, {}, 'queue', Sidekiq.method(:redis)) {:foo}.should == :foo}
+end

data/spec/autoscaler/sidekiq/sleep_wait_server_spec.rb

@@ -0,0 +1,45 @@
+require 'spec_helper'
+require 'autoscaler/sidekiq/sleep_wait_server'
+
+describe Autoscaler::Sidekiq::SleepWaitServer do
+  before do
+    @redis = Sidekiq.redis = REDIS
+    Sidekiq.redis {|c| c.flushdb }
+  end
+
+  let(:cut) {Autoscaler::Sidekiq::SleepWaitServer}
+  let(:scaler) {TestScaler.new(1)}
+  let(:server) {cut.new(scaler, 0, ['queue'])}
+
+  shared_examples "a sleepwait server" do
+  it "scales with no work" do
+    server.stub(:pending_work?).and_return(false)
+    when_run
+    scaler.workers.should == 0
+  end
+
+  it "does not scale with pending work" do
+    server.stub(:pending_work?).and_return(true)
+    when_run
+    scaler.workers.should == 1
+  end
+  end
+
+  describe "a middleware with no redis specified" do
+  it_behaves_like "a sleepwait server" do
+  def when_run
+    server.call(Object.new, {}, 'queue') {}
+  end
+  end
+  end
+
+  describe "a middleware with redis specified" do
+  it_behaves_like "a sleepwait server" do
+  def when_run
+    server.call(Object.new, {}, 'queue', Sidekiq.method(:redis)) {}
+  end
+  end
+  end
+
+  it('yields') {server.call(Object.new, {}, 'queue') {:foo}.should == :foo}
+end

data/spec/autoscaler/sidekiq/specified_queue_system_spec.rb

@@ -0,0 +1,63 @@
+require 'spec_helper'
+require 'autoscaler/sidekiq/specified_queue_system'
+
+describe Autoscaler::Sidekiq::SpecifiedQueueSystem do
+  let(:cut) {Autoscaler::Sidekiq::SpecifiedQueueSystem}
+
+  before do
+    @redis = Sidekiq.redis = REDIS
+    Sidekiq.redis {|c| c.flushdb }
+  end
+
+  def with_work_in_set(queue, set)
+    payload = Sidekiq.dump_json('queue' => queue)
+    Sidekiq.redis { |c| c.zadd(set, (Time.now.to_f + 30.to_f).to_s, payload)}
+  end
+
+  def with_scheduled_work_in(queue)
+    with_work_in_set(queue, 'schedule')
+  end
+
+  def with_retry_work_in(queue)
+    with_work_in_set(queue, 'retry')
+  end
+
+  subject {cut.new(['queue'])}
+
+  it {subject.queue_names.should == ['queue']}
+  it {subject.workers.should == 0}
+
+  describe 'no queued work' do
+    it "with no work" do
+      subject.stub(:sidekiq_queues).and_return({'queue' => 0, 'another_queue' => 1})
+      subject.queued.should == 0
+    end
+
+    it "with scheduled work in another queue" do
+      with_scheduled_work_in('another_queue')
+      subject.scheduled.should == 0
+    end
+
+    it "with retry work in another queue" do
+      with_retry_work_in('another_queue')
+      subject.retrying.should == 0
+    end
+  end
+
+  describe 'with queued work' do
+    it "with enqueued work" do
+      subject.stub(:sidekiq_queues).and_return({'queue' => 1})
+      subject.queued.should == 1
+    end
+
+    it "with schedule work" do
+      with_scheduled_work_in('queue')
+      subject.scheduled.should == 1
+    end
+
+    it "with retry work" do
+      with_retry_work_in('queue')
+      subject.retrying.should == 1
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,16 @@
+require 'sidekiq'
+REDIS = Sidekiq::RedisConnection.create(:url => 'http://localhost:9736', :namespace => 'autoscaler')
+
+RSpec.configure do |config|
+  config.mock_with :rspec
+
+  config.filter_run_excluding :online => true unless ENV['HEROKU_APP']
+end
+
+class TestScaler
+  attr_accessor :workers
+
+  def initialize(n = 0)
+    self.workers = n
+  end
+end

data/spec/test_system.rb

@@ -0,0 +1,11 @@
+class TestSystem
+  def initialize(pending, current = 0)
+    @pending = pending
+    @current = current
+  end
+
+  def workers; @current; end
+  def queued; @pending; end
+  def scheduled; 0; end
+  def retrying; 0; end
+end

metadata

@@ -0,0 +1,187 @@
+--- !ruby/object:Gem::Specification
+name: autoscale
+version: !ruby/object:Gem::Version
+  version: 0.9.1
+platform: ruby
+authors:
+- Justin Love
+- Fix Peña
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-10-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sidekiq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: heroku-api
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: guard-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard-process
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Currently provides a Sidekiq middleware that does 0/1 scaling of Heroku
+  processes
+email:
+- git@JustinLove.name
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- Guardfile
+- README.md
+- examples/complex.rb
+- examples/simple.rb
+- lib/autoscaler.rb
+- lib/autoscaler/binary_scaling_strategy.rb
+- lib/autoscaler/counter_cache_memory.rb
+- lib/autoscaler/counter_cache_redis.rb
+- lib/autoscaler/delayed_shutdown.rb
+- lib/autoscaler/heroku_scaler.rb
+- lib/autoscaler/ignore_scheduled_and_retrying.rb
+- lib/autoscaler/linear_scaling_strategy.rb
+- lib/autoscaler/sidekiq.rb
+- lib/autoscaler/sidekiq/activity.rb
+- lib/autoscaler/sidekiq/celluloid_monitor.rb
+- lib/autoscaler/sidekiq/client.rb
+- lib/autoscaler/sidekiq/entire_queue_system.rb
+- lib/autoscaler/sidekiq/monitor_middleware_adapter.rb
+- lib/autoscaler/sidekiq/queue_system.rb
+- lib/autoscaler/sidekiq/sleep_wait_server.rb
+- lib/autoscaler/sidekiq/specified_queue_system.rb
+- lib/autoscaler/stub_scaler.rb
+- lib/autoscaler/version.rb
+- spec/autoscaler/binary_scaling_strategy_spec.rb
+- spec/autoscaler/counter_cache_memory_spec.rb
+- spec/autoscaler/counter_cache_redis_spec.rb
+- spec/autoscaler/delayed_shutdown_spec.rb
+- spec/autoscaler/heroku_scaler_spec.rb
+- spec/autoscaler/ignore_scheduled_and_retrying_spec.rb
+- spec/autoscaler/linear_scaling_strategy_spec.rb
+- spec/autoscaler/sidekiq/activity_spec.rb
+- spec/autoscaler/sidekiq/celluloid_monitor_spec.rb
+- spec/autoscaler/sidekiq/client_spec.rb
+- spec/autoscaler/sidekiq/entire_queue_system_spec.rb
+- spec/autoscaler/sidekiq/monitor_middleware_adapter_spec.rb
+- spec/autoscaler/sidekiq/sleep_wait_server_spec.rb
+- spec/autoscaler/sidekiq/specified_queue_system_spec.rb
+- spec/spec_helper.rb
+- spec/test_system.rb
+homepage: ''
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: autoscaler
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Start/stop Sidekiq workers on Heroku
+test_files:
+- Guardfile
+- spec/autoscaler/binary_scaling_strategy_spec.rb
+- spec/autoscaler/counter_cache_memory_spec.rb
+- spec/autoscaler/counter_cache_redis_spec.rb
+- spec/autoscaler/delayed_shutdown_spec.rb
+- spec/autoscaler/heroku_scaler_spec.rb
+- spec/autoscaler/ignore_scheduled_and_retrying_spec.rb
+- spec/autoscaler/linear_scaling_strategy_spec.rb
+- spec/autoscaler/sidekiq/activity_spec.rb
+- spec/autoscaler/sidekiq/celluloid_monitor_spec.rb
+- spec/autoscaler/sidekiq/client_spec.rb
+- spec/autoscaler/sidekiq/entire_queue_system_spec.rb
+- spec/autoscaler/sidekiq/monitor_middleware_adapter_spec.rb
+- spec/autoscaler/sidekiq/sleep_wait_server_spec.rb
+- spec/autoscaler/sidekiq/specified_queue_system_spec.rb
+- spec/spec_helper.rb
+- spec/test_system.rb
+has_rdoc: