# encoding: UTF-8
require 'test_helper'
require 'ndr_import/universal_importer_helper'

# This tests the UniversalImporterHelper mixin
class UniversalImporterHelperTest < ActiveSupport::TestCase
  # This is a test importer class to test the excel file helper mixin
  class TestImporter
    include NdrImport::UniversalImporterHelper

    def initialize
      @table_mappings = [
        NdrImport::Table.new(:filename_pattern => /\.xls\z/i,
                             :tablename_pattern => /\Asheet1\z/i)
      ]
    end

    def get_notifier(_)
    end

    def unzip_path
      SafePath.new('test_space_rw')
    end
  end

  def setup
    @permanent_test_files = SafePath.new('permanent_test_files')
    @test_importer = TestImporter.new
  end

  test 'extract with matching mapping' do
    source_file = @permanent_test_files.join('sample_xls.xls')
    enumerator_ran = false
    @test_importer.extract(source_file) do |table, rows|
      assert_instance_of NdrImport::Table, table
      assert_instance_of Enumerator, rows
      enumerator_ran = true
    end
    assert enumerator_ran
  end

  test 'extract without matching mapping' do
    source_file = @permanent_test_files.join('sample_xlsx.xlsx')
    enumerator_ran = false
    @test_importer.extract(source_file) do |_table, _rows|
      enumerator_ran = true
    end
    refute enumerator_ran
  end

  test 'extract and transform with bespoke delimiter' do
    table_mappings = [
      NdrImport::Table.new(filename_pattern: /\.txt\z/i,
                           format: 'delimited',
                           delimiter: '¬',
                           header_lines: 1,
                           footer_lines: 0,
                           klass: 'SomeTestClass',
                           columns: [{ 'column' => 'one' },
                                     { 'column' => 'two' },
                                     { 'column' => 'three' }])
    ]
    source_file = @permanent_test_files.join('not_sign_delimited.txt')
    @test_importer.stubs(:get_table_mapping).returns(table_mappings.first)
    @test_importer.extract(source_file) do |table, rows|
      mapped_rows = table.transform(rows)

      assert_instance_of NdrImport::Table, table
      assert_instance_of Enumerator, rows
      expected_mapped_data = [{ rawtext: { 'one' => 'some', 'two' => 'data', 'three' => 'here' } },
                              { rawtext: { 'one' => 'more', 'two' => 'data', 'three' => 'here' } }]

      assert_equal expected_mapped_data, (mapped_rows.to_a.map { |_klass, fields| fields })
    end
  end

  test 'get_notifier' do
    class TestImporterWithoutNotifier
      include NdrImport::UniversalImporterHelper
    end

    assert_raise(NotImplementedError) do
      TestImporterWithoutNotifier.new.get_notifier(10_000)
    end
  end
end