BatchProcessor modified to use force_flush instead of sleep time

Author: mohammadVatandoost

opentelemetry-sdk/src/logs/batch_log_processor.rs

@@ -913,8 +913,7 @@ mod tests {
             )
             .build();
 
-        // wait for the batch processor to process the resource.
-        tokio::time::sleep(Duration::from_millis(100)).await;
+        provider.force_flush().unwrap();
 
         assert_eq!(exporter.get_resource().unwrap().into_iter().count(), 5);
         let _ = provider.shutdown();

opentelemetry-sdk/src/logs/log_processor_with_async_runtime.rs

@@ -541,8 +541,7 @@ mod tests {
             )
             .build();
 
-        // wait for the batch processor to process the resource.
-        tokio::time::sleep(Duration::from_millis(100)).await;
+        provider.force_flush().unwrap();
 
         assert_eq!(exporter.get_resource().unwrap().into_iter().count(), 5);
         let _ = provider.shutdown();
@@ -804,7 +803,7 @@ mod tests {
                 KeyValue::new("k5", "v5"),
             ]))
             .build();
-        tokio::time::sleep(Duration::from_millis(500)).await; // set resource in batch log processor is not blocking. Should we make it blocking?
+        provider.force_flush().unwrap();
         assert_eq!(exporter.get_resource().unwrap().into_iter().count(), 5);
         let _ = provider.shutdown();
     }

opentelemetry-sdk/src/trace/span_processor.rs

@@ -1268,7 +1268,7 @@ mod tests {
             processor.on_end(span);
         }
 
-        tokio::time::sleep(Duration::from_millis(1000)).await;
+        processor.force_flush().unwrap();
 
         let exported_spans = exporter_shared.lock().unwrap();
         assert_eq!(exported_spans.len(), 4);
@@ -1292,7 +1292,7 @@ mod tests {
             processor.on_end(span);
         }
 
-        tokio::time::sleep(Duration::from_millis(1000)).await;
+        processor.force_flush().unwrap();
 
         let exported_spans = exporter_shared.lock().unwrap();
         assert_eq!(exported_spans.len(), 4);
@@ -1326,8 +1326,7 @@ mod tests {
             handle.await.unwrap();
         }
 
-        // Allow time for batching and export
-        tokio::time::sleep(Duration::from_millis(1000)).await;
+        processor.force_flush().unwrap();
 
         // Verify exported spans
         let exported_spans = exporter_shared.lock().unwrap();