Fixed #3438: missing handling of restriction of virtual metaclass typ…

…e against metaclass
crystal-lang · Oct 19, 2016 · 203da3a · 203da3a
1 parent 5c3ba1f
commit 203da3a
Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/spec/compiler/semantic/restrictions_spec.cr b/spec/compiler/semantic/restrictions_spec.cr
@@ -381,4 +381,20 @@ describe "Restrictions" do
       Foo.new(1)
       )) { generic_class "Foo", int32 }
   end
+
+  it "restricts virtual metaclass type against metaclass (#3438)" do
+    assert_type(%(
+      class Parent
+      end
+
+      class Child < Parent
+      end
+
+      def foo(x : Parent.class)
+        x
+      end
+
+      foo(Parent || Child)
+      )) { types["Parent"].metaclass.virtual_type! }
+  end
 end
diff --git a/src/compiler/crystal/semantic/restrictions.cr b/src/compiler/crystal/semantic/restrictions.cr
@@ -815,6 +815,12 @@ module Crystal
     end
   end
 
+  class VirtualMetaclassType
+    def restrict(other : Metaclass, context)
+      instance_type.restrict(other.name, context).try &.metaclass
+    end
+  end
+
   class NonGenericModuleType
     def restrict(other, context)
       super || including_types.try(&.restrict(other, context))