ndr_import 3.0.0

data/test/readme_test.rb

@@ -0,0 +1,53 @@
+# encoding: UTF-8
+require 'test_helper'
+
+# This tests the README page example
+class ReadmeTest < ActiveSupport::TestCase
+  test 'readme example' do
+    require 'ndr_import/non_tabular/table'
+    require 'ndr_import/file/registry'
+
+    unzip_path = SafePath.new('test_space_rw')
+    source_file = SafePath.new('permanent_test_files').join('flat_file.pdf')
+    options = { 'unzip_path' => unzip_path }
+
+    table = NdrImport::NonTabular::Table.new(
+      'start_in_a_record' => false,
+      'end_in_a_record' => false,
+      'klass' => 'SomeTestKlass',
+      'start_line_pattern' => /\A------\z/,
+      'remove_lines' => { 'footer' => [/\A== Page \d+ of \d+ ==\z/i] },
+      'columns' => [
+        {
+          'column' => 'one',
+          'non_tabular_cell' => { 'lines' => Range.new(0, -1, true), 'capture' => /^(.*)$/i }
+        }
+      ]
+    )
+
+    # Use the Registry to enumerate over the files and their tables
+    files = NdrImport::File::Registry.files(source_file, options)
+    files.each do |filename|
+      tables = NdrImport::File::Registry.tables(filename, nil, options)
+      tables.each do |_tablename, table_content|
+        # Use the NonTabular::Table to tabulate the "table" contents
+        table.transform(table_content).each do |_klass, _fields, _index|
+          # Your code goes here
+        end
+
+        # Now we test the example
+        results = []
+        table.transform(table_content).each do |_klass, fields, _index|
+          results << fields[:rawtext]['one']
+        end
+        assert table.is_a?(NdrImport::NonTabular::Table)
+        assert_equal 4, results.count
+        assert results.first.start_with?('1')
+        assert results.last.start_with?('4')
+        assert results.any? { |result| result =~ /This is captured/ }
+        refute results.any? { |result| result =~ /This is never captured/ }
+        refute results.any? { |result| result =~ /== Page/ }
+      end
+    end
+  end
+end

data/test/resources/bomd.csv

@@ -0,0 +1,3 @@
+A,B,C,D,E,F,G,H,I,J,K,L,M,N,O,P,Q,R,S,T,U,V,W,X,Y,Z
+1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1
+2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2

data/test/resources/broken.csv

@@ -0,0 +1,3 @@
+I,really,think
+"this","is "not" "
+going,to,work

data/test/resources/filesystem_paths.yml

@@ -0,0 +1,26 @@
+<% require 'tmpdir' %>
+# This allows us different filesystem paths for different platforms
+--- 
+test_files: &test_files
+  root: <%= Dir.mktmpdir %>
+
+? !ruby/regexp /.*/
+: 
+  test_space_r:
+    <<: *test_files
+    prms:
+      - r
+  test_space_w:
+    <<: *test_files
+    prms:
+      - w
+  test_space_rw:
+    <<: *test_files
+    prms:
+      - r
+      - w
+  permanent_test_files:
+    root: <%= NdrImport.root + '/test/resources' %>
+    prms:
+      - r
+      - w

data/test/resources/flat_file.txt

@@ -0,0 +1,27 @@
+0
+This is never captured
+------
+1
+== Page 1 of 5 ==
+Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut
+labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris
+nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit
+esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt
+== Page 2 of 5 ==
+in culpa qui officia deserunt mollit anim id est laborum.
+------
+2
+This is captured
+Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
+== Page 3 of 5 ==
+------
+3
+Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.
+------
+== Page 4 of 5 ==
+4
+Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.
+------
+-1
+== Page 5 of 5 ==
+This is never captured

data/test/resources/flat_file.yml

@@ -0,0 +1,20 @@
+--- !ruby/object:NdrImport::NonTabular::Table
+# canonical_name: somename
+# filename_pattern:
+# format: pipe
+klass: SomeTestKlass
+start_line_pattern: !ruby/regexp /\A------\z/
+# end_line_pattern:
+start_in_a_record: false
+end_in_a_record: false
+remove_lines:
+  footer:
+  - !ruby/regexp /\A== Page \d+ of \d+ ==\z/i
+columns:
+- column: one
+  non_tabular_cell:
+    lines: !ruby/range
+      begin: 0
+      end: -1
+      excl: true
+    capture: !ruby/regexp /^(.*)$/i

data/test/resources/hello_utf8.txt

@@ -0,0 +1,2 @@
+Hello world
+This is a thorny þ issue!

data/test/resources/hello_windows.txt

@@ -0,0 +1,2 @@
+Hello windows world
+This is a thorny � issue!

data/test/resources/hello_world.txt

@@ -0,0 +1,2 @@
+Hello world,
+this is a text document

data/test/resources/high_ascii_delimited.txt

@@ -0,0 +1,2 @@
+345465�1234567890�Dr Bob��BLOGGS�JOE�M�05 NOV 1990�1 noddy street, anytown�CB22 3AD�IP�234534654�42�25 Jun 2015�14253�WWJ�SMITH, John�SGH�AA�LAB���Made discvery|�Dr Josh|
+345465�1234567890�Dr Bob��BLOGGS�JOE�M�05 NOV 1990�1 noddy street, anytown�CB22 3AD�IP�234534654�42�25 Jun 2015�14253�WWJ�SMITH, John�SGH�AA�LAB���Made discvery|�Dr Josh|

data/test/resources/malformed.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0"?>
+<root>
+  <note><![CDATA[
+    This is 
 a note!
+  ]]></note>
+</root>

data/test/resources/normal.csv

@@ -0,0 +1,3 @@
+A,B,C,D,E,F,G,H,I,J,K,L,M,N,O,P,Q,R,S,T,U,V,W,X,Y,Z
+1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1
+2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2

data/test/resources/normal_pipe.csv

@@ -0,0 +1,3 @@
+A|B|C|D|E|F|G|H|I|J|K|L|M|N|O|P|Q|R|S|T|U|V|W|X|Y|Z
+1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1|1
+2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2|2

data/test/resources/normal_thorn.csv

@@ -0,0 +1,3 @@
+A�B�C�D�E�F�G�H�I�J�K�L�M�N�O�P�Q�R�S�T�U�V�W�X�Y�Z
+1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1�1
+2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2�2

data/test/resources/standard_mappings.yml

@@ -0,0 +1,39 @@
+---
+surname:
+  column: surname
+  rawtext_name: surname
+  mappings:
+  - field: surname
+    clean: :name
+previoussurname:
+  column: previoussurname
+  rawtext_name: previoussurname
+  mappings:
+  - field: previoussurname
+    clean: :name
+forenames:
+  column: forenames
+  rawtext_name: forenames
+  mappings:
+  - field: forenames
+    clean: :name
+sex:
+  column: sex
+  rawtext_name: sex
+  mappings:
+  - field: sex
+    clean: :sex
+nhsnumber:
+  column: nhsnumber
+  rawtext_name: nhsnumber
+  mappings:
+  - field: nhsnumber
+    clean: :nhsnumber
+postcode:
+  column: postcode
+  rawtext_name: postcode
+  mappings:
+  - field: postcode
+    clean: :postcode
+test:
+  column: standard_mapping_column_name

data/test/resources/txt_file_xls_extension.xls

@@ -0,0 +1 @@
+Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.

data/test/resources/txt_file_xlsx_extension.xlsx

@@ -0,0 +1 @@
+Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.

data/test/resources/utf-8_xml.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0"?>
+<root>
+  <note id="alpha">
+    <letter>α</letter>
+  </note>
+  <note id="beta">
+    <letter>β</letter>
+  </note>
+</root>

data/test/resources/windows-1252_xml.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0"?>
+<root>
+  <note id="apostrophe">
+    <letter>�</letter>
+  </note>
+  <note id="dash">
+    <letter>�</letter>
+  </note>
+</root>

data/test/resources/windows.csv

@@ -0,0 +1,5 @@
+NHS NUMBER�,PERSON BIRTH DATE,POSTCODE OF USUAL ADDRESS (AT DIAGNOSIS),PERSON FAMILY NAME,PERSON GIVEN NAME,PATIENT USUAL ADDRESS (AT DIAGNOSIS),ORGANISATION CODE (CODE OF PROVIDER),MULTIDISCIPLINARY TEAM DISCUSSION DATE (CANCER),CONSULTANT CODE (FIRST SEEN),DATE OF DIAGNOSIS   (CLINICALLY AGREED),PERFORMANCE STATUS (ADULT) ,ASA SCORE (PROSTATE),SOURCE OF REFERRAL FOR OUT-PATIENTS ,PSA (DIAGNOSIS),PROSTATE BIOPSY TECHNIQUE,GLEASON GRADE (PRIMARY),GLEASON GRADE (SECONDARY),GLEASON GRADE (TERTIARY),MULTIPARAMETRIC MRI PERFORMED,T CATEGORY (FINAL PRE-TREATMENT),N CATEGORY (FINAL PRE-TREATMENT),M CATEGORY (FINAL PRE-TREATMENT),PERINEURAL INVASION,"NUMBER OF POSITIVE CORES 
+
+",TOTAL NUMBER,GREATEST PERCENTAGE OF CANCER IN SINGLE MOST INVOLVED CORE,"SPECIALIST REFERRAL
+APPOINTMENT
+",ORGANISATION SITE CODE (PROVIDER TREATMENT START DATE (CANCER) ,CONSULTANT CODE (TREATMENT),TYPE OF RADICAL PROSTATECTOMY (ACTUAL),PROCEDURE DATE,PROCEDURE � NERVE SPARING,T CATEGORY (PATHOLOGICAL),N CATEGORY (PATHOLOGICAL),ORGAN CONFINED,SEMINAL VESICLES INVASION,RADICAL PROSTATECTOMY MARGIN STATUS,NUMBER OF NODES EXAMINED,NUMBER OF NODES POSITIVE,PLANNED RADIOTHERAPY  INTENT (PROSTATE),PLANNED RADIOTHERAPY TYPE,PLANNED TYPE OF IMAGE-GUIDANCE FOR EXTERNAL BEAM RADIOTHERAPY,PLANNED RADIOTHERAPY FIELD,PLANNED BRACHYTHERAPY TYPE,PLANNED BRACHYTHERAPY TOTAL DOSE,PLANNED BRACHYTHERAPY TOTAL FRACTIONS,PLANNED DURATION OF NEOADJUVANT ANDROGEN DEPRIVATION THERAPY,PLANNED TOTAL DURATION OF ADJUVANT ANDROGEN DEPRIVATION THERAPY

data/test/standard_mappings_test.rb

@@ -0,0 +1,22 @@
+# encoding: UTF-8
+require 'test_helper'
+
+# This tests the StandardMappings configuration class
+class StandardMappingsTest < ActiveSupport::TestCase
+  test 'should not raise exception on reconfiguring NdrImport::StandardMappings' do
+    NdrImport::StandardMappings.mappings =
+      YAML.load_file(SafePath.new('permanent_test_files').join('standard_mappings.yml'))
+  end
+
+  test 'should raise exception on setting non-hash standard mappings' do
+    assert_raise ArgumentError do
+      NdrImport::StandardMappings.mappings = true
+    end
+  end
+
+  test 'should return mappings' do
+    safe_path = SafePath.new('permanent_test_files').join('standard_mappings.yml')
+    assert_instance_of Hash, NdrImport::StandardMappings.mappings
+    assert_equal YAML.load_file(safe_path), NdrImport::StandardMappings.mappings
+  end
+end

data/test/table_test.rb

@@ -0,0 +1,288 @@
+require 'test_helper'
+
+# This tests the NdrImport::Table mapping class
+class TableTest < ActiveSupport::TestCase
+  def test_deserialize_table
+    table = simple_deserialized_table
+    assert_instance_of NdrImport::Table, table
+    assert_equal 2, table.header_lines
+    assert_equal 1, table.footer_lines
+    assert_equal 'pipe', table.format
+    assert_equal 'SomeTestKlass', table.klass
+    assert_equal 'somename', table.canonical_name
+    assert_equal [{ 'column' => 'one' }, { 'column' => 'two' }, { 'column' => 'three' }],
+                 table.columns
+  end
+
+  def test_initialize
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1,
+                                 :format => 'pipe', :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+    assert_instance_of NdrImport::Table, table
+    assert_equal 2, table.header_lines
+    assert_equal 1, table.footer_lines
+    assert_equal 'pipe', table.format
+    assert_equal 'SomeTestKlass', table.klass
+    assert_equal [{ 'column' => 'one' }, { 'column' => 'two' }], table.columns
+  end
+
+  def test_should_raise_error_on_invalid_initialization
+    # incorrect parameter type
+    assert_raises ArgumentError do
+      NdrImport::Table.new([])
+    end
+    # invalid option
+    assert_raises ArgumentError do
+      NdrImport::Table.new(:potato => true)
+    end
+  end
+
+  def test_match_with_no_patterns
+    table = NdrImport::Table.new
+    assert table.match('example.csv', nil)
+    assert table.match('example.xslx', 'Sheet1')
+  end
+
+  def test_match_with_only_filename_pattern
+    table = NdrImport::Table.new(:filename_pattern => /\.(csv|xlsx)\z/i)
+    assert table.match('example.csv', nil)
+    assert table.match('example.xlsx', 'Sheet1')
+
+    table = NdrImport::Table.new(:filename_pattern => /\Ademo\.(csv|xlsx)\z/i)
+    refute table.match('example.csv', nil)
+    refute table.match('example.xlsx', 'Sheet1')
+  end
+
+  def test_match_with_both_patterns
+    table = NdrImport::Table.new(:filename_pattern => /\.xlsx\z/i,
+                                 :tablename_pattern => /\Asheet1\z/i)
+    assert table.match('example.xlsx', 'Sheet1')
+    refute table.match('example.xlsx', 'Sheet2')
+  end
+
+  def test_transform
+    lines = [%w(ONE TWO), %w(CARROT POTATO), %w(BACON SAUSAGE)].each
+    table = NdrImport::Table.new(:header_lines => 1, :footer_lines => 0,
+                                 :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+
+    output = []
+    table.transform(lines).each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    expected_output = [
+      ['SomeTestKlass', { :rawtext => { 'one' => 'CARROT', 'two' => 'POTATO' } }, 1],
+      ['SomeTestKlass', { :rawtext => { 'one' => 'BACON', 'two' => 'SAUSAGE' } }, 2]
+    ]
+    assert_equal expected_output, output
+  end
+
+  def test_process_line
+    # No header row, process the first line
+    table = NdrImport::Table.new(:header_lines => 0, :footer_lines => 0,
+                                 :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+
+    output = []
+    table.process_line(%w(CARROT POTATO)).each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    expected_output = [
+      ['SomeTestKlass', { :rawtext => { 'one' => 'CARROT', 'two' => 'POTATO' } }, 0]
+    ]
+    assert_equal expected_output.sort, output.sort
+
+    # One header row, don't process the first line
+    table = NdrImport::Table.new(:header_lines => 1, :footer_lines => 0,
+                                 :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+
+    output = []
+    table.process_line(%w(CARROT POTATO)).each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    assert_equal [], output
+  end
+
+  def test_transform_line
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1,
+                                 :columns => column_level_klass_mapping)
+    enum = table.transform_line(%w(CARROT POTATO PEA), 7)
+    assert_instance_of Enumerator, enum
+
+    output = []
+    enum.each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    expected_output = [
+      ['SomeTestKlass', { :rawtext => { 'one' => 'CARROT', 'two' => 'POTATO' } }, 7],
+      ['SomeOtherKlass', { :rawtext => { 'two' => 'POTATO', 'three' => 'PEA' } }, 7]
+    ]
+    assert_equal expected_output.sort, output.sort
+  end
+
+  def test_encode_with
+    # encode_with(coder)
+  end
+
+  def test_skip_footer_lines
+    table = simple_deserialized_table
+    lines = (1..10).each
+    assert_equal((1..7).to_a, table.send(:skip_footer_lines, lines, 3).to_a)
+    assert_equal((1..10).to_a, table.send(:skip_footer_lines, lines, 0).to_a)
+  end
+
+  def test_masked_mappings
+    # table level
+    table = simple_deserialized_table
+    table_level_klass_masked_mappings = {
+      'SomeTestKlass' => [{ 'column' => 'one' }, { 'column' => 'two' }, { 'column' => 'three' }]
+    }
+    assert_equal table_level_klass_masked_mappings, table.send(:masked_mappings)
+
+    # column level
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1,
+                                 :columns => column_level_klass_mapping)
+
+    assert_equal column_level_klass_masked_mappings, table.send(:masked_mappings)
+  end
+
+  def test_column_level_klass_masked_mappings
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1,
+                                 :columns => column_level_klass_mapping)
+
+    assert_equal column_level_klass_masked_mappings,
+                 table.send(:column_level_klass_masked_mappings)
+  end
+
+  def test_ensure_mappings_define_klass
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1, :columns => [
+      { 'column' => 'one', 'klass' => 'SomeTestKlass' },
+      { 'column' => 'two' }
+    ])
+    assert_raise(RuntimeError) { table.send(:ensure_mappings_define_klass) }
+
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1, :columns => [
+      { 'column' => 'one', 'klass' => 'SomeTestKlass' },
+      { 'column' => 'two', 'klass' => 'SomeOtherKlass' }
+    ])
+    table.send(:ensure_mappings_define_klass)
+  end
+
+  def test_mask_mappings_by_klass
+    table = NdrImport::Table.new(:header_lines => 2, :footer_lines => 1,
+                                 :columns => column_level_klass_mapping)
+    some_test_klass_mapping = [
+      { 'column' => 'one', 'klass' => 'SomeTestKlass' },
+      { 'column' => 'two', 'klass' => %w(SomeTestKlass SomeOtherKlass) },
+      { 'do_not_capture' => true }
+    ]
+    assert_equal some_test_klass_mapping,
+                 table.send(:mask_mappings_by_klass, 'SomeTestKlass')
+
+    some_other_klass_mapping = [
+      { 'do_not_capture' => true },
+      { 'column' => 'two', 'klass' => %w(SomeTestKlass SomeOtherKlass) },
+      { 'column' => 'three', 'klass' => 'SomeOtherKlass' }
+    ]
+    assert_equal some_other_klass_mapping,
+                 table.send(:mask_mappings_by_klass, 'SomeOtherKlass')
+  end
+
+  def test_valid_single_line_header
+    lines = [
+      %w(ONE TWO),
+      %w(CARROT POTATO),
+      %w(BACON SAUSAGE)
+    ].each
+
+    table = NdrImport::Table.new(:header_lines => 1, :footer_lines => 0,
+                                 :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+
+    output = []
+    table.transform(lines).each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    expected_output = [
+      ['SomeTestKlass', { :rawtext => { 'one' => 'CARROT', 'two' => 'POTATO' } }, 1],
+      ['SomeTestKlass', { :rawtext => { 'one' => 'BACON', 'two' => 'SAUSAGE' } }, 2]
+    ]
+    assert table.header_valid?
+    assert_equal expected_output, output
+  end
+
+  def test_valid_multi_line_header
+    lines = [
+      %w(NOTHEADING1 NOTHEADING2),
+      %w(ONE TWO),
+      %w(DEFINITELYNOTHEADING1 DEFINITELYNOTHEADING2),
+      %w(CARROT POTATO),
+      %w(BACON SAUSAGE)
+    ].each
+
+    table = NdrImport::Table.new(:header_lines => 3, :footer_lines => 0,
+                                 :klass => 'SomeTestKlass',
+                                 :columns => [{ 'column' => 'one' }, { 'column' => 'two' }])
+
+    output = []
+    table.transform(lines).each do |klass, fields, index|
+      output << [klass, fields, index]
+    end
+
+    expected_output = [
+      ['SomeTestKlass', { :rawtext => { 'one' => 'CARROT', 'two' => 'POTATO' } }, 3],
+      ['SomeTestKlass', { :rawtext => { 'one' => 'BACON', 'two' => 'SAUSAGE' } }, 4]
+    ]
+    assert table.header_valid?
+    assert_equal expected_output, output
+  end
+
+  private
+
+  def simple_deserialized_table
+    Psych.load <<YML
+--- !ruby/object:NdrImport::Table
+canonical_name: somename
+# filename_pattern: !ruby/regexp //
+header_lines: 2
+footer_lines: 1
+format: pipe
+klass: SomeTestKlass
+# non_tabular_row:
+#   ...
+columns:
+- column: one
+- column: two
+- column: three
+YML
+  end
+
+  def column_level_klass_mapping
+    [
+      { 'column' => 'one', 'klass' => 'SomeTestKlass' },
+      { 'column' => 'two', 'klass' => %w(SomeTestKlass SomeOtherKlass) },
+      { 'column' => 'three', 'klass' => 'SomeOtherKlass' }
+    ]
+  end
+
+  def column_level_klass_masked_mappings
+    {
+      'SomeTestKlass' => [
+        { 'column' => 'one', 'klass' => 'SomeTestKlass' },
+        { 'column' => 'two', 'klass' => %w(SomeTestKlass SomeOtherKlass) },
+        { 'do_not_capture' => true }
+      ],
+      'SomeOtherKlass' => [
+        { 'do_not_capture' => true },
+        { 'column' => 'two', 'klass' => %w(SomeTestKlass SomeOtherKlass) },
+        { 'column' => 'three', 'klass' => 'SomeOtherKlass' }
+      ]
+    }
+  end
+end