@@ -19,16 +19,14 @@ let sequelize_version = require('sequelize').version;
1919if ( ! sequelize_version )
2020 sequelize_version = require ( 'sequelize/package' ) . version ;
2121
22- const { wrapSequelize} = require ( '../index' ) ;
23- const { fields} = require ( '../util' ) ;
22+ const { wrapSequelize } = require ( '../index' ) ;
2423const chai = require ( "chai" ) ;
2524const expect = chai . expect ;
2625const seq_version = require ( 'sequelize' ) . version ;
27- const opencensus_tracing = require ( '@opencensus/nodejs' ) ;
28- const { context, trace} = require ( '@opentelemetry/api' ) ;
29- const { NodeTracerProvider} = require ( '@opentelemetry/node' ) ;
30- const { AsyncHooksContextManager} = require ( '@opentelemetry/context-async-hooks' ) ;
31- const { InMemorySpanExporter, SimpleSpanProcessor} = require ( '@opentelemetry/tracing' ) ;
26+ const { context, trace } = require ( '@opentelemetry/api' ) ;
27+ const { NodeTracerProvider } = require ( '@opentelemetry/node' ) ;
28+ const { AsyncHooksContextManager } = require ( '@opentelemetry/context-async-hooks' ) ;
29+ const { InMemorySpanExporter, SimpleSpanProcessor } = require ( '@opentelemetry/tracing' ) ;
3230
3331const createFakeSequelize = ( ) => {
3432 return {
@@ -42,11 +40,11 @@ const createFakeSequelize = () => {
4240 config : {
4341 database : 'fake' , client : 'fakesql' ,
4442 } ,
45- options : {
46- databaseVersion : 'fakesql-server:0.0.X' ,
47- dialect : 'fakesql' ,
48- timezone : '+00:00' ,
49- } ,
43+ options : {
44+ databaseVersion : 'fakesql-server:0.0.X' ,
45+ dialect : 'fakesql' ,
46+ timezone : '+00:00' ,
47+ } ,
5048 } ,
5149 } ,
5250 } ,
@@ -59,17 +57,16 @@ describe("Comments for Sequelize", () => {
5957 const fakeSequelize = createFakeSequelize ( ) ;
6058
6159 before ( ( ) => {
62- wrapSequelize ( fakeSequelize , { client_timezone :true , db_driver :true } ) ;
60+ wrapSequelize ( fakeSequelize , { client_timezone : true , db_driver : true } ) ;
6361 } ) ;
6462
6563 after ( ( ) => {
66- opencensus_tracing . stop ( ) ;
6764 } ) ;
6865
6966 describe ( "Cases" , ( ) => {
7067
7168 it ( "should add comment to generated sql" , ( done ) => {
72-
69+
7370 const want = `SELECT * FROM foo /*client_timezone='%2B00%3A00',db_driver='sequelize%3A${ seq_version } '*/` ;
7471 const query = 'SELECT * FROM foo' ;
7572
@@ -81,7 +78,7 @@ describe("Comments for Sequelize", () => {
8178 } ) ;
8279
8380 it ( "should NOT affix comments to statements with existing comments" , ( done ) => {
84-
81+
8582 const q = [
8683 'SELECT * FROM people /* existing */' ,
8784 'SELECT * FROM people -- existing'
@@ -120,9 +117,9 @@ describe("Comments for Sequelize", () => {
120117 } ) ;
121118
122119 it ( "chaining and repeated calls should NOT indefinitely chain SQL" , ( done ) => {
123-
120+
124121 const want = `SELECT * FROM foo /*client_timezone='%2B00%3A00',db_driver='sequelize%3A${ seq_version } '*/` ;
125-
122+
126123 const sql = 'SELECT * FROM foo' ;
127124
128125 fakeSequelize . dialect . Query . prototype . run ( sql )
@@ -142,20 +139,19 @@ describe("Excluding all variables", () => {
142139 const fakeSequelize = createFakeSequelize ( ) ;
143140
144141 before ( ( ) => {
145- wrapSequelize ( fakeSequelize , { non_existent : true } ) ;
142+ wrapSequelize ( fakeSequelize , { non_existent : true } ) ;
146143 } ) ;
147144
148145 after ( ( ) => {
149- opencensus_tracing . stop ( ) ;
150146 } ) ;
151147
152148 it ( "when all variables are excluded, no comment should be generated" , ( done ) => {
153149 // Allow a re-wrap.
154150 fakeSequelize . ___alreadySQLCommenterWrapped___ = false ;
155- wrapSequelize ( fakeSequelize , { foo :true } ) ;
151+ wrapSequelize ( fakeSequelize , { foo : true } ) ;
156152
157153 const want = `SELECT * FROM foo` ;
158- const sql = `SELECT * FROM foo` ;
154+ const sql = `SELECT * FROM foo` ;
159155
160156 fakeSequelize . dialect . Query . prototype . run ( sql ) . then ( ( got ) => {
161157 expect ( got ) . equals ( want ) ;
@@ -164,41 +160,6 @@ describe("Excluding all variables", () => {
164160 } ) ;
165161} ) ;
166162
167- describe ( "With OpenCensus tracing" , ( ) => {
168-
169- const fakeSequelize = createFakeSequelize ( ) ;
170-
171- before ( ( ) => {
172- wrapSequelize ( fakeSequelize , { traceparent : true , tracestate : true } , { TraceProvider : "OpenCensus" } ) ;
173- } ) ;
174-
175- after ( ( ) => {
176- opencensus_tracing . stop ( ) ;
177- } ) ;
178-
179- it ( 'Starting an OpenCensus trace should produce `traceparent`' , ( done ) => {
180- // TODO: Follow-up with https://github.com/census-instrumentation/opencensus-node/issues/580
181- // and get a proper guide or file bugs against the project to get the proper
182- // way to retrieve spans. For now let's skip this test.
183- // Remember: https://github.com/census-instrumentation/opencensus-node/issues/580
184-
185- const traceOptions = {
186- samplingRate : 1 , // Always sample
187- } ;
188- const tracer = opencensus_tracing . start ( traceOptions ) . tracer ;
189-
190- tracer . startRootSpan ( { name : 'with-tracing' } , rootSpan => {
191- const sql = 'SELECT * FROM foo' ;
192- fakeSequelize . dialect . Query . prototype . run ( sql ) . then ( ( augmentedSQL ) => {
193- const wantSQL = `SELECT * FROM foo /*traceparent='00-${ rootSpan . traceId } -${ rootSpan . id } -01'*/` ;
194- expect ( augmentedSQL ) . equals ( wantSQL ) ;
195- opencensus_tracing . tracer . stop ( ) ;
196- done ( ) ;
197- } ) ;
198- } ) ;
199- } ) ;
200- } ) ;
201-
202163describe ( "With OpenTelemetry tracing" , ( ) => {
203164
204165 const fakeSequelize = createFakeSequelize ( ) ;
@@ -215,7 +176,7 @@ describe("With OpenTelemetry tracing", () => {
215176 before ( ( ) => {
216177 contextManager = new AsyncHooksContextManager ( ) ;
217178 context . setGlobalContextManager ( contextManager . enable ( ) ) ;
218- wrapSequelize ( fakeSequelize , { traceparent : true , tracestate : true } , { TraceProvider : "OpenTelemetry" } ) ;
179+ wrapSequelize ( fakeSequelize , { traceparent : true , tracestate : true } , { TraceProvider : "OpenTelemetry" } ) ;
219180 } ) ;
220181
221182 after ( ( ) => {
@@ -226,7 +187,7 @@ describe("With OpenTelemetry tracing", () => {
226187 it ( 'Starting an OpenTelemetry trace should produce `traceparent`' , ( done ) => {
227188 const rootSpan = tracer . startSpan ( 'rootSpan' ) ;
228189
229- context . with ( trace . setSpan ( context . active ( ) , rootSpan ) , async ( ) => {
190+ context . with ( trace . setSpan ( context . active ( ) , rootSpan ) , async ( ) => {
230191 const sql = 'SELECT * FROM foo' ;
231192 let augmentedSQL = await fakeSequelize . dialect . Query . prototype . run ( sql ) ;
232193 const wantSQL = `SELECT * FROM foo /*traceparent='00-${ rootSpan . spanContext ( ) . traceId } -${ rootSpan . spanContext ( ) . spanId } -01'*/` ;
0 commit comments