Fix parsing of type members and templates

Signed-off-by: Alexandre Terrasa <[email protected]>

lib/rbi/parser.rb

@@ -787,10 +787,7 @@ def set_root_tree_loc
 
       sig { params(node: T.nilable(Prism::Node)).returns(T::Boolean) }
       def type_variable_definition?(node)
-        return false unless node.is_a?(Prism::CallNode)
-        return false unless node.block
-
-        node.message == "type_member" || node.message == "type_template"
+        node.is_a?(Prism::CallNode) && (node.message == "type_member" || node.message == "type_template")
       end
     end
 

test/rbi/parser_test.rb

@@ -271,7 +271,7 @@ def test_parse_sorbet_type_members_and_templates
         class Foo
           A = type_member
           B = type_member(:in)
-          C = type_member(:out)
+          C = type_member(:out) {}
           D = type_member(lower: A)
           E = type_member(upper: A)
           F = type_member(:in, fixed: A)
@@ -281,8 +281,14 @@ class Foo
         end
       RBI
 
-      out = Parser.parse_string(rbi)
-      assert_equal(rbi, out.string)
+      tree = Parser.parse_string(rbi)
+
+      # Make sure the type members and templates are not parsed as constants
+      cls = T.must(tree.nodes.grep(Class).first)
+      assert_equal(0, cls.nodes.grep(Const).size)
+      assert_equal(9, cls.nodes.grep(TypeMember).size)
+
+      assert_equal(rbi, tree.string)
     end
 
     def test_parse_root_tree_location