Add snapshot tests

[blakedietz]

Need to add jest and snapshot testing for all of this.