Refactor ViewScreenshotTestRule slightly to make it a bit extensible

Bug: 271331584
Test: atest AODClockOPRTest
Change-Id: I0a2563c5bb8a5e3bbf191eed18b48de8c2214798
1 file changed