LPS-36113 Adding LayoutStagedModelDataHandlerTest
LPS-36113 Fixing bugs found during writing the test