holidays 5.7.0 → 6.0.0

data/test/defs/test_defs_ve.rb

@@ -7,22 +7,29 @@ require File.expand_path(File.dirname(__FILE__)) + '/../test_helper'
 class VeDefinitionTests < Test::Unit::TestCase  # :nodoc:
 
   def test_ve
-{
-Date.civil(2013, 1, 1) => "Año Nuevo",
-Date.civil(2013, 2, 11) => "Lunes Carnaval",
-Date.civil(2013, 2, 12) => "Martes Carnaval",
-Date.civil(2013, 3, 28) => "Jueves Santo",
-Date.civil(2013, 3, 29) => "Viernes Santo",
-Date.civil(2013, 4, 19) => "Declaración Independencia",
-Date.civil(2013, 5, 1) => "Día del Trabajador",
-Date.civil(2013, 6, 24) => "Aniversario Batalla de Carabobo",
-Date.civil(2013, 7, 5) => "Día de la Independencia",
-Date.civil(2013, 7, 24) => "Natalicio de Simón Bolívar",
-Date.civil(2013, 10, 12) => "Día de la Resistencia Indígena",
-Date.civil(2013, 12, 25) => "Día de Navidad"
-}.each do |date, name|
-  assert_equal name, (Holidays.on(date, :ve)[0] || {})[:name]
-end
+    assert_equal "Año Nuevo", (Holidays.on(Date.civil(2013, 1, 1), [:ve])[0] || {})[:name]
+
+    assert_equal "Lunes Carnaval", (Holidays.on(Date.civil(2013, 2, 11), [:ve])[0] || {})[:name]
+
+    assert_equal "Martes Carnaval", (Holidays.on(Date.civil(2013, 2, 12), [:ve])[0] || {})[:name]
+
+    assert_equal "Jueves Santo", (Holidays.on(Date.civil(2013, 3, 28), [:ve])[0] || {})[:name]
+
+    assert_equal "Viernes Santo", (Holidays.on(Date.civil(2013, 3, 29), [:ve])[0] || {})[:name]
+
+    assert_equal "Declaración Independencia", (Holidays.on(Date.civil(2013, 4, 19), [:ve])[0] || {})[:name]
+
+    assert_equal "Día del Trabajador", (Holidays.on(Date.civil(2013, 5, 1), [:ve])[0] || {})[:name]
+
+    assert_equal "Aniversario Batalla de Carabobo", (Holidays.on(Date.civil(2013, 6, 24), [:ve])[0] || {})[:name]
+
+    assert_equal "Día de la Independencia", (Holidays.on(Date.civil(2013, 7, 5), [:ve])[0] || {})[:name]
+
+    assert_equal "Natalicio de Simón Bolívar", (Holidays.on(Date.civil(2013, 7, 24), [:ve])[0] || {})[:name]
+
+    assert_equal "Día de la Resistencia Indígena", (Holidays.on(Date.civil(2013, 10, 12), [:ve])[0] || {})[:name]
+
+    assert_equal "Día de Navidad", (Holidays.on(Date.civil(2013, 12, 25), [:ve])[0] || {})[:name]
 
   end
 end

data/test/defs/test_defs_vi.rb

@@ -7,14 +7,16 @@ require File.expand_path(File.dirname(__FILE__)) + '/../test_helper'
 class ViDefinitionTests < Test::Unit::TestCase  # :nodoc:
 
   def test_vi
-{Date.civil(2014,1,1) => 'Tết dương lịch',
- Date.civil(2014,4,30) => 'Ngày Giải phóng miền Nam, thống nhất đất nước',
- Date.civil(2014,5,1) => "Ngày Quốc tế Lao động",
- Date.civil(2014,9,2) => 'Quốc khánh',
- Date.civil(2017,4,6) => "Giỗ tổ Hùng Vương",
- Date.civil(2018,3,27) => "Giỗ tổ Hùng Vương"}.each do |date, name|
-  assert_equal name, (Holidays.on(date, :vi)[0] || {})[:name]
-end
+    assert_equal "Tết dương lịch", (Holidays.on(Date.civil(2014, 1, 1), [:vi])[0] || {})[:name]
+
+    assert_equal "Ngày Giải phóng miền Nam, thống nhất đất nước", (Holidays.on(Date.civil(2014, 4, 30), [:vi])[0] || {})[:name]
+
+    assert_equal "Ngày Quốc tế Lao động", (Holidays.on(Date.civil(2014, 5, 1), [:vi])[0] || {})[:name]
+
+    assert_equal "Quốc khánh", (Holidays.on(Date.civil(2014, 9, 2), [:vi])[0] || {})[:name]
+
+    assert_equal "Giỗ tổ Hùng Vương", (Holidays.on(Date.civil(2017, 4, 6), [:vi])[0] || {})[:name]
+assert_equal "Giỗ tổ Hùng Vương", (Holidays.on(Date.civil(2018, 3, 27), [:vi])[0] || {})[:name]
 
   end
 end

data/test/defs/test_defs_za.rb

@@ -7,20 +7,29 @@ require File.expand_path(File.dirname(__FILE__)) + '/../test_helper'
 class ZaDefinitionTests < Test::Unit::TestCase  # :nodoc:
 
   def test_za
-{Date.civil(2007,1,1) => 'New Year\'s Day',
- Date.civil(2007,3,21) => 'Human Rights Day',
- Date.civil(2007,4,6) => 'Good Friday',
- Date.civil(2007,4,9) => 'Family Day',
- Date.civil(2007,4,27) => 'Freedom Day',
- Date.civil(2007,5,1) => 'Workers Day',
- Date.civil(2007,6,16) => 'Youth Day',
- Date.civil(2007,8,9) => 'National Women\'s Day',
- Date.civil(2007,9,24) => 'Heritage Day',
- Date.civil(2007,12,16) => 'Day of Reconciliation',
- Date.civil(2007,12,25) => 'Christmas Day',
- Date.civil(2007,12,26) => 'Day of Goodwill'}.each do |date, name|
-  assert_equal name, (Holidays.on(date, :za, :informal)[0] || {})[:name]
-end
+    assert_equal "New Year's Day", (Holidays.on(Date.civil(2007, 1, 1), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Human Rights Day", (Holidays.on(Date.civil(2007, 3, 21), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Good Friday", (Holidays.on(Date.civil(2007, 4, 6), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Family Day", (Holidays.on(Date.civil(2007, 4, 9), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Freedom Day", (Holidays.on(Date.civil(2007, 4, 27), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Workers Day", (Holidays.on(Date.civil(2007, 5, 1), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Youth Day", (Holidays.on(Date.civil(2007, 6, 16), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "National Women's Day", (Holidays.on(Date.civil(2007, 8, 9), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Heritage Day", (Holidays.on(Date.civil(2007, 9, 24), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Day of Reconciliation", (Holidays.on(Date.civil(2007, 12, 16), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Christmas Day", (Holidays.on(Date.civil(2007, 12, 25), [:za], [:informal])[0] || {})[:name]
+
+    assert_equal "Day of Goodwill", (Holidays.on(Date.civil(2007, 12, 26), [:za], [:informal])[0] || {})[:name]
 
   end
 end

data/test/holidays/definition/context/test_generator.rb

@@ -14,10 +14,15 @@ class GeneratorTests < Test::Unit::TestCase
       source: "some source",
     )
 
+    @test_parser = mock()
+    @test_source_generator = mock()
+
     @generator = Holidays::Definition::Context::Generator.new(
       @custom_method_parser,
       @custom_method_source_decorator,
       @custom_methods_repository,
+      @test_parser,
+      @test_source_generator,
     )
   end
 
@@ -37,6 +42,8 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
+
     regions = @generator.parse_definition_files(files)[0]
 
     assert_equal [:custom_single_file], regions
@@ -46,6 +53,8 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
+
     rules_by_month = @generator.parse_definition_files(files)[1]
 
     expected_rules_by_month = {
@@ -65,6 +74,8 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
+
     custom_methods = @generator.parse_definition_files(files)[2]
 
     expected_custom_methods = {}
@@ -75,19 +86,21 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    tests = @generator.parse_definition_files(files)[3]
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
 
-    expected_tests = [[
-        "{Date.civil(2013,6,20) => 'Company Founding'}.each do |date, name|\n  assert_equal name, (Holidays.on(date, :custom_single_file)[0] || {})[:name]\nend"
-    ]]
+    parsed_tests = @generator.parse_definition_files(files)[3]
 
-    assert_equal expected_tests, tests
+    assert_equal ["parsed tests"], parsed_tests
   end
 
   def test_generate_definition_source_correctly_generate_module_src
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
+
+    @test_source_generator.expects(:call).with('test', ['test/data/test_single_custom_holiday_defs.yaml'], ['parsed tests']).returns("test source")
+
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
     module_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[0]
 
@@ -100,18 +113,22 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
+
+    @test_source_generator.expects(:call).with('test', ['test/data/test_single_custom_holiday_defs.yaml'], ['parsed tests']).returns("test source")
+
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
     test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[1]
 
-    expected_test_src = "# encoding: utf-8\nrequire File.expand_path(File.dirname(__FILE__)) + '/../test_helper'\n\n# This file is generated by the Ruby Holiday gem.\n#\n# Definitions loaded: test/data/test_single_custom_holiday_defs.yaml\nclass TestDefinitionTests < Test::Unit::TestCase  # :nodoc:\n\n  def test_test\n{Date.civil(2013,6,20) => 'Company Founding'}.each do |date, name|\n  assert_equal name, (Holidays.on(date, :custom_single_file)[0] || {})[:name]\nend\n  end\nend\n"
-
-    assert_equal expected_test_src, test_src
+    assert_equal 'test source', test_src
   end
 
   def test_parse_definitions_files_correctly_parse_year_range_by_month
     files = ['test/data/test_custom_year_range_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2017-01-01', 'regions' => ['custom_year_range_file']}, 'expect' => {'name' => 'after_year'}}]).returns(['parsed tests'])
+
     rules_by_month = @generator.parse_definition_files(files)[1]
 
     expected_rules_by_month = {
@@ -156,6 +173,10 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_custom_year_range_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2017-01-01', 'regions' => ['custom_year_range_file']}, 'expect' => {'name' => 'after_year'}}]).returns(['parsed tests'])
+
+    @test_source_generator.expects(:call).with('test', ['test/data/test_custom_year_range_holiday_defs.yaml'], ['parsed tests']).returns('test source')
+
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
     module_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[0]
     expected_module_src = "# encoding: utf-8\nmodule Holidays\n  # This file is generated by the Ruby Holidays gem.\n  #\n  # Definitions loaded: test/data/test_custom_year_range_holiday_defs.yaml\n  #\n  # All the definitions are available at https://github.com/holidays/holidays\n  module TEST # :nodoc:\n    def self.defined_regions\n      [:custom_year_range_file]\n    end\n\n    def self.holidays_by_month\n      {\n              6 => [{:mday => 1,  :year_ranges => [{:after => 2016}],:name => \"after_year\", :regions => [:custom_year_range_file]},\n            {:mday => 2,  :year_ranges => [{:before => 2017}],:name => \"before_year\", :regions => [:custom_year_range_file]},\n            {:mday => 3,  :year_ranges => [{:between => 2016..2018}],:name => \"between_year\", :regions => [:custom_year_range_file]},\n            {:mday => 4,  :year_ranges => [{:limited => [2016, 2018, 2019]}],:name => \"limited_year\", :regions => [:custom_year_range_file]},\n            {:mday => 5,  :year_ranges => [{:before => 2015},{:after => 2017}],:name => \"multiple_conditions\", :regions => [:custom_year_range_file]}]\n      }\n    end\n\n    def self.custom_methods\n      {\n        \n      }\n    end\n  end\nend\n"
@@ -170,6 +191,10 @@ class GeneratorTests < Test::Unit::TestCase
     @custom_methods_repository.expects(:find).twice.with('custom_method(year, month)').returns(nil)
     @custom_method_source_decorator.expects(:call).once.with(@parsed_custom_method).returns("\"custom_method(year, month)\" => Proc.new { |year, month|\nsource_stuff\n}")
 
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}, {'given' => {'date' => '2015-01-01', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Custom Holiday'}}]).returns(['parsed tests'])
+
+    @test_source_generator.expects(:call).with('test', ['test/data/test_single_custom_holiday_with_custom_procs.yaml'], ['parsed tests']).returns('test source')
+
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
     module_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[0]
 
@@ -181,16 +206,18 @@ class GeneratorTests < Test::Unit::TestCase
   def test_generate_definition_source_correctly_generate_test_src_with_custom_methods
     files = ['test/data/test_single_custom_holiday_with_custom_procs.yaml']
 
-    @custom_method_parser.expects(:call).with('custom_method' => {'arguments' => 'year, month', 'source' => "d = Date.civil(year, month, 1)\nd + 2\n"}).returns({:parsed => "method"})
+    @custom_method_parser.expects(:call).with('custom_method' => {'arguments' => 'year, month', 'source' => "d = Date.civil(year, month, 1)\nd + 2\n"}).returns({"custom_method(year, month)" => @parsed_custom_method})
     @custom_methods_repository.expects(:find).twice.with('custom_method(year, month)').returns(nil)
-    @custom_method_source_decorator.expects(:call).once.with("method").returns("\"custom_method(year, month)\" => Proc.new { |year, month|\nsource_stuff\n}")
+    @custom_method_source_decorator.expects(:call).once.with(@parsed_custom_method).returns("\"custom_method(year, month)\" => Proc.new { |year, month|\nsource_stuff\n}")
+
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}, {'given' => {'date' => '2015-01-01', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Custom Holiday'}}]).returns(['parsed tests'])
+
+    @test_source_generator.expects(:call).with('test', ['test/data/test_single_custom_holiday_with_custom_procs.yaml'], ['parsed tests']).returns('test source')
 
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
     test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[1]
 
-    expected_test_src = "# encoding: utf-8\nrequire File.expand_path(File.dirname(__FILE__)) + '/../test_helper'\n\n# This file is generated by the Ruby Holiday gem.\n#\n# Definitions loaded: test/data/test_single_custom_holiday_with_custom_procs.yaml\nclass TestDefinitionTests < Test::Unit::TestCase  # :nodoc:\n\n  def test_test\n{Date.civil(2013,6,20) => 'Company Founding'}.each do |date, name|\n  assert_equal name, (Holidays.on(date, :custom_single_file)[0] || {})[:name]\nend\n\n{Date.civil(2015, 1, 1) => 'Custom Holiday'}.each do |date, name|\n  assert_equal name, (Holidays.on(date, :custom_single_file)[0] || {}[:name]\nend\n\n  end\nend\n"
-
-    assert_equal expected_test_src, test_src
+    assert_equal 'test source', test_src
   end
 
   #TODO Missing test scenarios. Adding here so I don't forget when I split this

data/test/holidays/definition/decorator/test_test.rb

@@ -0,0 +1,123 @@
+# Heh at this file name
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/decorator/test'
+require 'holidays/definition/entity/test'
+
+class DecoratorTestTests < Test::Unit::TestCase
+  def setup
+    @decorator = Holidays::Definition::Decorator::Test.new
+  end
+
+  def test_call_generates_source_from_entity_single_date
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01')],
+      :regions => [:us],
+      :name => 'Test Holiday',
+      :holiday? => true,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_equal \"Test Holiday\", (Holidays.on(Date.civil(2016, 1, 1), [:us])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_multiple_dates
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01'), DateTime.parse('2017-01-01')],
+      :regions => [:us],
+      :name => 'Test Holiday',
+      :holiday? => true
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_equal \"Test Holiday\", (Holidays.on(Date.civil(2016, 1, 1), [:us])[0] || {})[:name]\nassert_equal \"Test Holiday\", (Holidays.on(Date.civil(2017, 1, 1), [:us])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_single_date_with_options
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01')],
+      :regions => [:us],
+      :options => ['option1'],
+      :name => 'Test Holiday',
+      :holiday? => true,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_equal \"Test Holiday\", (Holidays.on(Date.civil(2016, 1, 1), [:us], [:option1])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_single_date_with_holiday_flag
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01')],
+      :regions => [:us],
+      :holiday? => false,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_nil (Holidays.on(Date.civil(2016, 1, 1), [:us])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_single_date_multiple_regions
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01')],
+      :regions => [:us, :us_ca, :ca],
+      :name => 'Test Holiday',
+      :holiday? => true,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_equal \"Test Holiday\", (Holidays.on(Date.civil(2016, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_multiple_dates_multiple_regions
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01'), DateTime.parse('2017-01-01')],
+      :regions => [:us, :us_ca, :ca],
+      :name => 'Test Holiday',
+      :holiday? => true,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_equal \"Test Holiday\", (Holidays.on(Date.civil(2016, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\nassert_equal \"Test Holiday\", (Holidays.on(Date.civil(2017, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_multiple_dates_multiple_regions_holiday_flag_false
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01'), DateTime.parse('2017-01-01')],
+      :regions => [:us, :us_ca, :ca],
+      :name => 'Test Holiday',
+      :holiday? => false,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_nil (Holidays.on(Date.civil(2016, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\nassert_nil (Holidays.on(Date.civil(2017, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_single_date_multiple_regions_holiday_flag_false
+    entity = Holidays::Definition::Entity::Test.new(
+      :dates => [DateTime.parse('2016-01-01')],
+      :regions => [:us, :us_ca, :ca],
+      :name => 'Test Holiday',
+      :holiday? => false,
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "assert_nil (Holidays.on(Date.civil(2016, 1, 1), [:us, :us_ca, :ca])[0] || {})[:name]\n"
+
+    assert_equal expected_source, source
+  end
+end

data/test/holidays/definition/generator/test_test.rb

@@ -0,0 +1,113 @@
+# Heh at this file name
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/generator/test'
+require 'holidays/definition/entity/test'
+
+class TestGeneratorTests < Test::Unit::TestCase
+  def setup
+    @module_name = "Test"
+    @files = ["file1.rb"]
+    @tests = [
+      Holidays::Definition::Entity::Test.new(
+        :dates => [DateTime.parse("2016-01-01")],
+        :regions => [:test],
+        :name => "Test Holiday",
+      ),
+    ]
+
+    @decorator = mock()
+
+    @generator = Holidays::Definition::Generator::Test.new(@decorator)
+  end
+
+  def test_generate_tests_raises_error_if_module_name_is_invalid
+    assert_raises ArgumentError do
+      @generator.call(nil, @files, @tests)
+    end
+
+    assert_raises ArgumentError do
+      @generator.call("", @files, @tests)
+    end
+  end
+
+  def test_generate_tests_raises_error_if_files_is_invalid
+    assert_raises ArgumentError do
+      @generator.call(@module_name, nil, @tests)
+    end
+
+    assert_raises ArgumentError do
+      @generator.call(@module_name, "", @tests)
+    end
+  end
+
+  def test_generate_tests_raises_error_if_tests_are_missing
+    assert_raises ArgumentError do
+      @generator.call(@module_name, @files, nil)
+    end
+  end
+
+  def test_generate_source_from_single_test
+    @decorator.expects(:call).with(@tests.first).returns("test")
+
+    expected = <<-EOT
+# encoding: utf-8
+require File.expand_path(File.dirname(__FILE__)) + '/../test_helper'
+
+# This file is generated by the Ruby Holiday gem.
+#
+# Definitions loaded: #{@files.join(', ')}
+class #{@module_name.to_s.capitalize}DefinitionTests < Test::Unit::TestCase  # :nodoc:
+
+  def test_#{@module_name.to_s.downcase}
+    test
+  end
+end
+EOT
+
+    assert_equal(expected, @generator.call(@module_name, @files, @tests))
+  end
+
+  def test_generate_source_from_multiple_tests
+    @tests = [
+      Holidays::Definition::Entity::Test.new(
+        :dates => [DateTime.parse("2016-01-01")],
+        :regions => [:test],
+        :name => "Test Holiday",
+      ),
+      Holidays::Definition::Entity::Test.new(
+        :dates => [DateTime.parse("2016-02-01")],
+        :regions => [:test2],
+        :name => "Test Holiday2",
+      ),
+      Holidays::Definition::Entity::Test.new(
+        :dates => [DateTime.parse("2016-03-01")],
+        :regions => [:test3],
+        :name => "Test Holiday3",
+      ),
+    ]
+
+    @decorator.expects(:call).with(@tests[0]).returns("test")
+    @decorator.expects(:call).with(@tests[1]).returns("test2")
+    @decorator.expects(:call).with(@tests[2]).returns("test3")
+
+    expected = <<-EOT
+# encoding: utf-8
+require File.expand_path(File.dirname(__FILE__)) + '/../test_helper'
+
+# This file is generated by the Ruby Holiday gem.
+#
+# Definitions loaded: #{@files.join(', ')}
+class #{@module_name.to_s.capitalize}DefinitionTests < Test::Unit::TestCase  # :nodoc:
+
+  def test_#{@module_name.to_s.downcase}
+    test
+    test2
+    test3
+  end
+end
+EOT
+
+    assert_equal(expected, @generator.call(@module_name, @files, @tests))
+  end
+end