diff options
| -rw-r--r-- | app/helpers/table_builder_helper/column.rb | 4 | ||||
| -rw-r--r-- | spec/helpers/table_builder_helper/column_spec.rb | 14 |
2 files changed, 16 insertions, 2 deletions
diff --git a/app/helpers/table_builder_helper/column.rb b/app/helpers/table_builder_helper/column.rb index b4c569882..5834cbc4e 100644 --- a/app/helpers/table_builder_helper/column.rb +++ b/app/helpers/table_builder_helper/column.rb @@ -35,8 +35,8 @@ module TableBuilderHelper !@link_to.nil? end - def link_to(obj) - @link_to.call(obj) + def link_to(*objs) + @link_to.call(*objs) end end diff --git a/spec/helpers/table_builder_helper/column_spec.rb b/spec/helpers/table_builder_helper/column_spec.rb index e0bfd8a6a..1366d8c46 100644 --- a/spec/helpers/table_builder_helper/column_spec.rb +++ b/spec/helpers/table_builder_helper/column_spec.rb @@ -58,5 +58,19 @@ describe TableBuilderHelper::Column do ).link_to(train) ).to eq('TGV') end + + it "takes any number of arguments" do + train = double('train', kind: 'Shinkansen') + + expect( + TableBuilderHelper::Column.new( + name: 'unused', + attribute: nil, + link_to: lambda do |prefix, train| + "#{prefix}: #{train.kind}" + end + ).link_to('Type', train) + ).to eq('Type: Shinkansen') + end end end |
