setLayoutStateContextForTesting