samhendley-god 0.7.13

data/test/configs/contact/simple_server.rb

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

data/test/configs/daemon_events/daemon_events.god

@@ -0,0 +1,37 @@
+God.watch do |w|
+  w.name = "daemon-events"
+  w.interval = 5.seconds
+  w.start = 'ruby ' + File.join(File.dirname(__FILE__), *%w[simple_server.rb]) + ' start'
+  w.stop = 'ruby ' + File.join(File.dirname(__FILE__), *%w[simple_server_stop.rb])
+  w.pid_file = '/var/run/daemon-events.pid'
+  w.log = File.join(File.dirname(__FILE__), 'daemon_events.log')
+  w.uid = 'tom'
+  w.gid = 'tom'
+  
+  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
+    
+    # failsafe
+    on.condition(:tries) do |c|
+      c.times = 2
+      c.transition = :start
+    end
+  end
+
+  # start if process is not running
+  w.transition(:up, :start) do |on|
+    on.condition(:process_exits)
+  end
+end

data/test/configs/daemon_events/simple_server.rb

@@ -0,0 +1,8 @@
+require 'rubygems'
+require 'daemons'
+
+puts 'simple server ahoy!'
+
+Daemons.run_proc('daemon-events', {:dir_mode => :system}) do
+  loop { puts 'server'; sleep 1 }
+end

data/test/configs/daemon_events/simple_server_stop.rb

@@ -0,0 +1,11 @@
+# exit!(2)
+
+3.times do
+  puts 'waiting'
+  sleep 1
+end
+
+p ENV
+
+command = '/usr/local/bin/ruby ' + File.join(File.dirname(__FILE__), *%w[simple_server.rb]) + ' stop'
+system(command)

data/test/configs/daemon_polls/daemon_polls.god

@@ -0,0 +1,17 @@
+God.watch do |w|
+  w.name = "daemon-polls"
+  w.interval = 5.seconds
+  w.start = 'ruby ' + File.join(File.dirname(__FILE__), *%w[simple_server.rb]) + ' start'
+  w.stop = 'ruby ' + File.join(File.dirname(__FILE__), *%w[simple_server.rb]) + ' stop'
+  w.pid_file = '/var/run/daemon-polls.pid'
+  w.start_grace = 2.seconds
+  w.log = File.join(File.dirname(__FILE__), *%w[out.log])
+  
+  w.behavior(:clean_pid_file)
+  
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.running = false
+    end
+  end
+end

data/test/configs/daemon_polls/simple_server.rb

@@ -0,0 +1,6 @@
+require 'rubygems'
+require 'daemons'
+
+Daemons.run_proc('daemon-polls', {:dir_mode => :system}) do
+  loop { STDOUT.puts('server'); STDOUT.flush; sleep 1 }
+end

data/test/configs/degrading_lambda/degrading_lambda.god

@@ -0,0 +1,31 @@
+God.watch do |w|
+  w.name = 'degrading-lambda'
+  w.start = 'ruby ' + File.join(File.dirname(__FILE__), *%w[tcp_server.rb])
+  w.interval = 5
+  w.grace = 2
+  w.group = 'test'
+  
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.running = false
+    end
+  end
+  
+  w.restart_if do |restart|
+    restart.condition(:degrading_lambda) do |c|
+      require 'socket'
+      c.lambda = lambda {
+        begin
+          sock = TCPSocket.open('127.0.0.1', 9090)
+          sock.send "2\n", 0
+          retval = sock.gets
+          puts "Retval is #{retval}"
+          sock.close
+          retval
+        rescue
+          false
+        end
+      }
+    end
+  end
+end

data/test/configs/degrading_lambda/tcp_server.rb

@@ -0,0 +1,15 @@
+#! /usr/bin/env ruby
+
+require 'socket'
+server = TCPServer.new('127.0.0.1', 9090)
+while (session = server.accept)
+  puts "Found a session"
+  request = session.gets
+  puts "Request: #{request}"
+  time = request.to_i
+  puts "Sleeping for #{time}"
+  sleep time
+  session.print "Slept for #{time} seconds"
+  session.close
+  puts "Session closed"
+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/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/helper.rb

@@ -0,0 +1,151 @@
+require File.join(File.dirname(__FILE__), *%w[.. lib god])
+God::EventHandler.load
+
+require 'test/unit'
+require 'set'
+
+include God
+
+if Process.uid != 0
+  abort <<-EOF
+\n
+*********************************************************************
+*                                                                   *
+*               You need to run these tests as root                 *
+*           chroot and netlink (linux only) require it              *
+*                                                                   *
+*********************************************************************
+EOF
+end
+
+begin
+  require 'mocha'
+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
+
+def no_stdout
+  old_stdout = $stdout.dup
+  $stdout.reopen(File.open((PLATFORM =~ /mswin/ ? "NUL" : "/dev/null"), 'w'))
+  yield
+  $stdout.reopen(old_stdout)
+end
+
+def no_stderr
+  old_stderr = $stderr.dup
+  $stderr.reopen(File.open((PLATFORM =~ /mswin/ ? "NUL" : "/dev/null"), 'w'))
+  yield
+  $stderr.reopen(old_stderr)
+end
+
+module Kernel
+  def abort(text)
+    raise SystemExit
+  end
+  def exit(code)
+    raise SystemExit
+  end
+end
+
+module Test::Unit::Assertions
+  def assert_abort
+    assert_raise SystemExit do
+      yield
+    end
+  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 =~ /^__/
+    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