Query span tests

Azure · Oct 24, 2024 · 0b34e33 · 0b34e33
1 parent 70ab041
commit 0b34e33
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/sdk/data/azcosmos/emulator_cosmos_container_test.go b/sdk/data/azcosmos/emulator_cosmos_container_test.go
@@ -11,7 +11,7 @@ import (
 func TestContainerCRUD(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{"create_container aContainer", "read_container aContainer", "replace_container aContainer", "read_container_throughput aContainer", "replace_container_throughput aContainer", "delete_container aContainer"},
+		ExpectedSpans: []string{"create_container aContainer", "read_container aContainer", "query_containers containerCRUD", "replace_container aContainer", "read_container_throughput aContainer", "replace_container_throughput aContainer", "delete_container aContainer"},
 	}))
 
 	database := emulatorTests.createDatabase(t, context.TODO(), client, "containerCRUD")

diff --git a/sdk/data/azcosmos/emulator_cosmos_database_test.go b/sdk/data/azcosmos/emulator_cosmos_database_test.go
@@ -11,7 +11,7 @@ import (
 func TestDatabaseCRUD(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{"create_database baseDbTest", "read_database baseDbTest", "delete_database baseDbTest", "read_database_throughput baseDbTest"},
+		ExpectedSpans: []string{"create_database baseDbTest", "read_database baseDbTest", "query_databases localhost", "delete_database baseDbTest", "read_database_throughput baseDbTest"},
 	}))
 
 	database := DatabaseProperties{ID: "baseDbTest"}

diff --git a/sdk/data/azcosmos/emulator_cosmos_query_test.go b/sdk/data/azcosmos/emulator_cosmos_query_test.go
@@ -13,7 +13,7 @@ import (
 func TestSinglePartitionQueryWithIndexMetrics(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{},
+		ExpectedSpans: []string{"query_items aContainer"},
 	}))
 
 	database := emulatorTests.createDatabase(t, context.TODO(), client, "queryTests")
@@ -88,7 +88,7 @@ func TestSinglePartitionQueryWithIndexMetrics(t *testing.T) {
 func TestSinglePartitionQuery(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{},
+		ExpectedSpans: []string{"query_items aContainer"},
 	}))
 
 	database := emulatorTests.createDatabase(t, context.TODO(), client, "queryTests")
@@ -176,7 +176,7 @@ func TestSinglePartitionQuery(t *testing.T) {
 func TestSinglePartitionQueryWithParameters(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{},
+		ExpectedSpans: []string{"query_items aContainer"},
 	}))
 
 	database := emulatorTests.createDatabase(t, context.TODO(), client, "queryTests")
@@ -228,7 +228,7 @@ func TestSinglePartitionQueryWithParameters(t *testing.T) {
 func TestSinglePartitionQueryWithProjection(t *testing.T) {
 	emulatorTests := newEmulatorTests(t)
 	client := emulatorTests.getClient(t, newSpanValidator(t, &spanMatcher{
-		ExpectedSpans: []string{},
+		ExpectedSpans: []string{"query_items aContainer"},
 	}))
 
 	database := emulatorTests.createDatabase(t, context.TODO(), client, "queryTests")