Extract a method named getRawClassForType to reuse some code and add a test case for the method.

Comment From: chenqimiao

Duplicate of #24525