Cannot get the format of the entity

[haven2world]

Describe the bug
When we support formatting api on entities by setting the formattableRoots in the beforeFormat event, we cannot get the updated format when we call getFormatState

Expected behavior
The format state should reflect the format correctly if we allow to change it.

Additional context
A quick fix I found is when we call expandEntitySelections in formatSegmentWithContentModel we also push the entity's format into formatsAndSegments

[haven2world]

created a PR for the change #3196