Add image generation for UI tests

Change-Id: I615416c8fc65b8755c0a3e757fd5dc53d9b6e15f
Merge "Add image generation for UI tests"