Add ExprAnalysisResultStorage::getExprAnalysisResult()

Author: ondrejmirtes

src/Analyser/Generator/ExprAnalysisResultStorage.php

@@ -3,7 +3,10 @@
 namespace PHPStan\Analyser\Generator;
 
 use PhpParser\Node\Expr;
+use PHPStan\ShouldNotHappenException;
 use SplObjectStorage;
+use function get_class;
+use function sprintf;
 
 final class ExprAnalysisResultStorage
 {
@@ -26,4 +29,13 @@ public function findExprAnalysisResult(Expr $expr): ?ExprAnalysisResult
 		return $this->expressionAnalysisResults[$expr] ?? null;
 	}
 
+	public function getExprAnalysisResult(Expr $expr): ExprAnalysisResult
+	{
+		if (!isset($this->expressionAnalysisResults[$expr])) {
+			throw new ShouldNotHappenException(sprintf('Result for expr %s not found', get_class($expr)));
+		}
+
+		return $this->expressionAnalysisResults[$expr];
+	}
+
 }