Fix #140 adding test for snapshot ready

deepstreamIO · Jul 27, 2016 · 59843ac · 59843ac
1 parent d8c03fc
commit 59843ac
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/test-e2e/specs/record-snapshotSpec.js b/test-e2e/specs/record-snapshotSpec.js
@@ -20,7 +20,7 @@ describe( 'record snapshot', function() {
 
 	it( 'creates clientA', function( done ) {
 		clientA = deepstreamClient( 'localhost:6021' );
-		clientA.login( null, function(){ done(); });
+		clientA.login( null, done );
 	});
 
 	 /**************** TEST ****************/
@@ -59,12 +59,23 @@ describe( 'record snapshot', function() {
 				clientA.record.snapshot( 'localRecord', function( error, data ) {
 					expect( error ).toBeNull();
 					expect( data ).toEqual( { key: 'value' } );
-					done();
+					record.on( 'discard', done );
+					record.discard();
 				} );
 			 }, 10 );
 		} );
 	});
 
+	it( 'retrieves remote snapshot if client record is not ready', function( done ){
+		var record = clientA.record.getRecord( 'localRecord' );
+
+		clientA.record.snapshot( 'localRecord', function( error, data ) {
+			expect( error ).toBeNull();
+			expect( data ).toEqual( { key: 'value' } );
+			done();
+		} );
+	});
+
 	/**************** TEAR DOWN ****************/
 	it( 'closes the clients', function() {
 		clientA.close();