tidy_reset 0.1.22 → 0.1.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 20a24f795d8fbf73b6df538ca30cb0465ba94aad
-  data.tar.gz: 9a3667fcf27897987dba479c3e4d89c032061236
+  metadata.gz: 5dfac396bf01cba613e7d43a88f111983eca7786
+  data.tar.gz: 7bf9c15e7f6ad4bd1937eba79ebb26e358583f01
 SHA512:
-  metadata.gz: fb240a163789c03810a667195b6184570868caae9096e586b58283ea2190de455c31ddbf9dcce13f8cd06c8fa4be12a49a2e26a516bf705e3862d74d9dacafb3
-  data.tar.gz: b918b6225c6aa679353164181e20a6417aeae01f281c09e9355e0c83e994458fadc811b4863819728182e369bbaaa65dbca9453fe3ad7535950e38c15c33ddcb
+  metadata.gz: b5d0052b3e2ccda5a0516daafec4b6b989a96d7e926b8f2cee692b4c60ab34c73708831d7e8a1da2055179b78da7f03c7b36e03e56c52c09c52752c7d3a77f8e
+  data.tar.gz: 6b82ccbbd804e68549f06239eb1bbb97c3f35c0019c8794cfdd232bdc6ca4d0ed47ec067d9bc3d406d566abeee5d3571db0960d0de105f1abfd7776787c941b5

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -5,10 +5,12 @@ source "http://rubygems.org"
 
 gem "bundler", "~> 1.0"
 gem "activerecord", "~> 4.0.5"
+gem "platform-api"
 
 # Add dependencies to develop your gem here.
 # Include everything needed to run rake, tests, features, etc.
 group :development do
+  gem "rspec", "~> 2.8.0"
   gem "shoulda", ">= 0"
   gem "rdoc", "~> 3.12"
   gem "jeweler", "~> 2.0.1"

data/Gemfile.lock

@@ -21,7 +21,10 @@ GEM
     builder (3.1.4)
     descendants_tracker (0.0.4)
       thread_safe (~> 0.3, >= 0.3.1)
+    diff-lcs (1.1.3)
     docile (1.1.5)
+    erubis (2.7.0)
+    excon (0.41.0)
     faraday (0.9.0)
       multipart-post (>= 1.2, < 3)
     git (1.2.7)
@@ -34,6 +37,12 @@ GEM
       nokogiri (~> 1.6.0)
       oauth2
     hashie (3.0.0)
+    heroics (0.0.12)
+      erubis (~> 2.7.0)
+      excon
+      moneta
+      multi_json (>= 1.9.2)
+      netrc
     highline (1.6.21)
     i18n (0.6.9)
     jeweler (2.0.1)
@@ -49,9 +58,11 @@ GEM
     jwt (1.0.0)
     mini_portile (0.6.0)
     minitest (4.7.5)
+    moneta (0.8.0)
     multi_json (1.10.1)
     multi_xml (0.5.5)
     multipart-post (2.0.0)
+    netrc (0.8.0)
     nokogiri (1.6.2.1)
       mini_portile (= 0.6.0)
     oauth2 (0.9.4)
@@ -60,10 +71,20 @@ GEM
       multi_json (~> 1.3)
       multi_xml (~> 0.5)
       rack (~> 1.2)
+    platform-api (0.2.0)
+      heroics (~> 0.0.10)
     rack (1.5.2)
     rake (10.3.2)
     rdoc (3.12.2)
       json (~> 1.4)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
     shoulda (3.5.0)
       shoulda-context (~> 1.0, >= 1.0.1)
       shoulda-matchers (>= 1.4.1, < 3.0)
@@ -85,6 +106,8 @@ DEPENDENCIES
   activerecord (~> 4.0.5)
   bundler (~> 1.0)
   jeweler (~> 2.0.1)
+  platform-api
   rdoc (~> 3.12)
+  rspec (~> 2.8.0)
   shoulda
   simplecov

data/Rakefile

@@ -23,20 +23,19 @@ Jeweler::Tasks.new do |gem|
 end
 Jeweler::RubygemsDotOrgTasks.new
 
-require 'rake/testtask'
-Rake::TestTask.new(:test) do |test|
-  test.libs << 'lib' << 'test'
-  test.pattern = 'test/**/test_*.rb'
-  test.verbose = true
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
 end
 
 desc "Code coverage detail"
 task :simplecov do
   ENV['COVERAGE'] = "true"
-  Rake::Task['test'].execute
+  Rake::Task['spec'].execute
 end
 
-task :default => :test
+task :default => :spec
 
 require 'rdoc/task'
 Rake::RDocTask.new do |rdoc|

data/VERSION

@@ -1 +1 @@
-0.1.22
+0.1.23

data/lib/tidy_reset.rb

@@ -1,11 +1,14 @@
 require 'tidy_reset/configuration'
 require 'tidy_reset/version'
+require 'tidy_reset/application_manager'
+require 'yaml'
 require 'open3'
 
 module TidyReset
   class RakeExecuteException < StandardError; end
   class RakeStatusException < StandardError; end
   class RakeCompletedException < StandardError; end
+  class NoTokenException < StandardError; end
  
   class << self
     attr_accessor :configuration
@@ -65,15 +68,23 @@ module TidyReset
     end
 
     def app_stop(stage)
-      execute("heroku maintenance:on --app #{app_name(stage)}")
-      scale_command = dyno_config[stage].inject("") { |command, config| command += "#{config.first}='0' " }
-      execute("heroku ps:scale #{scale_command} --app #{app_name(stage)}")
+      application_name = app_name(stage)
+      application_manager = create_application_manager(application_name)
+      application_manager.enable_maintenance_mode
+      application_manager.shutdown_all_dynos
     end
 
     def app_start(stage)
-      execute("heroku maintenance:off --app #{app_name(stage)}")
-      scale_command = dyno_config[stage].inject("") { |command, config| command += "#{config.first}='#{config.last}' " }
-      execute("heroku ps:scale #{scale_command} --app #{app_name(stage)}")
+      application_name = app_name(stage)
+      application_manager = create_application_manager(application_name)
+      application_manager.scale_dynos_to_configuration(dyno_config[stage])
+      application_manager.disable_maintenance_mode
+    end
+
+    def create_application_manager(application_name)
+      token = configuration.token
+      raise NoTokenException unless token
+      self::ApplicationManager.new(name: application_name, token: token)
     end
 
     def master_connection_from_db_config(db_config)

data/lib/tidy_reset/application_manager.rb

@@ -0,0 +1,41 @@
+require 'platform-api'
+
+module TidyReset
+  class ApplicationManager
+    attr_accessor :name
+
+    def initialize(options)
+      @name = options.fetch(:name)
+      token = options.fetch(:token)
+      @heroku = PlatformAPI.connect_oauth(token)
+    end
+
+    def enable_maintenance_mode
+      @heroku.app.update(@name, { maintenance: true })
+    end
+
+    def disable_maintenance_mode
+      @heroku.app.update(@name, { maintenance: false })
+    end
+
+    def shutdown_all_dynos
+      live_formations = formation_list.select { |f| Integer(f['quantity']) > 0 }
+      live_formations.each do |formation|
+        @heroku.formation.update(@name, formation['id'], { quantity: 0 })
+      end
+    end
+
+    def scale_dynos_to_configuration(dyno_config)
+      formation_list.keep_if do |formation| 
+        dyno_config.keys.include?(formation['type']) 
+      end.each do |formation|
+        @heroku.formation.update(@name, formation['id'], { quantity: dyno_config[formation['type']] } )
+      end
+    end
+
+    private
+    def formation_list
+      @heroku.formation.list(@name)
+    end
+  end
+end

data/lib/tidy_reset/configuration.rb

@@ -1,11 +1,12 @@
 module TidyReset
   class Configuration
-    attr_accessor :databases, :app, :database_encoding
+    attr_accessor :databases, :app, :database_encoding, :token
 
     def initialize
       @databases = []
       @database_encoding = 'utf8'
       @app = ''
+      @token = nil
     end
   end
 end

data/{test/helper.rb → spec/spec_helper.rb}

@@ -14,21 +14,16 @@ end
 ENV["COVERAGE"] && SimpleCov.start do
   add_filter "/.rvm/"
 end
-require 'rubygems'
-require 'bundler'
-begin
-  Bundler.setup(:default, :development)
-rescue Bundler::BundlerError => e
-  $stderr.puts e.message
-  $stderr.puts "Run `bundle install` to install missing gems"
-  exit e.status_code
-end
-require 'test/unit'
-require 'shoulda'
-
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'rspec'
 require 'tidy_reset'
 
-class Test::Unit::TestCase
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+
 end

data/spec/tidy_reset/application_manager_spec.rb

@@ -0,0 +1,69 @@
+require 'spec_helper'
+
+describe "ApplicationManager" do
+  before do 
+    @formation1 = {"command"=>"LIBRATO_AUTORUN=1 bundle exec sidekiq -c 1 -q venice", "created_at"=>"2014-09-17T21:52:05Z", "id"=>"f9d3d59b-324f-4b97-927f-c66c23abcc95", "type"=>"venice_worker", "quantity"=>0, "size"=>"1X", "updated_at"=>"2014-11-17T21:22:38Z"}
+    @formation2 = {"command"=>"LIBRATO_AUTORUN=1 bundle exec sidekiq -c 1 -q redshift_engineering", "created_at"=>"2014-08-26T17:28:01Z", "id"=>"ca48884d-cb04-41ec-b34b-347d542542e2", "type"=>"redshift_engineering_worker", "quantity"=>1, "size"=>"1X", "updated_at"=>"2014-10-29T18:21:59Z"}
+    @formation_list = [@formation1, @formation2]
+
+    @heroku_app = double('PlatformAPI::App')
+
+    @heroku_formation = double('PlatformAPI::Formation')
+    @heroku_formation.stub(:list).and_return(@formation_list)
+
+    @heroku_client = double('PlatformAPI::Client')
+    @heroku_client.stub(:app).and_return(@heroku_app)
+    @heroku_client.stub(:dyno).and_return(@heroku_dyno)
+    @heroku_client.stub(:formation).and_return(@heroku_formation)
+    PlatformAPI.stub(:connect_oauth).and_return(@heroku_client)
+    @subject = TidyReset::ApplicationManager.new(name: 'application-name', token: 'fake-token')
+  end
+
+  describe "#enable_maintenance_mode" do 
+    it "should use the heroku client to update the app's maintenance mode status to true" do 
+      @heroku_app.should_receive(:update).with(@subject.name, {maintenance: true})
+      @subject.enable_maintenance_mode
+    end
+    it "should return the response of the heroku client" do 
+      @heroku_app.stub(:update).and_return({})
+      @subject.enable_maintenance_mode.should == {}
+    end
+  end
+
+  describe "#disable_maintenance_mode" do
+    it "should use the heroku client to update the app's maintenance mode status to false" do 
+      @heroku_app.should_receive(:update).with(@subject.name, {maintenance: false})
+      @subject.disable_maintenance_mode
+    end
+    it "should return the response of the heroku client" do 
+      @heroku_app.stub(:update).and_return({})
+      @subject.disable_maintenance_mode.should == {}
+    end
+  end
+
+  describe "#shutdown_all_dynos" do 
+    it "should use the heroku client to update the quantity of live formations to 0" do
+      @heroku_formation.should_not_receive(:update).with(@subject.name, @formation1['id'], { quantity: 0 })
+      @heroku_formation.should_receive(:update).with(@subject.name, @formation2['id'], { quantity: 0 })
+      @subject.shutdown_all_dynos
+    end
+    it "should return the shutdown formations" do 
+      @heroku_formation.stub(:update)
+      @subject.shutdown_all_dynos.should == [@formation2]
+    end
+  end
+
+  describe "#scale_dynos_to_configuration" do
+    it "should update the formation by type and quantity for each configuration passed in" do 
+      dyno_config = { 'redshift_engineering_worker' => 2 } 
+      @heroku_formation.should_receive(:update).with(@subject.name, @formation2['id'], { quantity: 2 })
+      @subject.scale_dynos_to_configuration(dyno_config)
+    end
+
+    it "should no update formations that do not match the dyno type passed in" do 
+      dyno_config = { 'redshift_engineering_worker' => 2 } 
+      @heroku_formation.should_not_receive(:update).with(@subject.name, @formation1['id'], { quantity: 2 })
+      @subject.scale_dynos_to_configuration(dyno_config)
+    end
+  end
+end

data/spec/tidy_reset/tidy_reset_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+
+describe "TidyReset" do
+  describe "#app_stop" do 
+    before do 
+      @application_manager = double(TidyReset::ApplicationManager)
+      @application_manager.stub(:enable_maintenance_mode)
+      @application_manager.stub(:shutdown_all_dynos)
+      TidyReset.stub(:create_application_manager).and_return(@application_manager)
+      TidyReset.configure do |config|
+        config.app = 'application'
+      end
+    end
+    it "should generate the applicaion name from the stage parameter" do 
+      TidyReset.should_receive(:app_name).with('test')
+      TidyReset.app_stop('test')
+    end
+    it "should create a new application manager" do
+      TidyReset.should_receive(:create_application_manager).with('application-test')
+      TidyReset.app_stop('test')
+    end
+    it "should have the application manager set the app to maintenance mode" do
+      @application_manager.should_receive(:enable_maintenance_mode)
+      TidyReset.app_stop('test')
+    end
+    it "should have the application manager shut down all dynos" do 
+      @application_manager.should_receive(:shutdown_all_dynos)
+      TidyReset.app_stop('test')
+    end
+  end
+
+  describe "#app_start" do 
+    before do 
+      @application_manager = double(TidyReset::ApplicationManager)
+      @application_manager.stub(:disable_maintenance_mode)
+      @application_manager.stub(:scale_dynos_to_configuration)
+      TidyReset.stub(:create_application_manager).and_return(@application_manager)
+      dyno_config = <<-EOF
+      production:
+        web: 100
+        clock: 1
+        redshift: 2
+      test: 
+        web: 2
+        clock: 1
+        generic_worker: 0
+      EOF
+      TidyReset.stub(:dyno_config).and_return(YAML.load(dyno_config))
+      TidyReset.configure do |config|
+        config.app = 'application'
+      end
+    end
+    it "should create a new application manager" do
+      TidyReset.should_receive(:create_application_manager).with('application-test')
+      TidyReset.app_start('test')
+    end
+    it "should have the application manager scale up the configured dynos filtered by stage" do 
+      configuration = {"web"=>2, "clock"=>1, "generic_worker"=>0}
+      @application_manager.should_receive(:scale_dynos_to_configuration).with(configuration)
+      TidyReset.app_start('test')
+    end
+    it "should have the application manager disable maintenance mode" do 
+      @application_manager.should_receive(:disable_maintenance_mode)
+      TidyReset.app_start('test')
+    end
+  end
+
+  describe "#create_application_manager" do 
+    it "should return an instance of an application manager" do 
+      TidyReset.configure do |config|
+        config.token = 'token'
+      end
+      TidyReset.create_application_manager('application-name').should be_a TidyReset::ApplicationManager
+    end
+    it "should require an oauth_token to be set on the configuration" do 
+      TidyReset.configure do |config|
+        config.token = nil
+      end
+      expect { TidyReset.create_application_manager('application_manager') }.to raise_error(TidyReset::NoTokenException)
+    end
+  end
+end

data/tidy_reset.gemspec

@@ -2,16 +2,16 @@
 # DO NOT EDIT THIS FILE DIRECTLY
 # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
 # -*- encoding: utf-8 -*-
-# stub: tidy_reset 0.1.22 ruby lib
+# stub: tidy_reset 0.1.23 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "tidy_reset"
-  s.version = "0.1.22"
+  s.version = "0.1.23"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib"]
   s.authors = ["Thinknear"]
-  s.date = "2014-11-07"
+  s.date = "2014-11-25"
   s.description = "Reset dynos, environment variables, deploys master git branch, and purges the database."
   s.email = "software@thinknear.com"
   s.extra_rdoc_files = [
@@ -20,6 +20,7 @@ Gem::Specification.new do |s|
   ]
   s.files = [
     ".document",
+    ".rspec",
     "Gemfile",
     "Gemfile.lock",
     "LICENSE.txt",
@@ -27,11 +28,13 @@ Gem::Specification.new do |s|
     "Rakefile",
     "VERSION",
     "lib/tidy_reset.rb",
+    "lib/tidy_reset/application_manager.rb",
     "lib/tidy_reset/configuration.rb",
     "lib/tidy_reset/tasks/tidy.rake",
     "lib/tidy_reset/version.rb",
-    "test/helper.rb",
-    "test/test_tidy_reset.rb",
+    "spec/spec_helper.rb",
+    "spec/tidy_reset/application_manager_spec.rb",
+    "spec/tidy_reset/tidy_reset_spec.rb",
     "tidy_reset.gemspec"
   ]
   s.homepage = "http://github.com/ThinkNear/tidy"
@@ -44,6 +47,8 @@ Gem::Specification.new do |s|
     if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
       s.add_runtime_dependency(%q<bundler>, ["~> 1.0"])
       s.add_runtime_dependency(%q<activerecord>, ["~> 4.0.5"])
+      s.add_runtime_dependency(%q<platform-api>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.8.0"])
       s.add_development_dependency(%q<shoulda>, [">= 0"])
       s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
       s.add_development_dependency(%q<jeweler>, ["~> 2.0.1"])
@@ -51,6 +56,8 @@ Gem::Specification.new do |s|
     else
       s.add_dependency(%q<bundler>, ["~> 1.0"])
       s.add_dependency(%q<activerecord>, ["~> 4.0.5"])
+      s.add_dependency(%q<platform-api>, [">= 0"])
+      s.add_dependency(%q<rspec>, ["~> 2.8.0"])
       s.add_dependency(%q<shoulda>, [">= 0"])
       s.add_dependency(%q<rdoc>, ["~> 3.12"])
       s.add_dependency(%q<jeweler>, ["~> 2.0.1"])
@@ -59,6 +66,8 @@ Gem::Specification.new do |s|
   else
     s.add_dependency(%q<bundler>, ["~> 1.0"])
     s.add_dependency(%q<activerecord>, ["~> 4.0.5"])
+    s.add_dependency(%q<platform-api>, [">= 0"])
+    s.add_dependency(%q<rspec>, ["~> 2.8.0"])
     s.add_dependency(%q<shoulda>, [">= 0"])
     s.add_dependency(%q<rdoc>, ["~> 3.12"])
     s.add_dependency(%q<jeweler>, ["~> 2.0.1"])

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tidy_reset
 version: !ruby/object:Gem::Version
-  version: 0.1.22
+  version: 0.1.23
 platform: ruby
 authors:
 - Thinknear
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-07 00:00:00.000000000 Z
+date: 2014-11-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -38,6 +38,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 4.0.5
+- !ruby/object:Gem::Dependency
+  name: platform-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: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.0
 - !ruby/object:Gem::Dependency
   name: shoulda
   requirement: !ruby/object:Gem::Requirement
@@ -104,6 +132,7 @@ extra_rdoc_files:
 - README.md
 files:
 - ".document"
+- ".rspec"
 - Gemfile
 - Gemfile.lock
 - LICENSE.txt
@@ -111,11 +140,13 @@ files:
 - Rakefile
 - VERSION
 - lib/tidy_reset.rb
+- lib/tidy_reset/application_manager.rb
 - lib/tidy_reset/configuration.rb
 - lib/tidy_reset/tasks/tidy.rake
 - lib/tidy_reset/version.rb
-- test/helper.rb
-- test/test_tidy_reset.rb
+- spec/spec_helper.rb
+- spec/tidy_reset/application_manager_spec.rb
+- spec/tidy_reset/tidy_reset_spec.rb
 - tidy_reset.gemspec
 homepage: http://github.com/ThinkNear/tidy
 licenses: []

data/test/test_tidy_reset.rb

@@ -1,7 +0,0 @@
-require 'helper'
-
-class TestTidyReset < Test::Unit::TestCase
-  should "probably rename this file and start testing for real" do
-    flunk "hey buddy, you should probably rename this file and start testing for real"
-  end
-end