51 lines
1.4 KiB
Ruby
51 lines
1.4 KiB
Ruby
#!/usr/bin/ruby
|
|
|
|
require 'rubygems'
|
|
require 'test/unit'
|
|
require File.dirname(__FILE__) + '/../classes/Scheduler.rb'
|
|
|
|
class TestScheduler < Test::Unit::TestCase
|
|
def setup
|
|
@scheduler = Scheduler.instance
|
|
end
|
|
|
|
def test_schedule
|
|
[
|
|
[
|
|
{
|
|
'start' => DateTime.parse('2010-01-01'),
|
|
'interval' => Scheduler::WEEKLY
|
|
},
|
|
2,
|
|
[ DateTime.parse('2010-01-01'), DateTime.parse('2010-01-08') ]
|
|
],
|
|
[
|
|
{
|
|
'start' => DateTime.parse('2010-01-01'),
|
|
'interval' => [ 'monday' ]
|
|
},
|
|
2,
|
|
[ DateTime.parse('2010-01-04'), DateTime.parse('2010-01-11') ]
|
|
],
|
|
[
|
|
{
|
|
'start' => DateTime.parse('2010-01-01'),
|
|
'interval' => Scheduler::DAILY,
|
|
'breaks' => [
|
|
{ 'from' => DateTime.parse('2010-01-03'), 'to' => DateTime.parse('2010-01-05') }
|
|
]
|
|
},
|
|
4,
|
|
[ DateTime.parse('2010-01-01'), DateTime.parse('2010-01-02'), DateTime.parse('2010-01-06'), DateTime.parse('2010-01-07') ]
|
|
],
|
|
].each do |parameters, to_produce, expected_results|
|
|
assert_equal expected_results, @scheduler.schedule(parameters, to_produce)
|
|
end
|
|
end
|
|
|
|
def test_skip_to_dow
|
|
assert_equal DateTime.parse('2010-01-02'), @scheduler.skip_to_dow(DateTime.parse('2010-01-01'), 6)
|
|
assert_equal DateTime.parse('2010-01-02'), @scheduler.skip_to_dow(DateTime.parse('2010-01-01'), 'saturday')
|
|
end
|
|
end
|