holidays 4.2.0 → 8.4.1

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

@@ -14,10 +14,18 @@ class GeneratorTests < Test::Unit::TestCase
       source: "some source",
     )
 
+    @test_parser = mock()
+    @test_source_generator = mock()
+
+    @module_source_generator = mock()
+
     @generator = Holidays::Definition::Context::Generator.new(
       @custom_method_parser,
       @custom_method_source_decorator,
       @custom_methods_repository,
+      @test_parser,
+      @test_source_generator,
+      @module_source_generator,
     )
   end
 
@@ -37,7 +45,9 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
+    @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
   end
@@ -46,7 +56,9 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
+    @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 = {
       6 => [
@@ -65,7 +77,9 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
+    @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 = {}
     assert_equal expected_custom_methods, custom_methods
@@ -75,23 +89,26 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
+    @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'])
+
+    @module_source_generator.expects(:call).with("test", ["test/data/test_single_custom_holiday_defs.yaml"], [:custom_single_file], ["      6 => [{:mday => 20, :name => \"Company Founding\", :regions => [:custom_single_file]}]"], "").returns("module source")
+    @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, test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)
+    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_single_custom_holiday_defs.yaml\n  #\n  # To use the definitions in this file, load it right after you load the\n  # Holiday gem:\n  #\n  #   require 'holidays'\n  #   require 'generated_definitions/test'\n  #\n  # All the definitions are available at https://github.com/holidays/holidays\n  module TEST # :nodoc:\n    def self.defined_regions\n      [:custom_single_file]\n    end\n\n    def self.holidays_by_month\n      {\n              6 => [{:mday => 20, :name => \"Company Founding\", :regions => [:custom_single_file]}]\n      }\n    end\n\n    def self.custom_methods\n      {\n        \n      }\n    end\n  end\nend\n"
+    expected_module_src = "module source"
 
     assert_equal expected_module_src, module_src
   end
@@ -100,19 +117,25 @@ class GeneratorTests < Test::Unit::TestCase
     files = ['test/data/test_single_custom_holiday_defs.yaml']
     @custom_method_parser.expects(:call).with(nil).returns({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
-    module_src, test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)
+    @test_parser.expects(:call).with([{'given' => {'date' => '2013-06-20', 'regions' => ['custom_single_file']}, 'expect' => {'name' => 'Company Founding'}}]).returns(['parsed tests'])
 
-    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"
+    @module_source_generator.expects(:call).returns("module_source")
+
+    @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]
 
-    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({})
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
+    @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 = {
       6 => [
@@ -120,31 +143,25 @@ class GeneratorTests < Test::Unit::TestCase
           :name => "after_year",
           :regions => [:custom_year_range_file],
           :mday => 1,
-          :year_ranges => [{"after" => 2016}]
+          :year_ranges => {:from => 2016}
         },
         {
           :name => "before_year",
           :regions => [:custom_year_range_file],
           :mday => 2,
-          :year_ranges => [{"before" => 2017}]
+          :year_ranges => {:until => 2017}
         },
         {
           :name => "between_year",
           :regions => [:custom_year_range_file],
           :mday => 3,
-          :year_ranges => [{"between" => "2016..2018"}]
+          :year_ranges => {:between => 2016..2018 }
         },
         {
           :name => "limited_year",
           :regions => [:custom_year_range_file],
           :mday => 4,
-          :year_ranges => [{"limited" => [2016,2018,2019]}]
-        },
-        {
-          :name => "multiple_conditions",
-          :regions => [:custom_year_range_file],
-          :mday => 5,
-          :year_ranges => [{"before" => 2015}, {"after" => 2017}]
+          :year_ranges => {:limited => [2016,2018,2019]}
         }
       ]
     }
@@ -156,9 +173,15 @@ 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')
+
+    @module_source_generator.expects(:call).with("test", ["test/data/test_custom_year_range_holiday_defs.yaml"], [:custom_year_range_file], ["      6 => [{:mday => 1, :year_ranges => { :from => 2016 },:name => \"after_year\", :regions => [:custom_year_range_file]},\n            {:mday => 2, :year_ranges => { :until => 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]}]"], "").returns('module_source')
+
     regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
-    module_src, test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)
-    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  # To use the definitions in this file, load it right after you load the\n  # Holiday gem:\n  #\n  #   require 'holidays'\n  #   require 'generated_definitions/test'\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"
+    module_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[0]
+    expected_module_src = "module_source"
 
     assert_equal expected_module_src, module_src
   end
@@ -170,10 +193,15 @@ 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'])
+
+    @module_source_generator.expects(:call).with("test", ["test/data/test_single_custom_holiday_with_custom_procs.yaml"], [:custom_single_file], ["      0 => [{:function => \"custom_method(year, month)\", :function_arguments => [:year, :month], :function_modifier => 5, :name => \"Custom Holiday\", :regions => [:custom_single_file]}]", "      6 => [{:mday => 20, :name => \"Company Founding\", :regions => [:custom_single_file]}]"], "\"custom_method(year, month)\" => Proc.new { |year, month|\nsource_stuff\n},\n\n").returns("module source")
+    @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, test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)
+    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_single_custom_holiday_with_custom_procs.yaml\n  #\n  # To use the definitions in this file, load it right after you load the\n  # Holiday gem:\n  #\n  #   require 'holidays'\n  #   require 'generated_definitions/test'\n  #\n  # All the definitions are available at https://github.com/holidays/holidays\n  module TEST # :nodoc:\n    def self.defined_regions\n      [:custom_single_file]\n    end\n\n    def self.holidays_by_month\n      {\n              0 => [{:function => \"custom_method(year, month)\", :function_arguments => [:year, :month], :function_modifier => 5, :name => \"Custom Holiday\", :regions => [:custom_single_file]}],\n      6 => [{:mday => 20, :name => \"Company Founding\", :regions => [:custom_single_file]}]\n      }\n    end\n\n    def self.custom_methods\n      {\n        \"custom_method(year, month)\" => Proc.new { |year, month|\nsource_stuff\n},\n\n\n      }\n    end\n  end\nend\n"
+    expected_module_src = "module source"
 
     assert_equal expected_module_src, module_src
   end
@@ -181,15 +209,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}")
 
-    regions, rules_by_month, custom_methods, tests = @generator.parse_definition_files(files)
-    module_src, test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)
+    @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'])
+
+    @module_source_generator.expects(:call).returns("module_source")
+    @test_source_generator.expects(:call).with('test', ['test/data/test_single_custom_holiday_with_custom_procs.yaml'], ['parsed tests']).returns('test source')
 
-    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"
+    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]
 
-    assert_equal expected_test_src, test_src
+    assert_equal 'test source', test_src
   end
 end

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

@@ -0,0 +1,37 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/context/load'
+
+class LoadTests < Test::Unit::TestCase
+  def setup
+    @definition_merger = mock()
+    full_definitions_path = File.expand_path(File.dirname(__FILE__)) + '/../../../data'
+
+    @subject = Holidays::Definition::Context::Load.new(
+      @definition_merger,
+      full_definitions_path,
+    )
+  end
+
+  def test_region_is_found_and_loaded_and_merged
+    @definition_merger.expects(:call).with(:test_region, {}, {})
+    @subject.call(:test_region)
+  end
+
+  def test_region_file_not_found
+    assert_raises Holidays::UnknownRegionError do
+      @subject.call(:unknown)
+    end
+  end
+
+  def test_region_can_be_loaded_but_file_is_invalid
+    assert_raises Holidays::UnknownRegionError do
+      @subject.call(:test_invalid_region)
+    end
+  end
+
+  def test_returns_list_of_loaded_regions
+    @definition_merger.expects(:call).with(:test_region, {}, {})
+    assert_equal([:test_region, :test_region2], @subject.call(:test_region), "Should cache subregions under the parent region's name")
+  end
+end

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_module.rb

@@ -0,0 +1,268 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/generator/module'
+
+class GeneratorModuleTests < Test::Unit::TestCase
+  def setup
+    @generator = Holidays::Definition::Generator::Module.new
+
+    @module_name = "TEST"
+    @files = ["file1.rb", "file2.rb"]
+    @regions = [:test, :test2]
+    @month_strings = ["first-string", "second-string"]
+    @custom_methods = "custom-methods"
+  end
+
+  def subject
+    @generator.call(
+      @module_name,
+      @files,
+      @regions,
+      @month_strings,
+      @custom_methods,
+    )
+  end
+
+  def test_generates_source
+    expected = <<-EOF
+# encoding: utf-8
+module Holidays
+  # This file is generated by the Ruby Holidays gem.
+  #
+  # Definitions loaded: file1.rb, file2.rb
+  #
+  # All the definitions are available at https://github.com/holidays/holidays
+  module TEST # :nodoc:
+    def self.defined_regions
+      [:test, :test2]
+    end
+
+    def self.holidays_by_month
+      {
+          first-string,
+second-string
+      }
+    end
+
+    def self.custom_methods
+      {
+          custom-methods
+      }
+    end
+  end
+end
+EOF
+
+    assert_equal expected, subject
+  end
+
+  def test_module_name_more_than_one_word
+    @module_name = "TESTWITHHYPHEN"
+
+    expected = <<-EOF
+# encoding: utf-8
+module Holidays
+  # This file is generated by the Ruby Holidays gem.
+  #
+  # Definitions loaded: file1.rb, file2.rb
+  #
+  # All the definitions are available at https://github.com/holidays/holidays
+  module TESTWITHHYPHEN # :nodoc:
+    def self.defined_regions
+      [:test, :test2]
+    end
+
+    def self.holidays_by_month
+      {
+          first-string,
+second-string
+      }
+    end
+
+    def self.custom_methods
+      {
+          custom-methods
+      }
+    end
+  end
+end
+EOF
+
+    assert_equal expected, subject
+  end
+
+  def test_generates_source_with_blank_custom_methods
+    @custom_methods = ""
+
+    expected = <<-EOF
+# encoding: utf-8
+module Holidays
+  # This file is generated by the Ruby Holidays gem.
+  #
+  # Definitions loaded: file1.rb, file2.rb
+  #
+  # All the definitions are available at https://github.com/holidays/holidays
+  module TEST # :nodoc:
+    def self.defined_regions
+      [:test, :test2]
+    end
+
+    def self.holidays_by_month
+      {
+          first-string,
+second-string
+      }
+    end
+
+    def self.custom_methods
+      {
+          
+      }
+    end
+  end
+end
+EOF
+
+    assert_equal expected, subject
+  end
+
+  def test_generates_source_with_nil_custom_methods
+    @custom_methods = nil
+
+    expected = <<-EOF
+# encoding: utf-8
+module Holidays
+  # This file is generated by the Ruby Holidays gem.
+  #
+  # Definitions loaded: file1.rb, file2.rb
+  #
+  # All the definitions are available at https://github.com/holidays/holidays
+  module TEST # :nodoc:
+    def self.defined_regions
+      [:test, :test2]
+    end
+
+    def self.holidays_by_month
+      {
+          first-string,
+second-string
+      }
+    end
+
+    def self.custom_methods
+      {
+          
+      }
+    end
+  end
+end
+EOF
+
+    assert_equal expected, subject
+  end
+
+  def test_generates_source_with_symbol_module_name
+    @module_name = :test
+
+    expected = <<-EOF
+# encoding: utf-8
+module Holidays
+  # This file is generated by the Ruby Holidays gem.
+  #
+  # Definitions loaded: file1.rb, file2.rb
+  #
+  # All the definitions are available at https://github.com/holidays/holidays
+  module TEST # :nodoc:
+    def self.defined_regions
+      [:test, :test2]
+    end
+
+    def self.holidays_by_month
+      {
+          first-string,
+second-string
+      }
+    end
+
+    def self.custom_methods
+      {
+          custom-methods
+      }
+    end
+  end
+end
+EOF
+
+    assert_equal expected, subject
+  end
+
+  def test_raise_error_if_module_name_is_blank
+    @module_name = ""
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_module_name_is_nil
+    @module_name = nil
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_files_are_empty
+    @files = []
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_files_are_not_all_strings
+    @files = [:test, 1]
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_files_is_nil
+    @files = nil
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_regions_are_empty
+    @regions = []
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_regions_is_nil
+    @regions = nil
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_month_strings_are_empty
+    @month_strings = []
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+
+  def test_raise_error_if_month_strings_is_nil
+    @month_strings = nil
+
+    assert_raises ArgumentError do
+      subject
+    end
+  end
+end