fix existing test cases

elastic · Jan 21, 2025 · c908c66 · c908c66
1 parent 3906d04
commit c908c66
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/internal/inventory/azurefetcher/fetcher_storage_test.go b/internal/inventory/azurefetcher/fetcher_storage_test.go
@@ -51,6 +51,11 @@ func TestStorageFetcher_Fetch(t *testing.T) {
 		Name:        "queue",
 		DisplayName: "queue",
 	}
+	azureTableService := azurelib_inventory.AzureAsset{
+		Id:          "queue",
+		Name:        "queue",
+		DisplayName: "queue",
+	}
 
 	expected := []inventory.AssetEvent{
 		inventory.NewAssetEvent(
@@ -83,6 +88,16 @@ func TestStorageFetcher_Fetch(t *testing.T) {
 				ServiceName: "Azure",
 			}),
 		),
+		inventory.NewAssetEvent(
+			inventory.AssetClassificationAzureStorageTableService,
+			azureQueue.Id,
+			azureQueue.Name,
+			inventory.WithRawAsset(azureTableService),
+			inventory.WithCloud(inventory.Cloud{
+				Provider:    inventory.AzureCloudProvider,
+				ServiceName: "Azure",
+			}),
+		),
 	}
 
 	// setup
@@ -119,6 +134,12 @@ func TestStorageFetcher_Fetch(t *testing.T) {
 		[]azurelib_inventory.AzureAsset{azureQueue}, nil,
 	)
 
+	provider.EXPECT().ListStorageAccountTableServices(
+		mock.Anything, mock.Anything,
+	).Return(
+		[]azurelib_inventory.AzureAsset{azureTableService}, nil,
+	)
+
 	fetcher := newStorageFetcher(logger, provider)
 	// test & compare
 	testutil.CollectResourcesAndMatch(t, fetcher, expected)