Create a reusable component for snapshot tests

[alizedebray]

The code for snapshot tests is repetitive and quite complicated at the moment. Creating a reusable lit component would make the snapshot test easier to write and to maintain.