dohruby 0.2.1 → 0.3

data/test/mysql/tc_migrate.rb

@@ -0,0 +1,50 @@
+require File.join(File.dirname(__FILE__), 'db_unit_test')
+require 'doh/mysql/migrate'
+require 'doh/mysql/version'
+
+module DohDb
+
+class TC_migrate < Test::Unit::TestCase
+  def cleanup_tables
+    DohDb::query("DROP TABLE IF EXISTS version")
+    DohDb::query("DROP TABLE IF EXISTS migrate_test_version2")
+  end
+
+  def create_version_table
+    DohDb::query("CREATE TABLE IF NOT EXISTS version (version INT UNSIGNED NOT NULL) ENGINE=MyISAM")
+    DohDb::query("INSERT INTO version VALUES (0)")
+  end
+
+  def test_stuff
+    cleanup_tables
+    create_version_table
+    dbmg = DatabaseMigrator.new(DohDb::connector_instance.database, nil, File.dirname(__FILE__))
+
+    dbmg.update
+    table_info = DohDb::select("DESC migrate_test_version2")
+    assert_equal(2, table_info.size)
+    assert_equal(2, DohDb::current_database_version)
+
+    dbmg.make_version(1)
+    table_info = DohDb::select("DESC migrate_test_version2")
+    assert_equal(1, table_info.size)
+    assert_equal(1, DohDb::current_database_version)
+
+    dbmg.make_version(0)
+    assert_raise(Mysql::Error) {DohDb::query("DESC migrate_test_version2")}
+    assert(DohTest::pop_error.exception.to_s =~ /doesn't exist/)
+    assert_equal(0, DohDb::current_database_version)
+
+    dbmg.make_version(1)
+    table_info = DohDb::select("DESC migrate_test_version2")
+    assert_equal(1, table_info.size)
+    assert_equal(1, DohDb::current_database_version)
+  end
+
+  def teardown
+    cleanup_tables
+    super
+  end
+end
+
+end

data/test/mysql/tc_parse.rb

@@ -3,7 +3,7 @@ require 'doh/mysql/parse'
 
 module DohDb
 
-class TC_Convert < Test::Unit::TestCase
+class TC_Parse < Test::Unit::TestCase
   def test_bool
     assert_equal(false, DohDb::parse_bool('0'))
     assert_equal(true, DohDb::parse_bool('1'))
@@ -15,6 +15,7 @@ class TC_Convert < Test::Unit::TestCase
     assert_equal(Date.new(2008,2,14), DohDb::parse_date('2008-02-14'))
     assert_raise(ArgumentError) {DohDb::parse_date('20080214')}
     assert_not_equal(Date.new(2008,2,14), DohDb::parse_date('2008-02-15'))
+    assert_equal(nil, DohDb::parse_date('0000-00-00'))
   end
 
   def test_datetime
@@ -22,6 +23,7 @@ class TC_Convert < Test::Unit::TestCase
     assert_raise(ArgumentError) {DohDb::parse_datetime('zzzzzzzzzzzzzzzzzzz')}
     assert_equal(DateTime.new(2008,2,14,10,20,30), DohDb::parse_datetime('2008-02-14 10:20:30'))
     assert_not_equal(DateTime.new(2008,2,14,10,20,30), DohDb::parse_datetime('2008-02-14 10:20:31'))
+    assert_equal(nil, DohDb::parse_datetime('0000-00-00 00:00:00'))
   end
 
   def test_decimal

data/test/mysql/tc_readonly_row.rb

@@ -5,11 +5,11 @@ module DohDb
 
 class TC_ReadOnlyRow < Test::Unit::TestCase
   def setup
-    @row = ReadOnlyRow.new(['id', 'fname', 'lname', 'other'], [3, 'george', 'bob', nil])
+    @row = ReadOnlyRow.new(['id', 'fname', 'lname', 'other', 'flag', 'emptystr'], [3, 'george', 'bob', nil, true, ''])
   end
 
   def test_keys
-    assert_equal(['id', 'fname', 'lname', 'other'], @row.keys)
+    assert_equal(['id', 'fname', 'lname', 'other', 'flag', 'emptystr'], @row.keys)
   end
 
   def test_at_bad
@@ -50,32 +50,36 @@ class TC_ReadOnlyRow < Test::Unit::TestCase
 
   def test_to_a
     ary = @row.to_a
-    assert_equal(4, ary.size)
+    assert_equal(6, ary.size)
     assert_equal(['id', 3], ary.at(0))
     assert_equal(['fname','george'], ary.at(1))
     assert_equal(['lname', 'bob'], ary.at(2))
     assert_equal(['other', nil], ary.at(3))
+    assert_equal(['flag', true], ary.at(4))
+    assert_equal(['emptystr', ''], ary.at(5))
   end
 
   def test_to_h
     hsh = @row.to_h
-    assert_equal(4, hsh.size)
+    assert_equal(6, hsh.size)
     assert_equal(3, hsh['id'])
     assert_equal('george', hsh['fname'])
     assert_equal('bob', hsh['lname'])
     assert_equal(nil, hsh['other'])
   end
 
-  def test_inspect
-    assert_equal(@row.inspect, @row.to_h.inspect)
-  end
-
   def test_method_missing
-    assert_equal(3, @row.id)
     assert_equal('george', @row.fname)
     assert_equal('bob', @row.lname)
     assert_equal(nil, @row.other)
-    assert_raise(RuntimeError) {@row.this_name_doesnt_exist}
+    assert_raise(RuntimeError) {@row.this_field_doesnt_exist}
+  end
+
+  def test_empty_field
+    assert(!@row.empty_field?('id'))
+    assert(@row.empty_field?('other'))
+    assert(@row.empty_field?('emptystr'))
+    assert(@row.empty_field?('unknown_field_name'))
   end
 end
 

data/test/mysql/tc_smart_row.rb

@@ -0,0 +1,22 @@
+require 'doh/unit_test'
+require 'doh/mysql/smart_row'
+
+module DohDb
+
+class TC_SmartRow < Test::Unit::TestCase
+  def test_display
+    row = SmartRow.new(['flag'], [true])
+    assert_equal('', row.display('this_field_doesnt_exist'))
+    assert_raise(RuntimeError) { row.display.this_field_doesnt_exist }
+    assert_equal('yes', row.display('flag'))
+    assert_equal('yes', row.display.flag)
+    row2 = row.dup
+    assert_equal(true, row2.flag)
+    row2.flag = false
+    assert_equal(false, row2.flag)
+    assert_equal(true, row.flag)
+    assert_raise(RuntimeError) { row.this_field_doesnt_exist }
+  end
+end
+
+end

data/test/mysql/tc_to_sql.rb

@@ -0,0 +1,20 @@
+require 'test/unit'
+require 'doh/mysql/to_sql'
+
+module DohDb
+
+class TC_to_sql < Test::Unit::TestCase
+  def test_stuff
+    assert_equal('"blah"', 'blah'.to_sql)
+    assert_equal('NULL', nil.to_sql)
+    assert_equal('3', 3.to_sql)
+    assert_equal('"2008-09-24 09:30:04"', DateTime.new(2008,9,24,9,30,4).to_sql)
+    assert_equal('1', true.to_sql)
+    assert_equal('0', false.to_sql)
+    assert_equal('10.0', BigDecimal.new('10').to_sql)
+    assert_equal('("a","b","c")', ['a', 'b', 'c'].to_sql)
+  end
+end
+
+end
+

data/test/mysql/tc_types.rb

@@ -0,0 +1,32 @@
+require File.join(File.dirname(__FILE__), 'db_unit_test')
+require 'doh/mysql/types'
+
+module DohDb
+
+class ConvertToString
+  def self.build(field, value)
+    value.to_s
+  end
+end
+
+class TC_Types < Test::Unit::TestCase
+  def test_stuff
+    dbh = DohDb::request_handle
+    tbl = "doh_mysql_types_stuff_test"
+    tbl2 = "island.doh_mysql_types_stuff_another_test"
+    DohDb::query("CREATE TEMPORARY TABLE #{tbl} (amount INT)")
+    DohDb::query("INSERT INTO #{tbl} SET amount = NULL")
+    assert_equal(nil, DohDb::select_field("SELECT amount FROM #{tbl}"))
+    DohDb::query("UPDATE #{tbl} SET amount = 5")
+    assert_equal(5, DohDb::select_field("SELECT amount FROM #{tbl}"))
+
+    DohDb::query("CREATE TEMPORARY TABLE #{tbl2} (other INT)")
+    DohDb::query("INSERT INTO #{tbl2} SET other = 7")
+    assert_equal(5, DohDb::select_field("SELECT amount, other FROM #{tbl}, #{tbl2}"))
+
+    DohDb::register_column_type(DohDb::connector_instance.database, tbl, 'amount', ConvertToString)
+    assert_equal('5', DohDb::select_field("SELECT amount FROM #{tbl}"))
+  end
+end
+
+end

data/test/mysql/tc_unquoted.rb

@@ -9,6 +9,7 @@ class TC_Unquoted < Test::Unit::TestCase
     assert_equal('blah', Unquoted.new('blah'))
     assert_equal('blah', Unquoted.new('blah').to_s)
     assert_equal('blah', Unquoted.new('blah').to_sql)
+    assert_equal('SELECT * FROM blah WHERE smoe = NULL', "SELECT * FROM blah WHERE smoe = #{DohDb::NULL.to_sql}")
   end
 end
 

data/test/mysql/tc_writable_row.rb

@@ -0,0 +1,22 @@
+require 'doh/unit_test'
+require 'doh/mysql/writable_row'
+
+module DohDb
+
+class TC_WritableRow < Test::Unit::TestCase
+  def test_bracket_assign_not_mutate
+    keys = []; values = []
+    row = WritableRow.new(keys, values)
+    keys.push('blah')
+    values.push('blee')
+    assert(row.keys.empty?)
+    assert(row.values.empty?)
+    row['frog'] = 'chicken'
+    assert_equal(['blah'], keys)
+    assert_equal(['blee'], values)
+    assert_equal(['frog'], row.keys)
+    assert_equal(['chicken'], row.values)
+  end
+end
+
+end

data/test/ts_core.rb

@@ -0,0 +1,4 @@
+require './core/tc_bigdecimal'
+require './core/tc_date'
+require './core/tc_string'
+require './core/tc_socket'

data/test/ts_logger.rb

@@ -0,0 +1,4 @@
+require './logger/tc_acceptor'
+require './logger/tc_event'
+require './logger/tc_formatter'
+require './logger/tc_interface'

data/test/ts_mysql.rb

@@ -0,0 +1,6 @@
+require './mysql/tc_cache_connector'
+require './mysql/tc_connector_instance'
+require './mysql/tc_handle'
+require './mysql/tc_parse'
+require './mysql/tc_readonly_row'
+require './mysql/tc_unquoted'

data/test/ts_util.rb

@@ -0,0 +1,6 @@
+require './util/tc_banking_workday'
+require './util/tc_file_edit'
+require './util/tc_jsval'
+require './util/tc_to_display'
+require './util/tc_xml_util'
+

data/test/util/slow_doh_socket.rb

@@ -0,0 +1,102 @@
+require 'test/unit'
+require 'doh/util/doh_socket'
+require 'doh/util/time_util'
+module Doh
+
+#single client timeout server
+class TimeoutServer
+  attr_accessor :shutdown
+  attr_reader :listening
+  def initialize(port)
+    @port = port
+  end
+
+  def run
+    @server = TCPServer.new(nil, @port)
+    @listening = true
+    loop do
+      while (!select([@server], nil, nil, 0.1) && !@shutdown) do
+      end
+      return if !@shutdown
+
+      session = @server.accept
+      request = session.gets
+      sleep request.to_i
+      session.write("slept: #{request.to_i} seconds -- sleeping another #{request.to_i} before sending the newline")
+      sleep request.to_i
+      session.write("\n")
+      session.close
+    end
+  end
+end
+
+class TC_slow_doh_socket < Test::Unit::TestCase
+  def test_stuff
+    server = TimeoutServer.new(3233)
+    thread = Thread.new do
+      server.run
+    end
+    while !server.listening
+    end
+
+    subtest_timeout_block
+    subtest_timed_readln
+    subtest_combination
+
+    server.shutdown = true
+    thread.join
+  end
+
+  def subtest_timeout_block
+    time_took = Doh::time_block do
+      assert_raises(TimeoutError) do
+        DohSocket::timeout_block(0.2) do
+          sock = DohSocket.new('127.0.0.1', 3233)
+          sock.write("1\n")
+          sock.timed_readln
+        end
+      end
+    end
+    assert((time_took - 0.2).abs < 0.001)
+  end
+
+  def subtest_timed_readln
+    time_took = Doh::time_block do
+      assert_raises(TimeoutError) do
+        sock = DohSocket.new('127.0.0.1', 3233)
+        sock.timeout = 0.2
+        sock.write("1\n")
+        sock.timed_readln
+      end
+    end
+    assert((time_took - 0.2).abs < 0.001)
+  end
+
+
+  def subtest_combination
+    time_took = Doh::time_block do
+      assert_raises(TimeoutError) do
+        DohSocket::timeout_block(1) do
+          time2_took = Doh::time_block do
+            assert_raises(TimeoutError) do
+              sock = DohSocket.new('127.0.0.1', 3233)
+              sock.timeout = 0.3
+              sock.write("4\n")
+              sock.timed_readln
+            end
+          end
+          assert((time2_took - 0.3).abs < 0.001)
+
+          sock = DohSocket.new('127.0.0.1', 3233)
+          sock.timeout = 1
+          sock.write("4\n")
+          sock.timed_readln
+        end
+      end
+    end
+    assert((time_took - 1).abs < 0.001)
+  end
+
+end
+
+end

data/test/util/tc_banking_workday.rb

@@ -3,6 +3,18 @@ require 'doh/util/banking_workday'
 module Doh
 
 class TC_banking_workday < Test::Unit::TestCase
+  # taken from http://www.federalreserve.gov/releases/k8/default.htm
+  @@federal_reserve_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),
+Date.new(2012, 1, 2), Date.new(2012, 1, 16), Date.new(2012, 2, 20), Date.new(2012, 5, 28), Date.new(2012, 7, 4), Date.new(2012, 9, 3), Date.new(2012, 10, 8), Date.new(2012, 11, 12), Date.new(2012, 11, 22), Date.new(2012, 12, 25),
+Date.new(2013, 1, 1), Date.new(2013, 1, 21), Date.new(2013, 2, 18), Date.new(2013, 5, 27), Date.new(2013, 7, 4), Date.new(2013, 9, 2), Date.new(2013, 10, 14), Date.new(2013, 11, 11), Date.new(2013, 11, 28), Date.new(2013, 12, 25),
+Date.new(2014, 1, 1), Date.new(2014, 1, 20), Date.new(2014, 2, 17), Date.new(2014, 5, 26), Date.new(2014, 7, 4), Date.new(2014, 9, 1), Date.new(2014, 10, 13), Date.new(2014, 11, 11), Date.new(2014, 11, 27), Date.new(2014, 12, 25)
+]
+
   def test_forward_banking_workday
     assert_equal(Date.new(2007, 2, 2).to_s, Date.new(2007, 2, 1).forward_banking_workday.to_s)
     assert_equal(Date.new(2007, 2, 5).to_s, Date.new(2007, 2, 2).forward_banking_workday.to_s)
@@ -22,6 +34,12 @@ class TC_banking_workday < Test::Unit::TestCase
   def test_banking_holiday
     assert(Date.new(2007, 2, 19).banking_holiday?)
   end
+
+  def test_all_federal_reserve_scraped
+    @@federal_reserve_holidays.each do |day|
+      assert(!day.banking_workday?)
+    end
+  end
 end
 
 end

data/test/util/tc_file_edit.rb

@@ -0,0 +1,54 @@
+require 'test/unit'
+require 'doh/util/file_edit'
+
+module Doh
+
+class TC_file_edit < Test::Unit::TestCase
+  def setup
+    @fname = '/tmp/file_edit_test.out'
+    file = File.open(@fname, 'w+')
+    @line1 = "line1 test"
+    @line2 = "test line2"
+    @line3 = "yet another line -- number 3"
+    file << @line1 << "\n"
+    file << @line2 << "\n"
+    file << @line3 << "\n"
+    file.close
+  end
+
+  def readfile
+    File.open(@fname, 'r').to_a.collect { |line| line[0..-2]}
+  end
+
+  def test_delete
+    assert_equal([@line1, @line2, @line3], readfile)
+    file = FileEdit.new(@fname)
+    file.delete
+    file.save
+    assert_equal([@line2, @line3], readfile)
+    file.move_after(/line2/)
+    file.delete
+    file.save
+    assert_equal([@line2], readfile)
+  end
+
+  def test_delete2
+    assert_equal([@line1, @line2, @line3], readfile)
+    file = FileEdit.new(@fname)
+    file.move_after(/line1/)
+    file.delete(2)
+    file.save
+    assert_equal([@line1], readfile)
+  end
+
+  def test_insert
+    file = FileEdit.new(@fname)
+    file.move_before(/line2/)
+    file.insert(@line3, @line2)
+    file.save
+    assert_equal([@line1, @line3, @line2, @line2, @line3], readfile)
+  end
+
+end
+
+end

data/test/util/tc_jsval.rb

@@ -0,0 +1,12 @@
+require 'test/unit'
+require 'doh/util/jsval'
+
+module Doh
+
+class TC_util_jsval < Test::Unit::TestCase
+  def test_jsval
+    assert_equal('new Date(2010,4,10)', Doh::jsval(Date.new(2010,5,10)))
+  end
+end
+
+end

data/test/util/tc_to_display.rb

@@ -0,0 +1,14 @@
+require 'test/unit'
+require 'doh/util/to_display'
+
+module Doh
+
+class TC_util_to_display < Test::Unit::TestCase
+  def test_to_display
+    assert_equal('', nil.to_display)
+    assert_equal('no', false.to_display)
+    assert_equal('yes', true.to_display)
+  end
+end
+
+end

data/test/util/tc_xml_util.rb

@@ -0,0 +1,17 @@
+require 'test/unit'
+require 'doh/util/xml_util'
+module Doh
+
+class TC_xml_util < Test::Unit::TestCase
+  def setup
+    @hash = {'person' => {'phones' => {'home' => '5552222', 'cell' => '5553333'}, 'name' => {'first' => 'john', 'last' => 'doe'}, 'height' => '184cm', 'weight' => '73kg'}}
+  end
+
+  def test_stuff
+    xml = Doh::hash_to_xml(@hash)
+    hash = Doh::xml_to_hash(xml)
+    assert_equal(@hash, hash)
+  end
+end
+
+end