eye 0.1.11

data/spec/dsl/with_server_spec.rb

@@ -0,0 +1,104 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "with_server feature" do
+
+  it "should load matched by string process" do
+    stub(Eye::System).host{ "server1" }
+
+    conf = <<-E
+      Eye.application("bla") do
+        with_server "server1" do
+          process("1"){ pid_file "1.pid" }
+        end
+
+        with_server "server2" do
+          process("2"){ pid_file "2.pid" }
+        end
+      end
+    E
+
+    Eye::Dsl.parse_apps(conf).should == {"bla"=>{:name => "bla", :groups=>{"__default__"=>{:name => "__default__", :application => "bla", :processes=>{"1"=>{:pid_file=>"1.pid", :application=>"bla", :group=>"__default__", :name=>"1"}}}}}}
+  end
+
+  it "should another host conditions" do
+    stub(Eye::System).host{ "server1" }
+
+    conf = <<-E
+      Eye.application("bla") do
+        with_server %w{server1 server2} do
+          process("1"){ pid_file "1.pid" }
+
+          if Eye::SystemResources == 'server2'
+            process("2"){ pid_file "2.pid" }
+          end
+        end
+      end
+    E
+
+    Eye::Dsl.parse_apps(conf).should == {"bla"=>{:name => "bla", :groups=>{"__default__"=>{:name => "__default__", :application => "bla", :processes=>{"1"=>{:pid_file=>"1.pid", :application=>"bla", :group=>"__default__", :name=>"1"}}}}}}
+  end
+
+  describe "matches" do
+    subject{ Eye::Dsl::Opts.new }
+
+    it "match string" do
+      stub(Eye::System).host{ "server1" }
+      subject.with_server("server1").should == true
+      subject.with_server("server2").should == false
+      subject.with_server('').should == true
+      subject.with_server(nil).should == true
+    end
+
+    it "match array" do
+      stub(Eye::System).host{ "server1" }
+      subject.with_server(%w{ server1 server2}).should == true
+      subject.with_server(%w{ server2 server3}).should == false
+    end
+
+    it "match regexp" do
+      stub(Eye::System).host{ "server1" }
+      subject.with_server(%r{server}).should == true
+      subject.with_server(%r{myserver}).should == false
+    end
+  end
+
+  describe "helpers" do
+    it "hostname on with server" do
+      conf = <<-E
+        Eye.application("bla"){ 
+          with_server('muga_server') do
+            working_dir "/tmp"
+          end
+        }
+      E
+      Eye::Dsl.parse_apps(conf).should == {"bla" => {:name => "bla"}}
+    end
+
+    it "with_server work" do
+      Eye::System.host = 'mega_server'
+
+      conf = <<-E
+        Eye.application("bla"){ 
+          with_server('mega_server') do
+            working_dir "/tmp"
+          end
+        }
+      E
+      Eye::Dsl.parse_apps(conf).should == {"bla" => {:working_dir=>"/tmp", :name => "bla"}}
+    end
+
+    it "hostname work" do
+      Eye::System.host = 'supa_server'
+
+      conf = <<-E
+        Eye.application("bla"){ 
+          working_dir "/tmp"
+          env "HOST" => hostname
+        }
+      E
+      Eye::Dsl.parse_apps(conf).should == {"bla" => {:name => "bla", :working_dir=>"/tmp", :environment=>{"HOST"=>"supa_server"}}}
+    end
+
+  end
+
+end

data/spec/example/em.rb

@@ -0,0 +1,57 @@
+require 'bundler/setup'
+require 'eventmachine'
+
+def answer(data)
+  case data
+    when 'ping' then "pong\n"
+    when 'bad' then "what\n"
+    when 'timeout' then sleep 5; "ok\n"
+    when 'exception' then raise 'haha'
+    when 'quit' then EM.stop
+    when 'big' then 'a' * 10_000_000    
+  end
+end
+
+class Echo < EM::Connection
+  def post_init
+    puts "-- someone connected to the echo server!"
+  end
+
+  def receive_data data
+    puts "receive #{data.inspect}"
+    send_data(answer(data))
+  end
+
+  def unbind
+     puts "-- someone disconnected from the echo server!"
+  end  
+end
+
+class EchoObj < EM::Connection
+  include EM::P::ObjectProtocol
+
+  def post_init
+    puts "-- someone connected to the echo server!"
+  end
+
+  def receive_object obj # {:command => 'ping'}
+    puts "receive #{obj.inspect}"
+    send_object(answer(obj[:command]).chop)
+  end
+
+  def unbind
+    puts "-- someone disconnected from the echo server!"
+  end  
+end
+
+trap "TERM" do
+  EM.stop
+  `rm /tmp/em_test_sock_spec` rescue nil
+end
+
+EM.run do
+  EM.start_server '127.0.0.1', 33231, Echo
+  EM.start_server '127.0.0.1', 33232, EchoObj
+  EM.start_server "/tmp/em_test_sock_spec", nil, Echo
+  puts 'started'
+end

data/spec/example/forking.rb

@@ -0,0 +1,20 @@
+require 'bundler/setup'
+require 'forking'
+
+root = File.expand_path(File.dirname(__FILE__))
+
+f = Forking.new(:name => 'forking', :working_dir => root,
+    :log_file => "#{root}/forking.log", 
+    :pid_file => "#{root}/forking.pid", :sync_log => true)
+
+3.times do |i|
+  f.spawn(:log_file => "#{root}/child#{i}.log", :sync_log => true) do
+    $0 = "forking child"
+    loop do
+      p "#{Time.now} - #{Time.now.to_f} - #{i} - tick" 
+      sleep 0.1
+    end
+  end
+end
+
+f.run!

data/spec/example/sample.rb

@@ -0,0 +1,154 @@
+#!/usr/bin/env ruby
+require 'optparse'
+
+# This hash will hold all of the options
+# parsed from the command-line by
+# OptionParser.
+options = {}
+
+optparse = OptionParser.new do|opts|
+  # This displays the help screen, all programs are
+  # assumed to have this option.
+  opts.on( '-h', '--help', 'Display this screen' ) do
+    puts opts
+    exit
+  end
+
+  opts.on( '-p', '--pid FILE', 'pid_file' ) do |a| 
+    options[:pid_file] = a
+  end
+
+  opts.on( '-l', '--log FILE', 'log_file' ) do |a| 
+    options[:log_file] = a
+  end
+
+  opts.on( '-L', '--lock FILE', 'lock_file' ) do |a| 
+    options[:lock_file] = a
+  end
+
+  opts.on( '-d', '--daemonize', 'Daemonize' ) do 
+    options[:daemonize] = true
+  end
+
+  opts.on( '-s', '--daemonize_delay DELAY', 'Daemonized time' ) do |d|
+    options[:daemonize_delay] = d
+  end
+
+  opts.on( '-r', '--raise', 'Raised execution' ) do 
+    options[:raise] = true
+  end
+
+  opts.on( '-w', '--watch_file FILE', 'Exit on touched file' ) do |w|
+    options[:watch_file] = w
+  end
+
+  opts.on( '-W', '--watch_file_delay DELAY', 'Exit on touched file, after delay' ) do |w|
+    options[:watch_file_delay] = w
+  end
+
+  opts.on( '-T', '--no_terminate', 'catch terminate signal and not die' ) do
+    options[:no_terminate] = true
+  end
+
+end
+
+optparse.parse!
+
+module Sample
+  def puts(mes = "")
+    tm = Time.now
+    STDOUT.puts "#{tm.to_s} (#{tm.to_f}) - #{mes}"
+    STDOUT.flush
+  end
+
+  def daemonize(pid_file, log_file, daemonize_delay = 0)
+    puts "daemonize start #{pid_file}, #{log_file}, #{daemonize_delay}"
+
+    if daemonize_delay && daemonize_delay.to_f > 0
+      puts "daemonize delay start #{daemonize_delay}"
+      sleep daemonize_delay.to_f
+      puts "daemonize delay end"
+    end
+
+    daemon
+    STDOUT.reopen(log_file, "a")
+    STDERR.reopen(log_file, "a")
+    File.open(pid_file, 'w'){|f| f.write $$.to_s}
+
+    puts "daemonized"
+  end
+
+  def daemon
+    exit if fork                     # Parent exits, child continues.
+    Process.setsid                   # Become session leader.
+    exit if fork                     # Zap session leader. See [1].
+
+    STDIN.reopen "/dev/null"         # Free file descriptors and
+    STDOUT.reopen "/dev/null", "a"   # point them somewhere sensible.
+    STDERR.reopen '/dev/null', 'a'
+    return 0
+  end
+end
+
+extend Sample
+
+if options[:daemonize]
+  daemonize(options[:pid_file], options[:log_file], options[:daemonize_delay])
+end
+
+puts "Started #{ARGV.inspect}, #{options.inspect}, #{ENV["ENV1"]}"
+
+if options[:lock_file]
+  if File.exists?(options[:lock_file])
+    puts "Lock file exists, exiting"
+    exit 1
+  else
+    File.open(options[:lock_file], 'w'){|f| f.write $$ }
+  end
+end
+
+if options[:raise]
+  puts "Raised"
+  File.unlink(options[:lock_file]) if options[:lock_file]
+  exit 1
+end
+
+trap("USR1") do
+  puts "USR1 signal!"
+end
+
+trap("USR2") do
+  puts "USR2 start memory leak"
+  $ar = []
+  300_000.times{|i| $ar << "memory leak #{i}" * 10}
+end
+
+if options[:no_terminate]
+  trap("TERM") do
+    puts "TERM signal! and we not die O_o"
+  end
+end
+
+loop do
+  sleep 0.1
+  puts "tick"
+
+  if options[:watch_file]
+    if File.exists?(options[:watch_file])
+      puts "watch file finded"
+      File.unlink(options[:watch_file])
+
+      if options[:watch_file_delay]
+        puts "watch_file delay start"
+        sleep options[:watch_file_delay].to_f
+        puts "watch_file delay end"
+      end
+
+      break
+    end
+  end
+end
+
+puts "exit"
+File.unlink(options[:lock_file]) if options[:lock_file]
+exit 0

data/spec/fixtures/dsl/0.rb

@@ -0,0 +1,8 @@
+require File.expand_path(File.join(File.dirname(__FILE__), %w{1.rb}))
+
+Eye.application("bla") do
+  working_dir "/tmp"
+
+  process_1(self, "11")
+  process_1(self, "12")
+end

data/spec/fixtures/dsl/0a.rb

@@ -0,0 +1,8 @@
+require File.join(File.dirname(__FILE__), %w{1.rb})
+
+Eye.application("bla") do |app|
+  working_dir "/tmp"
+
+  process_1(app, "11")
+  process_1(app, "12")
+end

data/spec/fixtures/dsl/0c.rb

@@ -0,0 +1,8 @@
+Eye.load(File.dirname(__FILE__) + "/1*.rb")
+
+Eye.application("bla") do |app|
+  working_dir "/tmp"
+
+  process_1(app, "11")
+  process_1(app, "12")
+end

data/spec/fixtures/dsl/1.rb

@@ -0,0 +1,5 @@
+def process_1(proxy, name)
+  proxy.process(name) do
+    pid_file "#{name}.pid"
+  end
+end

data/spec/fixtures/dsl/bad.eye

@@ -0,0 +1,6 @@
+Eye.application "bla" do
+  process "bad" do
+    # bad because not pid_file
+    working_dir "/tmp"
+  end
+end

data/spec/fixtures/dsl/configs/1.eye

@@ -0,0 +1,3 @@
+Eye.config do
+ logger "/tmp/a.log"
+end

data/spec/fixtures/dsl/configs/2.eye

@@ -0,0 +1 @@
+Eye.config{ http enable: true }

data/spec/fixtures/dsl/configs/3.eye

@@ -0,0 +1 @@
+Eye.config{ http enable: false }

data/spec/fixtures/dsl/configs/4.eye

@@ -0,0 +1,3 @@
+Eye.config do
+  self.logger = nil
+end

data/spec/fixtures/dsl/empty.eye

@@ -0,0 +1,20 @@
+Eye.application("bla") do
+  environment "RAILS_ENV" => "production"
+  keep_alive false
+  stdall "12.log"
+
+  group("ha") do
+    5.times{|i|
+      process("ha_#{i}") do
+        environment "HA" => "1"
+        pid_file "/tmp/#{i}"
+        stdout "11.log"
+      end
+    }
+  end
+
+  process("1") do
+    pid_file "1"
+    stdall "1"
+  end
+end

data/spec/fixtures/dsl/include_test.eye

@@ -0,0 +1,5 @@
+Eye.app :test do
+
+  include "include_test/1.rb"
+
+end

data/spec/fixtures/dsl/include_test/1.rb

@@ -0,0 +1,6 @@
+
+env "a" => 'b'
+
+process :bla do
+  pid_file "10"
+end

data/spec/fixtures/dsl/include_test/ha.rb

@@ -0,0 +1,4 @@
+
+group "ha" do
+  include "1.rb"
+end

data/spec/fixtures/dsl/include_test2.eye

@@ -0,0 +1,5 @@
+Eye.app :test2 do
+
+  include "include_test/ha.rb"
+
+end