Index ConstantAndWrite and ConstantOperatorWrite (#1049)

Author: vinistock

lib/ruby_indexer/lib/ruby_indexer/visitor.rb

@@ -35,7 +35,8 @@ def visit(node)
         add_index_entry(node, Index::Entry::Class)
       when YARP::ModuleNode
         add_index_entry(node, Index::Entry::Module)
-      when YARP::ConstantWriteNode, YARP::ConstantOrWriteNode
+      when YARP::ConstantWriteNode, YARP::ConstantOrWriteNode, YARP::ConstantAndWriteNode,
+        YARP::ConstantOperatorWriteNode
         name = fully_qualify_name(node.name.to_s)
         add_constant(node, name)
       when YARP::ConstantPathWriteNode, YARP::ConstantPathOrWriteNode, YARP::ConstantPathOperatorWriteNode,
@@ -90,6 +91,8 @@ def handle_private_constant(node)
         node: T.any(
           YARP::ConstantWriteNode,
           YARP::ConstantOrWriteNode,
+          YARP::ConstantAndWriteNode,
+          YARP::ConstantOperatorWriteNode,
           YARP::ConstantPathWriteNode,
           YARP::ConstantPathOrWriteNode,
           YARP::ConstantPathOperatorWriteNode,

lib/ruby_indexer/test/constant_test.rb

@@ -306,5 +306,23 @@ module A
       constant = @index["A::M"].first
       assert_instance_of(Index::Entry::Constant, constant)
     end
+
+    def test_indexing_or_and_operator_nodes
+      index(<<~RUBY)
+        A ||= 1
+        B &&= 2
+        C &= 3
+        D::E ||= 4
+        F::G &&= 5
+        H::I &= 6
+      RUBY
+
+      assert_entry("A", Index::Entry::Constant, "/fake/path/foo.rb:0-0:0-7")
+      assert_entry("B", Index::Entry::Constant, "/fake/path/foo.rb:1-0:1-7")
+      assert_entry("C", Index::Entry::Constant, "/fake/path/foo.rb:2-0:2-6")
+      assert_entry("D::E", Index::Entry::Constant, "/fake/path/foo.rb:3-0:3-10")
+      assert_entry("F::G", Index::Entry::Constant, "/fake/path/foo.rb:4-0:4-10")
+      assert_entry("H::I", Index::Entry::Constant, "/fake/path/foo.rb:5-0:5-9")
+    end
   end
 end