@@ -7,28 +7,22 @@ const suite = new helper.Suite()
77
88suite . test ( 'pipelining property can be set and retrieved' , ( cb ) => {
99 const client = new Client ( helper . config )
10-
1110 // Initially false
1211 helper . assert . equal ( client . pipelining , false )
13-
1412 client . connect ( ( err ) => {
1513 if ( err ) return cb ( err )
16-
1714 // Can be set to true after connection
1815 client . pipelining = true
1916 helper . assert . equal ( client . pipelining , true )
20-
2117 // Can be set back to false
2218 client . pipelining = false
2319 helper . assert . equal ( client . pipelining , false )
24-
2520 client . end ( cb )
2621 } )
2722} )
2823
2924suite . test ( 'cannot enable pipelining before connection' , ( cb ) => {
3025 const client = new Client ( helper . config )
31-
3226 try {
3327 client . pipelining = true
3428 helper . assert . fail ( 'Should have thrown error' )
@@ -40,55 +34,44 @@ suite.test('cannot enable pipelining before connection', (cb) => {
4034
4135suite . test ( 'pipelining mode allows multiple parameterized queries' , ( cb ) => {
4236 const client = new Client ( helper . config )
43-
4437 client . connect ( ( err ) => {
4538 if ( err ) return cb ( err )
46-
4739 client . pipelining = true
48-
4940 let completed = 0
5041 const results = [ ]
51-
5242 // Send multiple queries in pipeline mode
5343 client . query ( 'SELECT $1::text as message' , [ 'Hello' ] , ( err , result ) => {
5444 if ( err ) return cb ( err )
5545 results [ 0 ] = result
5646 completed ++
5747 if ( completed === 3 ) checkResults ( )
5848 } )
59-
6049 client . query ( 'SELECT $1::int as number' , [ 42 ] , ( err , result ) => {
6150 if ( err ) return cb ( err )
6251 results [ 1 ] = result
6352 completed ++
6453 if ( completed === 3 ) checkResults ( )
6554 } )
66-
6755 client . query ( 'SELECT $1::text as greeting' , [ 'World' ] , ( err , result ) => {
6856 if ( err ) return cb ( err )
6957 results [ 2 ] = result
7058 completed ++
7159 if ( completed === 3 ) checkResults ( )
7260 } )
73-
7461 function checkResults ( ) {
7562 helper . assert . equal ( results [ 0 ] . rows [ 0 ] . message , 'Hello' )
7663 helper . assert . equal ( results [ 1 ] . rows [ 0 ] . number , 42 )
7764 helper . assert . equal ( results [ 2 ] . rows [ 0 ] . greeting , 'World' )
78-
7965 client . end ( cb )
8066 }
8167 } )
8268} )
8369
8470suite . test ( 'pipelining mode rejects simple queries' , ( cb ) => {
8571 const client = new Client ( helper . config )
86-
8772 client . connect ( ( err ) => {
8873 if ( err ) return cb ( err )
89-
9074 client . pipelining = true
91-
9275 client . query ( 'SELECT 1' , ( err , result ) => {
9376 helper . assert ( err )
9477 helper . assert . equal ( err . message , 'Simple query protocol is not allowed in pipeline mode. Use parameterized queries.' )
@@ -99,18 +82,14 @@ suite.test('pipelining mode rejects simple queries', (cb) => {
9982
10083suite . test ( 'pipelining mode rejects multi-statement queries' , ( cb ) => {
10184 const client = new Client ( helper . config )
102-
10385 client . connect ( ( err ) => {
10486 if ( err ) return cb ( err )
105-
10687 client . pipelining = true
107-
10888 client . query ( 'SELECT $1; SELECT $2' , [ 1 , 2 ] , ( err , result ) => {
10989 helper . assert ( err )
11090 helper . assert . equal ( err . message , 'Multiple SQL commands in a single query are not allowed in pipeline mode' )
11191 client . end ( cb )
11292 } )
11393 } )
11494} )
115-
116- module . exports = suite
95+ module . exports = suite
0 commit comments