dohruby 0.2.1 → 0.3

data/lib/doh/test/run_tests.rb

@@ -1,82 +1,87 @@
 require 'doh/test/configure_logging'
-require 'doh/core/dir'
 require 'doh/app/activate_database'
 require 'doh/mysql/database_creator'
 require 'test/unit'
-require 'test/unit/collector/dir'
-require 'test/unit/ui/console/testrunner'
+if RUBY_VERSION.to_d < 1.9
+  require 'test/unit/collector/dir'
+  require 'test/unit/ui/console/testrunner'
+end
 require 'doh/test/clear_on_teardown'
+require 'doh/test/setup_once'
 require 'doh/data/require_datagen'
+require 'doh/test/test_result'
 
 module DohTest
 
-def self.find_tests(directory, match, files_found)
-  Dir.glob(File.join(directory, match)).each {|filename| files_found.push(filename)}
-  Dir.directories(directory).each {|subdir| find_tests(File.join(directory, subdir), match, files_found)}
-end
-
 def self.init_test_database
-  dbname = DohApp::config['primary_database']
-  DohDb::DatabaseCreator.new.create_database_copy('doh_test_' + dbname, dbname, true)
-end
-
-def self.init_live_database
-  DohDb::connector_instance.reset
-  DohDb::connector_instance.database = DohApp::config['primary_database']
+  if DohApp::config['database_type'] == 'mysql'
+    dbname = DohApp::config['primary_database']
+    DohDb::DatabaseCreator.new.create_database_copy('doh_test_' + dbname, dbname, true)
+  end
 end
 
-def self.execute_test_files(files, method_filter = nil)
+def self.execute_test_files(files, method_filter = nil, growl_notify = nil)
+if RUBY_VERSION.to_d < 1.9
   collector = Test::Unit::Collector::Dir.new
   collector.filter = [proc {|tstobj| !tstobj.method_name.index(method_filter).nil?}] if method_filter
   suite = collector.collect(*files)
   results = Test::Unit::UI::Console::TestRunner.run(suite)
+
+  dohlog.literal do
+    results.errors.each do |error|
+      dohlog.debug("got #{error.to_s}")
+    end
+    results.failures.each do |failure|
+      dohlog.debug("got #{failure.to_s}")
+    end
+  end
+
+  if growl_notify
+    if !results.passed?
+      `growlnotify -n run_tests_failed -m 'some tests failed' -p 1`
+    else
+      `growlnotify -n run_tests_passed -m 'all tests passed' -p -2`
+    end
+  end
   exit(1) unless results.passed?
   DohTest::check_for_unhandled_errors
+else
+  files.each {|onefile| require onefile}
+end
 end
 
-def self.run_tests_directory(directory, include_production_tests, method_filter = nil)
+def self.run_tests_directory(directory, include_production_tests, include_slow_tests, method_filter = nil, growl_notify = nil)
+  result = false
   if include_production_tests
     letters = 'pq'
-    run_integration_tests = false
   else
     letters = 'dp'
-    run_integration_tests = true
   end
+  letters += 's' if include_slow_tests
 
-  files = []; DohTest::find_tests(directory, "*.dt[#{letters}].rb", files)
+  files = Dir.glob(File.join(directory, '**', "*.dt[#{letters}].rb"))
   unless files.empty?
     init_test_database
-    execute_test_files(files, method_filter)
-  end
-
-  return unless run_integration_tests
-  files = []; DohTest::find_tests(directory, "*.dti.rb", files)
-  unless files.empty?
-    init_live_database
-    execute_test_files(files)
+    execute_test_files(files, method_filter, growl_notify)
+    result = true
   end
+  result
 end
 
-def self.run_tests_file(filename, method_filter = nil)
-  if filename.ends_with('.dti.rb')
-    init_live_database
-  else
-    init_test_database
-  end
-  execute_test_files([filename], method_filter)
+def self.run_tests_file(filename, method_filter = nil, growl_notify = nil)
+  init_test_database
+  execute_test_files([filename], method_filter, growl_notify)
+  true
 end
 
-def self.run_tests(what_to_run, include_production_tests, method_filter = nil)
+def self.run_tests(what_to_run, include_production_tests, include_slow_tests, method_filter = nil, growl_notify = nil)
   DohApp::activate_database
   DohData::require_datagen
-  what_to_run.each do |elem|
-    if File.directory?(elem)
-      run_tests_directory(elem, include_production_tests, method_filter)
-    else
-      run_tests_file(elem, method_filter)
-    end
+  if File.directory?(what_to_run)
+    run_tests_directory(what_to_run, include_production_tests, include_slow_tests, method_filter, growl_notify)
+  else
+    run_tests_file(what_to_run, method_filter, growl_notify)
   end
-  0
 end
 
 end

data/lib/doh/test/setup_once.rb

@@ -0,0 +1,15 @@
+module Test
+  module Unit
+    class TestCase
+      def setup
+        unless @setup_once
+          @setup_once = true
+          setup_once
+        end
+      end
+
+      def setup_once
+      end
+    end
+  end
+end

data/lib/doh/test/test_result.rb

@@ -0,0 +1,7 @@
+module Test
+  module Unit
+    class TestResult
+      attr_reader(:failures, :errors)
+    end
+  end
+end

data/lib/doh/unit_test.rb

@@ -0,0 +1,6 @@
+# this is intended for simple unit test files, particularly those for dohruby itself
+# for a more powerful test setup, use bin/run_tests.rb
+
+require 'doh/test/configure_logging'
+at_exit { DohTest::check_for_unhandled_errors }
+require 'test/unit'

data/lib/doh/util/banking_workday.rb

@@ -1,20 +1,24 @@
 require 'doh/core/date'
 
 class Date
-  # taken from http://www.federalreserve.gov/releases/k8/default.htm
-  @@banking_holidays = [Date.new(2007, 1, 1), Date.new(2007, 1, 15), Date.new(2007, 2, 19), Date.new(2007, 5, 28), Date.new(2007, 7, 4), Date.new(2007, 9, 3), Date.new(2007, 10,  8), Date.new(2007, 11, 12), Date.new(2007, 11, 22), Date.new(2007, 12, 25),
-                        Date.new(2008, 1, 1), Date.new(2008, 1, 21), Date.new(2008, 2, 18), Date.new(2008, 5, 26), Date.new(2008, 7, 4), Date.new(2008, 9, 1), Date.new(2008, 10, 13), Date.new(2008, 11, 11), Date.new(2008, 11, 27), Date.new(2008, 12, 25),
-                        Date.new(2009, 1, 1), Date.new(2009, 1, 19), Date.new(2009, 2, 16), Date.new(2009, 5, 25), Date.new(2009, 7, 4), Date.new(2009, 9, 7), Date.new(2009, 10, 12), Date.new(2009, 11, 11), Date.new(2009, 11, 26), Date.new(2009, 12, 25),
-                        Date.new(2010, 1, 1), Date.new(2010, 1, 18), Date.new(2010, 2, 15), Date.new(2010, 5, 31), Date.new(2010, 7, 5), Date.new(2010, 9, 6), Date.new(2010, 10, 11), Date.new(2010, 11, 11), Date.new(2010, 11, 25), Date.new(2010, 12, 25),
-                        Date.new(2011, 1, 1), Date.new(2011, 1, 17), Date.new(2011, 2, 21), Date.new(2011, 5, 30), Date.new(2011, 7, 4), Date.new(2011, 9, 5), Date.new(2011, 10, 10), Date.new(2011, 11, 11), Date.new(2011, 11, 24), Date.new(2011, 12, 26),
-                       ]
-
-  def banking_holiday?
-    @@banking_holidays.include?(self)
+  # algoritm based on http://www.federalreserve.gov/releases/k8/default.htm and wikipedia pages about the individual holidays
+  def banking_workday?
+    return false if wday == 0 || wday == 6
+    !(case month
+    when 1 then (mday == 1) || ((wday == 1) && ((mday == 2) || (mday > 14 && mday < 22)))
+    when 2 then (wday == 1) && (mday > 14 && mday < 22)
+    when 5 then (wday == 1) && (mday > 24)
+    when 7 then (mday == 4) || ((mday == 5) && (wday == 1))
+    when 9 then (wday == 1) && (mday < 8)
+    when 10 then (wday == 1) && (mday > 7) && (mday < 15)
+    when 11 then (mday == 11) || ((mday == 12) && (wday == 1)) || ((wday == 4) && (mday > 21 && mday < 29))
+    when 12 then (mday == 25) || ((mday == 26) && (wday == 1))
+    else false
+    end)
   end
 
-  def banking_workday?
-    weekday? && !banking_holiday?
+  def banking_holiday?
+    weekday? && !banking_workday?
   end
 
   def forward_banking_workday(count = 1)

data/lib/doh/util/class_basename.rb

@@ -0,0 +1,10 @@
+require 'doh/core/string'
+
+module Doh
+
+def self.class_basename(klass)
+  full_name = klass.to_s
+  full_name.rafter('::') || full_name
+end
+
+end

data/lib/doh/util/current_date.rb

@@ -2,64 +2,41 @@ require 'doh/core/date'
 
 module Doh
 
-@@current_date = nil
-@@current_datetime = nil
+@@current_date_objs = []
 
 def self.current_date(dflt = Date.today)
-  @@current_date || (@@current_datetime && @@current_datetime.date_only) || dflt
+  return dflt if @@current_date_objs.empty?
+  @@current_date_objs.last.date_only
 end
 
-def self.set_current_date(date)
-  @@current_date = date
+def self.current_datetime(dflt = DateTime.zow)
+  return dflt if @@current_date_objs.empty?
+  cdo = @@current_date_objs.last
+  return cdo if cdo.respond_to?(:hour)
+  DateTime.new(cdo.year, cdo.month, cdo.day, dflt.hour, dflt.min, dflt.sec, dflt.zone)
 end
 
-def self.current_date_set?
-  !@@current_date.nil?
+def self.push_current_date(date_obj)
+  @@current_date_objs.push(date_obj)
 end
 
-def self.clear_current_date
-  @@current_date = nil
-end
-
-def self.date_as(date, &block)
-  save_date = @@current_date
-  @@current_date = date
-  begin
-    retval = block.call
-  rescue
-    @@current_date = save_date
-    raise
-  end
-  @@current_date = save_date
-  retval
-end
-
-def self.current_datetime(dflt = DateTime.now)
-  @@current_datetime || (@@current_date && DateTime.new(@@current_date.year, @@current_date.month, @@current_date.day, rand(12) + 2, rand(60), rand(60))) || dflt
-end
-
-def self.set_current_datetime(datetime)
-  @@current_datetime = datetime
-end
-
-def self.current_datetime_set?
-  !@@current_datetime.nil?
+def self.pop_current_date
+  @@current_date_objs.pop
 end
 
-def self.clear_current_datetime
-  @@current_datetime = nil
+def self.clear_current_date
+  @@current_date_objs.clear
 end
 
-def self.datetime_as(datetime, &block)
-  save_datetime = @@current_datetime
-  @@current_datetime = datetime
+def self.date_as(date_obj, &block)
+  push_current_date(date_obj)
   begin
     retval = block.call
   rescue
-    @@current_datetime = save_datetime
     raise
+  ensure
+    pop_current_date
   end
-  @@current_datetime = save_datetime
   retval
 end
 

data/lib/doh/util/doh_socket.rb

@@ -0,0 +1,56 @@
+require 'socket'
+
+module Doh
+
+class TimeoutError < RuntimeError
+end
+
+class DohSocket
+  @@socket_end_times = []
+  attr_accessor :timeout
+
+  def initialize(server, port)
+    @sock = TCPSocket.new(server, port)
+    @timeout = 300
+  end
+
+  def self.timeout_block(duration)
+    push_duration(duration)
+    begin
+      yield
+    ensure
+      pop_duration
+    end
+  end
+
+  def write(str)
+    @sock.write(str)
+  end
+
+  def timed_readln
+    received = ''
+    while (received =~ /$/) == received.size
+      if (!select([@sock], nil, nil, time_left))
+        raise TimeoutError.new('DohSocket timed_readln timed out')
+      end
+      received += @sock.recvfrom(65536)[0]
+    end
+    received
+  end
+
+  private
+  def self.push_duration(duration)
+    @@socket_end_times.push(Time.now + duration)
+  end
+
+  def self.pop_duration
+    @@socket_end_times.pop
+  end
+
+  def time_left
+    result = [@@socket_end_times, Time.now + @timeout].flatten.min - Time.now
+    result = 0.001 if result <= 0
+    result
+  end
+end
+end

data/lib/doh/util/email.rb

@@ -0,0 +1,18 @@
+require 'net/smtp'
+module DohEmail
+
+@@mail_server = DohApp::config['mail_server'] || 'mail.' + DohApp::get_required_config_value('domain_name', 'mail_server or domain_name required for mail server')
+@@from_address = DohApp::config['from_address'] || 'admin@' + DohApp::get_required_config_value('domain_name', 'from_address or domain_name required for notification email address')
+
+def self.send_email(subject, body, recipients, from_address = @@from_address)
+  msg = "Subject:#{subject}\nFrom:#{from_address}\nTo:#{recipients.join(', ')}\nContent-Type: text/plain\n\n#{body}"
+  Net::SMTP.start(@@mail_server) do |smtp|
+    begin
+      smtp.send_message(msg, from_address, recipients)
+    rescue Exception => excpt
+      Doh::log.warn("got exception while sending email: #{excpt.inspect}")
+    end
+  end
+end
+
+end

data/lib/doh/util/file_edit.rb

@@ -0,0 +1,64 @@
+require 'socket'
+
+module Doh
+
+class TimeoutError < RuntimeError
+end
+
+class FileEdit
+  def initialize(filename, &block)
+    @filename = filename
+    @file = File.new(filename, 'r').to_a
+    @file.each_with_index do |line, index|
+      @file[index] = @file[index][0..-2]
+    end
+    @index = 0
+
+    if block
+      yield(self)
+      save
+    end
+  end
+
+  def move_after(regexp)
+    @file.each_with_index do |line, index|
+      if line =~ regexp
+        @index = index + 1
+      end
+    end
+  end
+
+  def move_before(regexp)
+    move_after(regexp)
+    @index -= 1
+  end
+
+  def insert(*lines)
+    lines.each do |line|
+      @file.insert(@index, line)
+      @index += 1
+    end
+  end
+
+  def gsub(regexp, replacement)
+    @file.each_with_index do |line, index|
+      @file[index] = line.gsub(regexp, replacement)
+    end
+  end
+
+  def delete(numlines = 1)
+    numlines.times do
+      @file.delete_at(@index)
+    end
+  end
+
+  def save
+    outf = File.open(@filename, 'w+')
+    @file.each do |line|
+      outf << line << "\n"
+    end
+    outf.close
+  end
+end
+
+end

data/lib/doh/util/http_helper.rb

@@ -0,0 +1,107 @@
+require 'net/http'
+require 'net/https'
+require 'uri'
+
+module Doh
+
+class HttpHelper
+  attr_reader :headers
+
+  def initialize(url)
+    @url = url
+    @headers = {}
+    init_sender
+  end
+
+  def post(data, get_body_only = true)
+    dohlog.debug("posting to url #@url")
+    response = if data.is_a?(Hash)
+      post_hash(data)
+    else
+      post_string(data)
+    end
+    check_for_errors(response)
+    return response.body.to_s if get_body_only
+    response
+  end
+
+  def get(get_body_only = true, allow_redirect = false, num_redirects = 0)
+    getstr = @parsed_url.path
+    getstr += '?' + @parsed_url.query unless @parsed_url.query.to_s.empty?
+    request = Net::HTTP::Get.new(getstr)
+    request.initialize_http_header(@headers)
+    response = @sender.start { |http| http.request(request) }
+    if response.is_a?(Net::HTTPRedirection) && allow_redirect
+      raise "too many redirects" if num_redirects > 20
+      @url = response['location']
+      init_sender
+      return get(get_body_only, true, num_redirects + 1)
+    end
+    return response.body.to_s if get_body_only
+    response
+  end
+
+  def xml
+    @headers['Content-Type'] = 'text/xml'
+    self
+  end
+
+  def soap_action(action)
+    @headers['SOAPAction'] = action
+    self
+  end
+
+  def nossl
+    @sender.use_ssl = false
+    @nossl = true
+    self
+  end
+
+  def timeout(seconds)
+    @sender.read_timeout = seconds
+    self
+  end
+
+  def hdr(name, value)
+    @headers[name] = value
+    self
+  end
+
+  #shortcut method
+  def self.load(url, use_ssl = true)
+    helper = HttpHelper.new(url)
+    helper.nossl if !use_ssl
+    helper.get
+  end
+private
+  def init_sender
+    @parsed_url = URI.parse(@url)
+    @sender = Net::HTTP.new(@parsed_url.host, @parsed_url.port)
+    @sender.use_ssl = true unless @nossl || (['127.0.0.1', 'localhost'].include?(@parsed_url.host) && @parsed_url.port != 443)
+    @sender.verify_mode = OpenSSL::SSL::VERIFY_NONE
+  end
+
+  def post_hash(data)
+    request = Net::HTTP::Post.new(@parsed_url.path)
+    request.set_form_data(data)
+    request.initialize_http_header(@headers)
+    @sender.start { |http| http.request(request) }
+  end
+
+  def post_string(data)
+    @sender.post(@parsed_url.path, data, @headers)
+  end
+
+  def check_for_errors(response)
+    # TODO: do something useful here
+#     case response
+#     when Net::HTTPSuccess
+#       puts "success"
+#       puts response.body.to_s
+#     else
+#       puts response.error!
+#     end
+  end
+end
+
+end

data/lib/doh/util/internal_ip.rb

@@ -3,7 +3,7 @@ require 'socket'
 module Doh
 
 def self.internal_ip
-  Socket.getaddrinfo(Socket.gethostname, nil, Socket::AF_INET, Socket::SOCK_STREAM)[0][3]
+  @@internal_ip ||= Socket.getaddrinfo(Socket.gethostname, nil, Socket::AF_INET, Socket::SOCK_STREAM)[0][3]
 end
 
 end

data/lib/doh/util/jsval.rb

@@ -0,0 +1,13 @@
+require 'date'
+
+module Doh
+
+def self.jsval(value)
+  if value.is_a?(Date)
+    "new Date(#{value.year},#{value.month-1},#{value.mday})"
+  else
+    value
+  end
+end
+
+end

data/lib/doh/util/post_hash.rb

@@ -0,0 +1,14 @@
+require 'net/http'
+require 'net/https'
+require 'uri'
+require 'doh/util/xml_util'
+require 'doh/util/http_helper'
+
+module Doh
+
+def self.post_hash_as_xml(hash, base_node = nil, url = 'http://localhost:4000/')
+  helper = HttpHelper.new(url).xml
+  helper.post("<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>#{hash_to_xml(hash, base_node)}")
+end
+
+end

data/lib/doh/util/xml_util.rb

@@ -0,0 +1,48 @@
+require 'rexml/document'
+
+module Doh
+
+def self.hash_to_xml(hash, root_node_name = nil)
+  doc = REXML::Document.new
+  doc << REXML::XMLDecl.new
+  if root_node_name
+    doc = doc.add_element(root_node_name)
+  end
+  elem_to_xml(hash, doc)
+  doc.to_s
+end
+
+def self.elem_to_xml(elem, node)
+  elem.each_pair do |field, value|
+    if value.class == Hash
+      elem_to_xml(value, node.add_element(field))
+    else
+      node.add_element(field).add_text(value.to_s)
+    end
+  end
+end
+
+def self.xml_to_hash(xml)
+  doc = REXML::Document.new(xml)
+  doc.children.each do |child|
+    if child.respond_to?('name')
+      return elem_to_hash(child)
+    end
+  end
+  raise "unable to find root node in xml: #{xml.inspect}"
+end
+
+def self.elem_to_hash(elem)
+  return {} if !elem.is_a?(REXML::Element)
+  if elem.has_elements?
+    children_hash = {}
+    elem.children.each do |child|
+      children_hash.merge!(elem_to_hash(child))
+    end
+    {elem.name => children_hash}
+  else
+    {elem.name => elem.text}
+  end
+end
+
+end

data/test/core/tc_array.rb

@@ -0,0 +1,12 @@
+require 'test/unit'
+require 'doh/core/array'
+
+module Doh
+
+class TC_core_array < Test::Unit::TestCase
+  def test_build_hash
+    assert_equal({'a' => 1, 'b' => 2}, [['a', 1], ['b', 2]].build_hash { |elem| elem})
+  end
+end
+
+end