Class: Cucumber::Ast::Table

Inherits:
Object show all
Includes:
Enumerable, Gherkin::Rubify
Defined in:
lib/cucumber/ast/table.rb

Overview

Step Definitions that match a plain text Step with a multiline argument table will receive it as an instance of Table. A Table object holds the data of a table parsed from a feature file and lets you access and manipulate the data in different ways.

For example:

Given I have:
  | a | b |
  | c | d |

And a matching StepDefinition:

Given /I have:/ do |table|
  data = table.raw
end

This will store [['a', 'b'], ['c', 'd']] in the data variable.

Direct Known Subclasses

OutlineTable

Defined Under Namespace

Classes: Builder, Cell, Cells, Different, SurplusCell

Constant Summary collapse

NULL_CONVERSIONS = 
Hash.new(lambda{ |cell_value| cell_value }).freeze

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(raw, conversion_procs = NULL_CONVERSIONS.dup) ⇒ Table

Creates a new instance. raw should be an Array of Array of String or an Array of Hash (similar to what #hashes returns). You don’t typically create your own Table objects - Cucumber will do it internally and pass them to your Step Definitions.



74
75
76
77
78
79
80
81
82
83
 
# File 'lib/cucumber/ast/table.rb', line 74

def initialize(raw, conversion_procs = NULL_CONVERSIONS.dup)
  @cells_class = Cells
  @cell_class = Cell

  raw = ensure_array_of_array(rubify(raw))
  # Verify that it's square
  transposed = raw.transpose
  create_cell_matrix(raw)
  @conversion_procs = conversion_procs
end

Instance Attribute Details

#file ⇒ Object

Returns the value of attribute file.



56
57
58
 
# File 'lib/cucumber/ast/table.rb', line 56

def file
  @file
end

Class Method Details

.default_arg_name ⇒ Object

:nodoc:



58
59
60
 
# File 'lib/cucumber/ast/table.rb', line 58

def self.default_arg_name #:nodoc:
  "table"
end

.parse(text, uri, offset) ⇒ Object 



62
63
64
65
66
67
 
# File 'lib/cucumber/ast/table.rb', line 62

def self.parse(text, uri, offset)
  builder = Builder.new
  lexer = Gherkin::Lexer::I18nLexer.new(builder)
  lexer.scan(text)
  new(builder.rows)
end

Instance Method Details

#accept(visitor) ⇒ Object

:nodoc:



180
181
182
183
184
185
186
 
# File 'lib/cucumber/ast/table.rb', line 180

def accept(visitor) #:nodoc:
  return if Cucumber.wants_to_quit
  cells_rows.each do |row|
    visitor.visit_table_row(row)
  end
  nil
end

#arguments_replaced(arguments) ⇒ Object

:nodoc:



407
408
409
410
411
412
413
414
415
416
417
418
419
420
 
# File 'lib/cucumber/ast/table.rb', line 407

def arguments_replaced(arguments) #:nodoc:
  raw_with_replaced_args = raw.map do |row|
    row.map do |cell|
      cell_with_replaced_args = cell
      arguments.each do |name, value|
        if cell_with_replaced_args && cell_with_replaced_args.include?(name)
          cell_with_replaced_args = value ? cell_with_replaced_args.gsub(name, value) : nil
        end
      end
      cell_with_replaced_args
    end
  end
  Table.new(raw_with_replaced_args)
end

#cell_matrix ⇒ Object

:nodoc:



440
441
442
 
# File 'lib/cucumber/ast/table.rb', line 440

def cell_matrix #:nodoc:
  @cell_matrix
end

#cells_rows ⇒ Object

:nodoc:



426
427
428
429
430
 
# File 'lib/cucumber/ast/table.rb', line 426

def cells_rows #:nodoc:
  @rows ||= cell_matrix.map do |cell_row|
    @cells_class.new(self, cell_row)
  end
end

#col_width(col) ⇒ Object

:nodoc:



444
445
446
 
# File 'lib/cucumber/ast/table.rb', line 444

def col_width(col) #:nodoc:
  columns[col].__send__(:width)
end

#column_names ⇒ Object

:nodoc:



167
168
169
 
# File 'lib/cucumber/ast/table.rb', line 167

def column_names	#:nodoc:
  @col_names ||= cell_matrix[0].map { |cell| cell.value }
end

#diff!(other_table, options = {}) ⇒ Object

Compares other_table to self. If other_table contains columns and/or rows that are not in self, new columns/rows are added at the relevant positions, marking the cells in those rows/columns as surplus. Likewise, if other_table lacks columns and/or rows that are present in self, these are marked as missing.

surplus and missing cells are recognised by formatters and displayed so that it’s easy to read the differences.

Cells that are different, but look identical (for example the boolean true and the string “true”) are converted to their Object#inspect representation and preceded with (i) - to make it easier to identify where the difference actually is.

Since all tables that are passed to StepDefinitions always have String objects in their cells, you may want to use #map_column! before calling #diff!. You can use #map_column! on either of the tables.

A Different error is raised if there are missing rows or columns, or surplus rows. An error is not raised for surplus columns. Whether to raise or not raise can be changed by setting values in options to true or false:

  • missing_row : Raise on missing rows (defaults to true)

  • surplus_row : Raise on surplus rows (defaults to true)

  • missing_col : Raise on missing columns (defaults to true)

  • surplus_col : Raise on surplus columns (defaults to false)

The other_table argument can be another Table, an Array of Array or an Array of Hash (similar to the structure returned by #hashes).

Calling this method is particularly useful in Then steps that take a Table argument, if you want to compare that table to some actual values.

Raises:



318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
 
# File 'lib/cucumber/ast/table.rb', line 318

def diff!(other_table, options={})
  options = {:missing_row => true, :surplus_row => true, :missing_col => true, :surplus_col => false}.merge(options)

  other_table = ensure_table(other_table)
  other_table.convert_columns!
  ensure_green!

  original_width = cell_matrix[0].length
  other_table_cell_matrix = pad!(other_table.cell_matrix)
  padded_width = cell_matrix[0].length

  missing_col = cell_matrix[0].detect{|cell| cell.status == :undefined}
  surplus_col = padded_width > original_width

  require_diff_lcs
  cell_matrix.extend(Diff::LCS)
  convert_columns!
  changes = cell_matrix.diff(other_table_cell_matrix).flatten

  inserted = 0
  missing  = 0

  row_indices = Array.new(other_table_cell_matrix.length) {|n| n}

  last_change = nil
  missing_row_pos = nil
  insert_row_pos  = nil
  
  changes.each do |change|
    if(change.action == '-')
      missing_row_pos = change.position + inserted
      cell_matrix[missing_row_pos].each{|cell| cell.status = :undefined}
      row_indices.insert(missing_row_pos, nil)
      missing += 1
    else # '+'
      insert_row_pos = change.position + missing
      inserted_row = change.element
      inserted_row.each{|cell| cell.status = :comment}
      cell_matrix.insert(insert_row_pos, inserted_row)
      row_indices[insert_row_pos] = nil
      inspect_rows(cell_matrix[missing_row_pos], inserted_row) if last_change && last_change.action == '-'
      inserted += 1
    end
    last_change = change
  end

  other_table_cell_matrix.each_with_index do |other_row, i|
    row_index = row_indices.index(i)
    row = cell_matrix[row_index] if row_index
    if row
      (original_width..padded_width).each do |col_index|
        surplus_cell = other_row[col_index]
        row[col_index].value = surplus_cell.value if row[col_index]
      end
    end
  end
  
  clear_cache!
  should_raise = 
    missing_row_pos && options[:missing_row] ||
    insert_row_pos  && options[:surplus_row] ||
    missing_col     && options[:missing_col] ||
    surplus_col     && options[:surplus_col]
  raise Different.new(self) if should_raise
end

#dup ⇒ Object

Creates a copy of this table, inheriting any column mappings. registered with #map_headers!



92
93
94
 
# File 'lib/cucumber/ast/table.rb', line 92

def dup
  self.class.new(raw.dup, @conversion_procs.dup)
end

#each_cells_row(&proc) ⇒ Object

:nodoc:



176
177
178
 
# File 'lib/cucumber/ast/table.rb', line 176

def each_cells_row(&proc) #:nodoc:
  cells_rows.each(&proc)
end

#has_text?(text) ⇒ Boolean

:nodoc:

Returns:

  • (Boolean) 


422
423
424
 
# File 'lib/cucumber/ast/table.rb', line 422

def has_text?(text) #:nodoc:
  raw.flatten.compact.detect{|cell_value| cell_value.index(text)}
end

#hashes ⇒ Object

Converts this table into an Array of Hash where the keys of each Hash are the headers in the table. For example, a Table built from the following plain text:

| a | b | sum |
| 2 | 3 | 5   |
| 7 | 9 | 16  |

Gets converted into the following:

[{'a' => '2', 'b' => '3', 'sum' => '5'}, {'a' => '7', 'b' => '9', 'sum' => '16'}]

Use #map_column! to specify how values in a column are converted.



125
126
127
128
129
 
# File 'lib/cucumber/ast/table.rb', line 125

def hashes
  @hashes ||= cells_rows[1..-1].map do |row|
    row.to_hash
  end
end

#header_cell(col) ⇒ Object

:nodoc:



436
437
438
 
# File 'lib/cucumber/ast/table.rb', line 436

def header_cell(col) #:nodoc:
  cells_rows[0][col]
end

#headers ⇒ Object

:nodoc:



432
433
434
 
# File 'lib/cucumber/ast/table.rb', line 432

def headers #:nodoc:
  raw.first
end

#index(cells) ⇒ Object

:nodoc:



395
396
397
 
# File 'lib/cucumber/ast/table.rb', line 395

def index(cells) #:nodoc:
  cells_rows.index(cells)
end

#map_column!(column_name, strict = true, &conversion_proc) ⇒ Object

Change how #hashes converts column values. The column_name argument identifies the column and conversion_proc performs the conversion for each cell in that column. If strict is true, an error will be raised if the column named column_name is not found. If strict is false, no error will be raised. Example:

Given /^an expense report for (.*) with the following posts:$/ do |table|
  posts_table.map_column!('amount') { |a| a.to_i }
  posts_table.hashes.each do |post|
    # post['amount'] is a Fixnum, rather than a String
  end
end



278
279
280
281
282
 
# File 'lib/cucumber/ast/table.rb', line 278

def map_column!(column_name, strict=true, &conversion_proc)
  verify_column(column_name.to_s) if strict
  @conversion_procs[column_name.to_s] = conversion_proc
  self
end

#map_headers(mappings = {}) ⇒ Object

Returns a new Table where the headers are redefined. See #map_headers!



260
261
262
263
264
 
# File 'lib/cucumber/ast/table.rb', line 260

def map_headers(mappings={})
  table = self.dup
  table.map_headers!(mappings)
  table
end

#map_headers!(mappings = {}, &block) ⇒ Object

Redefines the table headers. This makes it possible to use prettier and more flexible header names in the features. The keys of mappings are Strings or regular expressions (anything that responds to #=== will work) that may match column headings in the table. The values of mappings are desired names for the columns.

Example:

| Phone Number | Address |
| 123456       | xyz     |
| 345678       | abc     |

A StepDefinition receiving this table can then map the columns with both Regexp and String:

table.map_headers!(/phone( number)?/i => :phone, 'Address' => :address)
table.hashes
# => [{:phone => '123456', :address => 'xyz'}, {:phone => '345678', :address => 'abc'}]

You may also pass in a block if you wish to convert all of the headers:

table.map_headers! { |header| header.downcase }
table.hashes.keys
# => ['phone number', 'address']

When a block is passed in along with a hash then the mappings in the hash take precendence:

table.map_headers!('Address' => 'ADDRESS') { |header| header.downcase }
table.hashes.keys
# => ['phone number', 'ADDRESS']



240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
 
# File 'lib/cucumber/ast/table.rb', line 240

def map_headers!(mappings={}, &block)
  header_cells = cell_matrix[0]

  if block_given?
    header_values = header_cells.map { |cell| cell.value } - mappings.keys
    mappings = mappings.merge(Hash[*header_values.zip(header_values.map(&block)).flatten])
  end

  mappings.each_pair do |pre, post|
    mapped_cells = header_cells.select{|cell| pre === cell.value}
    raise "No headers matched #{pre.inspect}" if mapped_cells.empty?
    raise "#{mapped_cells.length} headers matched #{pre.inspect}: #{mapped_cells.map{|c| c.value}.inspect}" if mapped_cells.length > 1
    mapped_cells[0].value = post
    if @conversion_procs.has_key?(pre)
      @conversion_procs[post] = @conversion_procs.delete(pre)
    end
  end
end

#match(pattern) ⇒ Object

Matches pattern against the header row of the table. This is used especially for argument transforms.

Example:

| column_1_name | column_2_name |
| x             | y             |

table.match(/table:column_1_name,column_2_name/) #=> non-nil

Note: must use ‘table:’ prefix on match



198
199
200
201
 
# File 'lib/cucumber/ast/table.rb', line 198

def match(pattern)
  header_to_match = "table:#{headers.join(',')}"
  pattern.match(header_to_match)
end

#raw ⇒ Object

Gets the raw data of this table. For example, a Table built from the following plain text:

| a | b |
| c | d |

gets converted into the following:

[['a', 'b'], ['c', 'd']]



159
160
161
162
163
164
165
 
# File 'lib/cucumber/ast/table.rb', line 159

def raw
  cell_matrix.map do |row|
    row.map do |cell|
      cell.value
    end
  end
end

#rows ⇒ Object

Same as #raw, but skips the first (header) row



172
173
174
 
# File 'lib/cucumber/ast/table.rb', line 172

def rows
  raw[1..-1]
end

#rows_hash ⇒ Object

Converts this table into a Hash where the first column is used as keys and the second column is used as values

| a | 2 |
| b | 3 |

Gets converted into the following:

{'a' => '2', 'b' => '3'}

The table must be exactly two columns wide



143
144
145
146
147
 
# File 'lib/cucumber/ast/table.rb', line 143

def rows_hash
  return @rows_hash if @rows_hash
  verify_table_width(2)
  @rows_hash = self.transpose.hashes[0]
end

#to_hash(cells) ⇒ Object

:nodoc:



384
385
386
387
388
389
390
391
392
393
 
# File 'lib/cucumber/ast/table.rb', line 384

def to_hash(cells) #:nodoc:
  hash = Hash.new do |hash, key|
    hash[key.to_s] if key.is_a?(Symbol)
  end
  column_names.each_with_index do |column_name, column_index|
    value = @conversion_procs[column_name].call(cells.value(column_index))
    hash[column_name] = value
  end
  hash
end

#to_s(options = {}) ⇒ Object

:nodoc:



448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
 
# File 'lib/cucumber/ast/table.rb', line 448

def to_s(options = {}) #:nodoc:
  require 'cucumber/formatter/pretty'
  options = {:color => true, :indent => 2, :prefixes => TO_S_PREFIXES}.merge(options)
  io = StringIO.new

  c = Term::ANSIColor.coloring?
  Term::ANSIColor.coloring = options[:color]
  formatter = Formatter::Pretty.new(nil, io, options)
  formatter.instance_variable_set('@indent', options[:indent])
  TreeWalker.new(nil, [formatter]).visit_multiline_arg(self)
  
  Term::ANSIColor.coloring = c
  io.rewind
  s = "\n" + io.read + (" " * (options[:indent] - 2))
  s
end

#to_sexp ⇒ Object

For testing only



204
205
206
 
# File 'lib/cucumber/ast/table.rb', line 204

def to_sexp #:nodoc:
  [:table, *cells_rows.map{|row| row.to_sexp}]
end

#to_step_definition_arg ⇒ Object 



85
86
87
 
# File 'lib/cucumber/ast/table.rb', line 85

def to_step_definition_arg
  dup
end

#transpose ⇒ Object

Returns a new, transposed table. Example:

| a | 7 | 4 |
| b | 9 | 2 |

Gets converted into the following:

| a | b |
| 7 | 9 |
| 4 | 2 |



107
108
109
 
# File 'lib/cucumber/ast/table.rb', line 107

def transpose
  self.class.new(raw.transpose, @conversion_procs.dup)
end

#verify_column(column_name) ⇒ Object

:nodoc:



399
400
401
 
# File 'lib/cucumber/ast/table.rb', line 399

def verify_column(column_name) #:nodoc:
  raise %{The column named "#{column_name}" does not exist} unless raw[0].include?(column_name)
end

#verify_table_width(width) ⇒ Object

:nodoc:



403
404
405
 
# File 'lib/cucumber/ast/table.rb', line 403

def verify_table_width(width) #:nodoc:
  raise %{The table must have exactly #{width} columns} unless raw[0].size == width
end