Update TestMessageVisitor for IDiagnosticMessage