Adding optional block to run before screenshot

Adding optional block to run viewModel config after view is created but
before screenshot is taken

Bug: 280354248
Change-Id: Ic28a6e0ab6e3de3ce6dd7ae8cea9a16963bac82f
1 file changed