revised_holidays 9.0.0

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

@@ -0,0 +1,226 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/context/generator'
+
+class GeneratorTests < Test::Unit::TestCase
+  def setup
+    @custom_method_parser = mock()
+    @custom_method_source_decorator = mock()
+    @custom_methods_repository = mock()
+
+    @parsed_custom_method = Holidays::Definition::Entity::CustomMethod.new(
+      name: 'custom_method',
+      arguments: [:year, :month],
+      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
+
+  def test_parse_definition_files_raises_error_if_argument_is_nil
+    assert_raises ArgumentError do
+      @generator.parse_definition_files(nil)
+    end
+  end
+
+  def test_parse_definition_files_raises_error_if_files_are_empty
+    assert_raises ArgumentError do
+      @generator.parse_definition_files([])
+    end
+  end
+
+  def test_parse_definition_files_correctly_parse_regions
+    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
+  end
+
+  def test_parse_definitions_files_correctly_parse_rules_by_month
+    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 = {
+      6 => [
+        {
+          :mday    => 20,
+          :name    => "Company Founding",
+          :regions => [:custom_single_file]
+        }
+      ]
+    }
+
+    assert_equal expected_rules_by_month, rules_by_month
+  end
+
+  def test_parse_definition_files_correctly_parse_custom_methods
+    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 = {}
+    assert_equal expected_custom_methods, custom_methods
+  end
+
+  def test_parse_definition_files_correctly_parse_tests
+    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'])
+
+    parsed_tests = @generator.parse_definition_files(files)[3]
+
+    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 = @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
+
+  def test_generate_definition_source_correctly_generate_test_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).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 '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 = {
+      6 => [
+        {
+          :name => "after_year",
+          :regions => [:custom_year_range_file],
+          :mday => 1,
+          :year_ranges => {:from => 2016}
+        },
+        {
+          :name => "before_year",
+          :regions => [:custom_year_range_file],
+          :mday => 2,
+          :year_ranges => {:until => 2017}
+        },
+        {
+          :name => "between_year",
+          :regions => [:custom_year_range_file],
+          :mday => 3,
+          :year_ranges => {:between => 2016..2018 }
+        },
+        {
+          :name => "limited_year",
+          :regions => [:custom_year_range_file],
+          :mday => 4,
+          :year_ranges => {:limited => [2016,2018,2019]}
+        }
+      ]
+    }
+
+    assert_equal expected_rules_by_month, rules_by_month
+  end
+
+  def test_generate_definition_source_correctly_generate_yearrange_test_src
+    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 = @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
+
+  def test_generate_definition_source_correctly_generate_module_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({"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(@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 = @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
+
+  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({"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(@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).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)
+    test_src = @generator.generate_definition_source("test", files, regions, rules_by_month, custom_methods, tests)[1]
+
+    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/context/test_merger.rb

@@ -0,0 +1,25 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/context/merger'
+
+class MergerTests < Test::Unit::TestCase
+  def setup
+    @target_regions = [:new_region]
+    @target_holidays = {0 => [:mday => 1, :name => "Test", :regions => [:test2, :test]]}
+    @target_custom_methods = {"test_method" => Proc.new { |year| Date.civil(year, 1, 1) } }
+
+    @holidays_repo = mock()
+    @regions_repo = mock()
+    @custom_methods_repo = mock()
+
+    @subject = Holidays::Definition::Context::Merger.new(@holidays_repo, @regions_repo, @custom_methods_repo)
+  end
+
+  def test_repos_are_called_to_add_regions_and_holidays
+    @holidays_repo.expects(:add).with(@target_holidays)
+    @regions_repo.expects(:add).with(@target_regions)
+    @custom_methods_repo.expects(:add).with(@target_custom_methods)
+
+    @subject.call(@target_regions, @target_holidays, @target_custom_methods)
+  end
+end

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

@@ -0,0 +1,113 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/decorator/custom_method_proc'
+require 'holidays/definition/entity/custom_method'
+
+class DecoratorCustomMethodProcTests < Test::Unit::TestCase
+  def setup
+    @decorator = Holidays::Definition::Decorator::CustomMethodProc.new
+  end
+
+  def test_generates_lambda_from_entity
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+      source: "Date.civil(year, 1, 1)"
+    )
+
+    proc = @decorator.call(entity)
+    expected_proc = Proc.new { |year| eval("Date.civil(year, 1, 1)") }
+
+    assert_equal expected_proc.call(2015), proc.call(2015)
+  end
+
+  def test_generates_lamba_from_entity_with_multiple_lines
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+      source: "d = Date.civil(year, 1, 1)\nd + 2"
+    )
+
+    proc = @decorator.call(entity)
+    expected_proc = Proc.new { |year| eval("d = Date.civil(year, 1, 1)\nd + 2") }
+
+    assert_equal expected_proc.call(2015), proc.call(2015)
+  end
+
+  def test_generates_lamba_from_entity_with_multiple_arguments
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year", "month"],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    proc = @decorator.call(entity)
+    expected_proc = Proc.new { |year, month| eval("Date.civil(year, month, 1)") }
+
+    assert_equal expected_proc.call(2015, 6), proc.call(2015, 6)
+  end
+
+  def test_generate_returns_error_if_missing_name
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      arguments: ["year", "month"],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "",
+      arguments: ["year", "month"],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+
+  def test_generate_returns_error_if_arguments_is_missing
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: [],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+
+
+  def test_generate_returns_error_if_source_is_missing
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+      source: "",
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+end

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

@@ -0,0 +1,96 @@
+require File.expand_path(File.dirname(__FILE__)) + '/../../../test_helper'
+
+require 'holidays/definition/decorator/custom_method_source'
+require 'holidays/definition/entity/custom_method'
+
+class DecoratorCustomMethodSourceTests < Test::Unit::TestCase
+  def setup
+    @decorator = Holidays::Definition::Decorator::CustomMethodSource.new
+  end
+
+  def test_call_generates_source_from_entity
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+      source: "Date.civil(year, 1, 1)"
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "\"#{entity.name}(#{entity.arguments[0]})\" => Proc.new { |year|\n#{entity.source}}"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_with_multiple_arguments
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year", "month"],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    source = @decorator.call(entity)
+    expected_args = "#{entity.arguments[0]}, #{entity.arguments[1]}"
+
+    expected_source = "\"#{entity.name}(#{expected_args})\" => Proc.new { |#{expected_args}|\n#{entity.source}}"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_generates_source_from_entity_when_source_is_multiple_lines
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year"],
+      source: "d = Date.civil(year, 1, 1)\nd + 2"
+    )
+
+    source = @decorator.call(entity)
+    expected_source = "\"#{entity.name}(#{entity.arguments[0]})\" => Proc.new { |year|\n#{entity.source}}"
+
+    assert_equal expected_source, source
+  end
+
+  def test_call_raises_error_if_missing_name
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      arguments: ["year", "month"],
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+
+  def test_call_raises_error_if_missing_arguments
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+
+  def test_call_raises_error_if_arguments_is_not_an_array
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: "test",
+      source: "Date.civil(year, month, 1)"
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  end
+
+  def test_call_raises_error_if_missing_source
+    entity = Holidays::Definition::Entity::CustomMethod.new(
+      name: "test",
+      arguments: ["year", "month"],
+    )
+
+    assert_raises ArgumentError do
+      @decorator.call(entity)
+    end
+  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