Added tests for snapshot