Add snapshot store tests