add test

snowflakedb · Aug 26, 2024 · 4a6184f · 4a6184f
1 parent 1f04897
commit 4a6184f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/test/scala/com/snowflake/snowpark_test/OpenTelemetrySuite.scala b/src/test/scala/com/snowflake/snowpark_test/OpenTelemetrySuite.scala
@@ -447,6 +447,16 @@ class OpenTelemetrySuite extends OpenTelemetryEnabled {
     assert(l.size() == 1)
   }
 
+  test("actions should be processed in the span time period") {
+    val result = session.sql("select current_timestamp()").collect().head.getTimestamp(0)
+    val l = testSpanExporter.getFinishedSpanItems
+    val spanStart = l.get(0).getStartEpochNanos / 1000000
+    val time = result.getTime
+    val spanEnd = l.get(0).getEndEpochNanos / 1000000
+    assert(spanStart < time)
+    assert(time < spanEnd)
+  }
+
   override def beforeAll: Unit = {
     super.beforeAll
     createStage(stageName1)