mcproc 2016.2.20

data/test/configs/lifecycle/lifecycle.god

@@ -0,0 +1,25 @@
+God::Contacts::Twitter.settings = {
+  # this is for my 'mojombo2' twitter test account
+  # feel free to use it for testing your conditions
+  :username => 'mojombo@gmail.com',
+  :password  => 'gok9we3ot1av2e'
+}
+
+God.contact(:twitter) do |c|
+  c.name      = 'tom2'
+  c.group     = 'developers'
+end
+
+God.watch do |w|
+  w.name = "lifecycle"
+  w.interval = 5.seconds
+  w.start = "/dev/null"
+
+  # lifecycle
+  w.lifecycle do |on|
+    on.condition(:always) do |c|
+      c.what = true
+      c.notify = "tom2"
+    end
+  end
+end

data/test/configs/matias/matias.god

@@ -0,0 +1,50 @@
+$pid_file = "/tmp/matias.pid"
+
+God.task do |w|
+  w.name = "watcher"
+  w.interval = 5.seconds
+  w.valid_states = [:init, :up, :down]
+  w.initial_state = :init
+
+  # determine the state on startup
+  w.transition(:init, { true => :up, false => :down }) do |on|
+    on.condition(:process_running) do |c|
+      c.running = true
+      c.pid_file = $pid_file
+    end
+  end
+
+  # when process is up
+  w.transition(:up, :down) do |on|
+    # transition to 'start' if process goes down
+    on.condition(:process_running) do |c|
+      c.running = false
+      c.pid_file = $pid_file
+    end
+
+    # send up info
+    on.condition(:lambda) do |c|
+      c.lambda = lambda do
+        puts 'yay I am up'
+        false
+      end
+    end
+  end
+
+  # when process is down
+  w.transition(:down, :up) do |on|
+    # transition to 'up' if process comes up
+    on.condition(:process_running) do |c|
+      c.running = true
+      c.pid_file = $pid_file
+    end
+
+    # send down info
+    on.condition(:lambda) do |c|
+      c.lambda = lambda do
+        puts 'boo I am down'
+        false
+      end
+    end
+  end
+end

data/test/configs/real.rb

@@ -0,0 +1,59 @@
+if $0 == __FILE__
+  require File.join(File.dirname(__FILE__), *%w[.. .. lib god])
+end
+
+RAILS_ROOT = "/Users/tom/dev/git/helloworld"
+
+God.watch do |w|
+  w.name = "local-3000"
+  w.interval = 5 # seconds
+  w.start = "mongrel_rails start -P ./log/mongrel.pid -c #{RAILS_ROOT} -d"
+  w.stop = "mongrel_rails stop -P ./log/mongrel.pid -c #{RAILS_ROOT}"
+  w.grace = 5
+
+  pid_file = File.join(RAILS_ROOT, "log/mongrel.pid")
+
+  # clean pid files before start if necessary
+  w.behavior(:clean_pid_file) do |b|
+    b.pid_file = pid_file
+  end
+
+  # start if process is not running
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.running = false
+      c.pid_file = pid_file
+    end
+  end
+
+  # restart if memory or cpu is too high
+  w.restart_if do |restart|
+    restart.condition(:memory_usage) do |c|
+      c.interval = 20
+      c.pid_file = pid_file
+      c.above = (50 * 1024) # 50mb
+      c.times = [3, 5]
+    end
+
+    restart.condition(:cpu_usage) do |c|
+      c.interval = 10
+      c.pid_file = pid_file
+      c.above = 10 # percent
+      c.times = [3, 5]
+    end
+  end
+end
+
+# clear old session files
+# god.watch do |w|
+#   w.name = "local-session-cleanup"
+#   w.start = lambda do
+#     Dir["#{RAILS_ROOT}/tmp/sessions/ruby_sess.*"].select do |f|
+#       File.mtime(f) < Time.now - (7 * 24 * 60 * 60)
+#     end.each { |f| File.delete(f) }
+#   end
+#
+#   w.start_if do |start|
+#     start.condition(:always)
+#   end
+# end

data/test/configs/running_load/running_load.god

@@ -0,0 +1,16 @@
+God.watch do |w|
+  w.name = 'running-load'
+  w.start = '/Users/tom/dev/god/test/configs/child_polls/simple_server.rb'
+  w.stop = ''
+  w.interval = 5
+  w.grace = 2
+  w.uid = 'tom'
+  w.gid = 'tom'
+  w.group = 'test'
+
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.running = false
+    end
+  end
+end

data/test/configs/stop_options/simple_server.rb

@@ -0,0 +1,12 @@
+#! /usr/bin/env ruby
+
+trap :USR1 do
+
+end
+
+loop do
+  STDOUT.puts('server');
+  STDOUT.flush;
+
+  sleep 10
+end

data/test/configs/stop_options/stop_options.god

@@ -0,0 +1,39 @@
+God.watch do |w|
+  w.name = 'stop-options'
+  w.start = File.join(GOD_ROOT, *%w[test configs stop_options simple_server.rb])
+  w.stop_signal = 'USR1'
+  w.stop_timeout = 5
+  w.interval = 5
+  w.grace = 2
+
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.running = false
+    end
+  end
+
+  w.restart_if do |restart|
+    restart.condition(:cpu_usage) do |c|
+      c.above = 30.percent
+      c.times = [3, 5]
+    end
+
+    restart.condition(:memory_usage) do |c|
+      c.above = 10.megabytes
+      c.times = [3, 5]
+    end
+  end
+
+  # lifecycle
+  w.lifecycle do |on|
+    on.condition(:flapping) do |c|
+      c.to_state = [:start, :restart]
+      c.times = 3
+      c.within = 60.seconds
+      c.transition = :unmonitored
+      c.retry_in = 10.seconds
+      c.retry_times = 2
+      c.retry_within = 5.minutes
+    end
+  end
+end

data/test/configs/stress/simple_server.rb

@@ -0,0 +1,3 @@
+#! /usr/bin/env ruby
+
+loop { puts 'server'; sleep 1 }

data/test/configs/stress/stress.god

@@ -0,0 +1,15 @@
+('01'..'08').each do |i|
+  God.watch do |w|
+    w.name = "stress-#{i}"
+    w.start = "ruby " + File.join(File.dirname(__FILE__), *%w[simple_server.rb])
+    w.interval = 0
+    w.grace = 2
+    w.group = 'test'
+
+    w.start_if do |start|
+      start.condition(:process_running) do |c|
+        c.running = false
+      end
+    end
+  end
+end

data/test/configs/task/task.god

@@ -0,0 +1,26 @@
+LOG_DIR = File.join(File.dirname(__FILE__), *%w[logs])
+
+God.task do |t|
+  t.name = 'task'
+  t.valid_states = [:ok, :clean]
+  t.initial_state = :ok
+  t.interval = 5
+
+  # t.clean = lambda do
+  #   Dir[File.join(LOG_DIR, '*.log')].each do |f|
+  #     File.delete(f)
+  #   end
+  # end
+
+  t.clean = "rm #{File.join(LOG_DIR, '*.log')}"
+
+  t.transition(:clean, :ok)
+
+  t.transition(:ok, :clean) do |on|
+    on.condition(:lambda) do |c|
+      c.lambda = lambda do
+        Dir[File.join(LOG_DIR, '*.log')].size > 1
+      end
+    end
+  end
+end

data/test/configs/test.rb

@@ -0,0 +1,61 @@
+ENV['GOD_TEST_RAILS_ROOT'] || abort("Set a rails root for testing in an environment variable called GOD_TEST_RAILS_ROOT")
+
+RAILS_ROOT = ENV['GOD_TEST_RAILS_ROOT']
+
+port = 5000
+
+God.watch do |w|
+  w.name = "local-#{port}"
+  w.interval = 5.seconds
+  w.start = "mongrel_rails start -P ./log/mongrel.pid -c #{RAILS_ROOT} -p #{port} -d"
+  w.restart = "mongrel_rails restart -P ./log/mongrel.pid -c #{RAILS_ROOT}"
+  w.stop = "mongrel_rails stop -P ./log/mongrel.pid -c #{RAILS_ROOT}"
+  w.group = 'mongrels'
+  w.pid_file = File.join(RAILS_ROOT, "log/mongrel.pid")
+
+  # clean pid files before start if necessary
+  w.behavior(:clean_pid_file)
+
+  # determine the state on startup
+  w.transition(:init, { true => :up, false => :start }) do |on|
+    on.condition(:process_running) do |c|
+      c.running = true
+    end
+  end
+
+  # determine when process has finished starting
+  w.transition([:start, :restart], :up) do |on|
+    on.condition(:process_running) do |c|
+      c.running = true
+    end
+  end
+
+  # start if process is not running
+  w.transition(:up, :start) do |on|
+    on.condition(:process_exits)
+  end
+
+  # restart if memory or cpu is too high
+  w.transition(:up, :restart) do |on|
+    on.condition(:memory_usage) do |c|
+      c.interval = 1
+      c.above = 50.megabytes
+      c.times = [3, 5]
+    end
+
+    on.condition(:cpu_usage) do |c|
+      c.interval = 1
+      c.above = 10.percent
+      c.times = [3, 5]
+    end
+
+    on.condition(:http_response_code) do |c|
+      c.host = 'localhost'
+      c.port = port
+      c.path = '/'
+      c.code_is_not = 200
+      c.timeout = 10.seconds
+      c.times = [3, 5]
+    end
+  end
+end

data/test/configs/usr1_trapper.rb

@@ -0,0 +1,10 @@
+#! /usr/bin/env ruby
+
+Signal.trap 'USR1' do
+  puts "can't stop won't stop"
+end
+
+loop do
+  puts 'server'
+  sleep 1
+end

data/test/helper.rb

@@ -0,0 +1,172 @@
+$:.unshift File.expand_path('../../lib', __FILE__) # For use/testing when no gem is installed
+
+# Use this flag to actually load all of the god infrastructure
+$load_god = true
+
+require File.join(File.dirname(__FILE__), *%w[.. lib god sys_logger])
+require File.join(File.dirname(__FILE__), *%w[.. lib god])
+God::EventHandler.load
+
+require 'minitest/autorun'
+require 'minitest/unit'
+require 'set'
+
+include God
+
+if Process.uid != 0 and RbConfig::CONFIG['host_os'] == "linux"
+  abort <<-EOF
+\n
+*********************************************************************
+*                                                                   *
+*               You need to run these tests as root                 *
+*           chroot and netlink (linux only) require it              *
+*                                                                   *
+*********************************************************************
+EOF
+end
+
+begin
+  require 'mocha/setup'
+rescue LoadError
+  unless gems ||= false
+    require 'rubygems'
+    gems = true
+    retry
+  else
+    abort "=> You need the Mocha gem to run these tests."
+  end
+end
+
+module God
+  module Conditions
+    class FakeCondition < Condition
+      def test
+        true
+      end
+    end
+
+    class FakePollCondition < PollCondition
+      def test
+        true
+      end
+    end
+
+    class FakeEventCondition < EventCondition
+      def register
+      end
+      def deregister
+      end
+    end
+  end
+
+  module Behaviors
+    class FakeBehavior < Behavior
+      def before_start
+        'foo'
+      end
+      def after_start
+        'bar'
+      end
+    end
+  end
+
+  module Contacts
+    class FakeContact < Contact
+    end
+
+    class InvalidContact
+    end
+  end
+
+  def self.reset
+    self.watches = nil
+    self.groups = nil
+    self.server = nil
+    self.inited = nil
+    self.host = nil
+    self.port = nil
+    self.pid_file_directory = nil
+    self.registry.reset
+  end
+end
+
+def silence_warnings
+  old_verbose, $VERBOSE = $VERBOSE, nil
+  yield
+ensure
+  $VERBOSE = old_verbose
+end
+
+LOG.instance_variable_set(:@io, StringIO.new())
+
+def output_logs
+  io = LOG.instance_variable_get(:@io)
+  LOG.instance_variable_set(:@io, $stderr)
+  yield
+ensure
+  LOG.instance_variable_set(:@io, io)
+end
+
+# module Kernel
+#   def abort(text)
+#     raise SystemExit, text
+#   end
+#   def exit(code)
+#     raise SystemExit, "Exit code: #{code}"
+#   end
+# end
+
+module Minitest::Assertions
+  def assert_abort
+    assert_raises SystemExit do
+      yield
+    end
+  end
+
+  def assert_nothing_raised
+    yield
+  end
+end
+
+# This allows you to be a good OOP citizen and honor encapsulation, but
+# still make calls to private methods (for testing) by doing
+#
+#   obj.bypass.private_thingie(arg1, arg2)
+#
+# Which is easier on the eye than
+#
+#   obj.send(:private_thingie, arg1, arg2)
+#
+class Object
+  class Bypass
+    instance_methods.each do |m|
+      undef_method m unless m =~ /^(__|object_id)/
+    end
+
+    def initialize(ref)
+      @ref = ref
+    end
+
+    def method_missing(sym, *args)
+      @ref.__send__(sym, *args)
+    end
+  end
+
+  def bypass
+    Bypass.new(self)
+  end
+end
+
+# Make sure we return valid exit codes
+if defined?(RUBY_ENGINE) && RUBY_ENGINE == "ruby" && RUBY_VERSION >= "1.9"
+  module Kernel
+    alias :__at_exit :at_exit
+    def at_exit(&block)
+      __at_exit do
+        exit_status = $!.status if $!.is_a?(SystemExit)
+        block.call
+        exit exit_status if exit_status
+      end
+    end
+  end
+end