[SPARK-3170][CORE]: Bug Fix in Storage UI

current compeleted stage only need to remove its own partitions that are no longer cached. Currently, "Storage" in Spark UI may lost some rdds which are cached actually.
Could you describe in more detail the sequence that can cause this issue? Better yet, maybe write a unit test?