Added unit tests for basic snapshot functionality