Merge pull request #445 from oliverguenther/fix/dot-visitor-casted-node

Add Arel::Nodes::Casted to dot visitor
rails · Oct 7, 2016 · ea05e8e · ea05e8e
2 parents 029041a + fbbe05e
commit ea05e8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/arel/visitors/dot.rb b/lib/arel/visitors/dot.rb
@@ -156,6 +156,11 @@ def visit_Arel_Table o
         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
@@ -64,6 +64,7 @@ def test_named_function
         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)