[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.
There's actually a `StorageTabSuite` and this should go in there