@@ -75,6 +75,7 @@ void setup() {
7575
7676	@ AfterEach 
7777	void  tearDown () {
78+ 		
7879		Observation  currentObservation  = observationRegistry .getCurrentObservation ();
7980		if  (currentObservation  != null ) {
8081			currentObservation .stop ();
@@ -85,40 +86,32 @@ void tearDown() {
8586	@ Test 
8687	void  commandStartedShouldNotInstrumentWhenAdminDatabase () {
8788
88- 		// when 
8989		listener .commandStarted (new  CommandStartedEvent (null , 0 , 0 , null , "admin" , "" , null ));
9090
91- 		// then 
9291		assertThat (meterRegistry ).hasNoMetrics ();
9392	}
9493
9594	@ Test 
9695	void  commandStartedShouldNotInstrumentWhenNoRequestContext () {
9796
98- 		// when 
9997		listener .commandStarted (new  CommandStartedEvent (null , 0 , 0 , null , "some name" , "" , null ));
10098
101- 		// then 
10299		assertThat (meterRegistry ).hasNoMetrics ();
103100	}
104101
105102	@ Test 
106103	void  commandStartedShouldNotInstrumentWhenNoParentSampleInRequestContext () {
107104
108- 		// when 
109105		listener .commandStarted (new  CommandStartedEvent (new  MapRequestContext (), 0 , 0 , null , "some name" , "" , null ));
110106
111- 		// then 
112107		assertThat (meterRegistry ).hasMeterWithName ("spring.data.mongodb.command.active" );
113108	}
114109
115110	@ Test  // GH-4994 
116111	void  commandStartedShouldAlwaysIncludeCollection () {
117112
118- 		// when 
119113		listener .commandStarted (new  CommandStartedEvent (new  MapRequestContext (), 0 , 0 , null , "some name" , "hello" , null ));
120114
121- 		// then 
122115		// although command 'hello' is collection-less, metric must have tag "db.mongodb.collection" 
123116		assertThat (meterRegistry ).hasMeterWithNameAndTags (
124117				"spring.data.mongodb.command.active" ,
@@ -140,18 +133,15 @@ void reactiveContextCompletesNormally() {
140133				new  BsonDocument ("collection" , new  BsonString ("user" ))));
141134		listener .commandSucceeded (new  CommandSucceededEvent (context , 0 , 0 , null , "insert" , null , null , 0 ));
142135
143- 		// then 
144136		assertThatTimerRegisteredWithTags ();
145137	}
146138
147139	@ Test 
148140	void  successfullyCompletedCommandShouldCreateTimerWhenParentSampleInRequestContext () {
149141
150- 		// given 
151142		Observation  parent  = Observation .start ("name" , observationRegistry );
152143		RequestContext  traceRequestContext  = getContext ();
153144
154- 		// when 
155145		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , // 
156146				new  ConnectionDescription ( // 
157147						new  ServerId ( // 
@@ -160,18 +150,15 @@ void successfullyCompletedCommandShouldCreateTimerWhenParentSampleInRequestConte
160150				new  BsonDocument ("collection" , new  BsonString ("user" ))));
161151		listener .commandSucceeded (new  CommandSucceededEvent (traceRequestContext , 0 , 0 , null , "insert" , null , null , 0 ));
162152
163- 		// then 
164153		assertThatTimerRegisteredWithTags ();
165154	}
166155
167156	@ Test 
168157	void  successfullyCompletedCommandWithCollectionHavingCommandNameShouldCreateTimerWhenParentSampleInRequestContext () {
169158
170- 		// given 
171159		Observation  parent  = Observation .start ("name" , observationRegistry );
172160		RequestContext  traceRequestContext  = getContext ();
173161
174- 		// when 
175162		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , // 
176163				new  ConnectionDescription ( // 
177164						new  ServerId ( // 
@@ -181,18 +168,15 @@ void successfullyCompletedCommandWithCollectionHavingCommandNameShouldCreateTime
181168				new  BsonDocument ("aggregate" , new  BsonString ("user" ))));
182169		listener .commandSucceeded (new  CommandSucceededEvent (traceRequestContext , 0 , 0 , null , "aggregate" , null , null , 0 ));
183170
184- 		// then 
185171		assertThatTimerRegisteredWithTags ();
186172	}
187173
188174	@ Test 
189175	void  successfullyCompletedCommandWithoutClusterInformationShouldCreateTimerWhenParentSampleInRequestContext () {
190176
191- 		// given 
192177		Observation  parent  = Observation .start ("name" , observationRegistry );
193178		RequestContext  traceRequestContext  = getContext ();
194179
195- 		// when 
196180		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , null , "database" , "insert" ,
197181				new  BsonDocument ("collection" , new  BsonString ("user" ))));
198182		listener .commandSucceeded (new  CommandSucceededEvent (traceRequestContext , 0 , 0 , null , "insert" , null , null , 0 ));
@@ -207,11 +191,9 @@ void successfullyCompletedCommandWithoutClusterInformationShouldCreateTimerWhenP
207191	@ Test 
208192	void  commandWithErrorShouldCreateTimerWhenParentSampleInRequestContext () {
209193
210- 		// given 
211194		Observation  parent  = Observation .start ("name" , observationRegistry );
212195		RequestContext  traceRequestContext  = getContext ();
213196
214- 		// when 
215197		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , // 
216198				new  ConnectionDescription ( // 
217199						new  ServerId ( // 
@@ -222,20 +204,17 @@ void commandWithErrorShouldCreateTimerWhenParentSampleInRequestContext() {
222204		listener .commandFailed ( // 
223205				new  CommandFailedEvent (traceRequestContext , 0 , 0 , null , "db" , "insert" , 0 , new  IllegalAccessException ()));
224206
225- 		// then 
226207		assertThatTimerRegisteredWithTags ();
227208	}
228209
229210	@ Test  // GH-4481 
230211	void  completionShouldIgnoreIncompatibleObservationContext () {
231212
232- 		// given 
233213		RequestContext  traceRequestContext  = getContext ();
234214
235215		Observation  observation  = mock (Observation .class );
236216		traceRequestContext .put (ObservationThreadLocalAccessor .KEY , observation );
237217
238- 		// when 
239218		listener .commandSucceeded (new  CommandSucceededEvent (traceRequestContext , 0 , 0 , null , "insert" , null , null , 0 ));
240219
241220		verify (observation ).getContext ();
@@ -245,13 +224,11 @@ void completionShouldIgnoreIncompatibleObservationContext() {
245224	@ Test  // GH-4481 
246225	void  failureShouldIgnoreIncompatibleObservationContext () {
247226
248- 		// given 
249227		RequestContext  traceRequestContext  = getContext ();
250228
251229		Observation  observation  = mock (Observation .class );
252230		traceRequestContext .put (ObservationThreadLocalAccessor .KEY , observation );
253231
254- 		// when 
255232		listener .commandFailed (new  CommandFailedEvent (traceRequestContext , 0 , 0 , null , "db" , "insert" , 0 , null ));
256233
257234		verify (observation ).getContext ();
@@ -261,7 +238,6 @@ void failureShouldIgnoreIncompatibleObservationContext() {
261238	@ Test  // GH-4321 
262239	void  shouldUseObservationConvention () {
263240
264- 		// given 
265241		MongoHandlerObservationConvention  customObservationConvention  = new  MongoHandlerObservationConvention () {
266242			@ Override 
267243			public  boolean  supportsContext (Observation .Context  context ) {
@@ -276,22 +252,18 @@ public String getName() {
276252		this .listener  = new  MongoObservationCommandListener (observationRegistry , mock (ConnectionString .class ),
277253				customObservationConvention );
278254
279- 		// when 
280255		listener .commandStarted (new  CommandStartedEvent (new  MapRequestContext (), 0 , 0 , null , "some name" , "" , null ));
281256
282- 		// then 
283257		assertThat (meterRegistry ).hasMeterWithName ("custom.name.active" );
284258	}
285259
286260	@ Test  // GH-5064 
287261	void  completionRestoresParentObservation () {
288262
289- 		// given 
290263		Observation  parent  = Observation .start ("name" , observationRegistry );
291264		observationRegistry .setCurrentObservationScope (parent .openScope ());
292265		RequestContext  traceRequestContext  = getContext ();
293266
294- 		// when 
295267		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , null , "database" , "insert" ,
296268				new  BsonDocument ("collection" , new  BsonString ("user" ))));
297269
@@ -306,12 +278,10 @@ void completionRestoresParentObservation() {
306278	@ Test  // GH-5064 
307279	void  failureRestoresParentObservation () {
308280
309- 		// given 
310281		Observation  parent  = Observation .start ("name" , observationRegistry );
311282		observationRegistry .setCurrentObservationScope (parent .openScope ());
312283		RequestContext  traceRequestContext  = getContext ();
313284
314- 		// when 
315285		listener .commandStarted (new  CommandStartedEvent (traceRequestContext , 0 , 0 , null , "database" , "insert" ,
316286				new  BsonDocument ("collection" , new  BsonString ("user" ))));
317287
0 commit comments