Add screenshot util to unit tests.

Test: atest
Bug: b/277911350
Change-Id: I13add386dc4cd7bd3a21a539d7d034ec14f37be7
31 files changed