add default switch branches

goatshriek · Jul 26, 2024 · 771c4eb · 771c4eb
1 parent 25867f1
commit 771c4eb
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/wrapture/scope.rb b/lib/wrapture/scope.rb
@@ -255,10 +255,12 @@ def overloads?(parent)
     # Returns the ClassSpec for the given +type+ in the scope, if one exists.
     def type(type)
       name = case type
-             when String
-               type
              when TypeSpec
                type.base
+             when String
+               type
+             else
+               type.to_s
              end
 
       @classes.find { |class_spec| class_spec.name == name }
@@ -267,10 +269,12 @@ def type(type)
     # Returns true if there is a class matching the given +type+ in this scope.
     def type?(type)
       name = case type
-             when String
-               type
              when TypeSpec
                type.base
+             when String
+               type
+             else
+               type.to_s
              end
 
       @classes.any? { |class_spec| class_spec.name == name }