diff options
| -rw-r--r-- | lib/arel/visitors/dot.rb | 5 | ||||
| -rw-r--r-- | test/visitors/test_dot.rb | 1 |
2 files changed, 6 insertions, 0 deletions
diff --git a/lib/arel/visitors/dot.rb b/lib/arel/visitors/dot.rb index 85edc50eb7..9aa22d33f6 100644 --- a/lib/arel/visitors/dot.rb +++ b/lib/arel/visitors/dot.rb @@ -156,6 +156,11 @@ module Arel visit_edge o, "name" end + def visit_Arel_Nodes_Casted o + visit_edge o, 'val' + visit_edge o, 'attribute' + end + def visit_Arel_Attribute o visit_edge o, "relation" visit_edge o, "name" diff --git a/test/visitors/test_dot.rb b/test/visitors/test_dot.rb index 6db9b9b6de..179a2c297c 100644 --- a/test/visitors/test_dot.rb +++ b/test/visitors/test_dot.rb @@ -64,6 +64,7 @@ module Arel Arel::Nodes::As, Arel::Nodes::DeleteStatement, Arel::Nodes::JoinSource, + Arel::Nodes::Casted, ].each do |klass| define_method("test_#{klass.name.gsub('::', '_')}") do binary = klass.new(:a, :b) |