Temp add test coverage

Azure · Dec 2, 2024 · 049161c · 049161c
1 parent ed350d8
commit 049161c
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 0 deletions.
diff --git a/tests/BlobTestServerFactory.ts b/tests/BlobTestServerFactory.ts
@@ -2,6 +2,7 @@ import BlobConfiguration from "../src/blob/BlobConfiguration";
 import BlobServer from "../src/blob/BlobServer";
 import SqlBlobConfiguration from "../src/blob/SqlBlobConfiguration";
 import SqlBlobServer from "../src/blob/SqlBlobServer";
+import { AzuriteTelemetryClient } from "../src/common/Telemetry";
 import { StoreDestinationArray } from "../src/common/persistence/IExtentStore";
 import { DEFAULT_SQL_OPTIONS } from "../src/common/utils/constants";
 
@@ -75,6 +76,7 @@ export default class BlobTestServerFactory {
         undefined,
         inMemoryPersistence
       );
+      AzuriteTelemetryClient.init("", true, undefined);
       return new BlobServer(config);
     }
   }

diff --git a/tests/blob/blobCorsRequest.test.ts b/tests/blob/blobCorsRequest.test.ts
@@ -15,6 +15,7 @@ import {
 } from "../testutils";
 import OPTIONSRequestPolicyFactory from "./RequestPolicy/OPTIONSRequestPolicyFactory";
 import OriginPolicyFactory from "./RequestPolicy/OriginPolicyFactory";
+import { AzuriteTelemetryClient } from "../../src/common/Telemetry";
 
 // Set true to enable debug log
 configLogger(false);
@@ -41,11 +42,13 @@ describe("Blob Cors requests test", () => {
 
   before(async () => {
     await server.start();
+    await AzuriteTelemetryClient.TraceStartEvent("Blob Test");
   });
 
   after(async () => {
     await server.close();
     await server.clean();
+    AzuriteTelemetryClient.TraceStopEvent("Blob Test");
   });
 
   it("OPTIONS request without cors rules in server should be fail @loki @sql", async () => {

diff --git a/tests/queue/oauth.test.ts b/tests/queue/oauth.test.ts
@@ -8,6 +8,7 @@ import { StoreDestinationArray } from "../../src/common/persistence/IExtentStore
 import { EMULATOR_ACCOUNT_KEY, generateJWTToken, getUniqueName } from "../testutils";
 import { SimpleTokenCredential } from "../simpleTokenCredential";
 import QueueTestServerFactory from "./utils/QueueTestServerFactory";
+import { AzuriteTelemetryClient } from "../../src/common/Telemetry";
 
 // Set true to enable debug log
 configLogger(false);
@@ -41,11 +42,13 @@ describe("Queue OAuth Basic", () => {
       oauth: "basic"
     })
     await server.start();
+    await AzuriteTelemetryClient.TraceStartEvent("Queue Test");
   });
 
   after(async () => {
     await server.close();
     await server.clean();
+    AzuriteTelemetryClient.TraceStopEvent("Queue Test");
   });
 
   it(`Should work with create container @loki @sql`, async () => {

diff --git a/tests/queue/utils/QueueTestServerFactory.ts b/tests/queue/utils/QueueTestServerFactory.ts
@@ -1,3 +1,4 @@
+import { AzuriteTelemetryClient } from "../../../src/common/Telemetry"
 import { StoreDestinationArray } from "../../../src/common/persistence/IExtentStore"
 import QueueConfiguration from "../../../src/queue/QueueConfiguration"
 import QueueServer from "../../../src/queue/QueueServer"
@@ -43,6 +44,7 @@ export default class QueueTestServerFactory {
       undefined,
       inMemoryPersistence
     );
+    AzuriteTelemetryClient.init("", true, undefined);
     return new QueueServer(config);
   }
 }
diff --git a/tests/table/apis/table.entity.azure.data-tables.test.ts b/tests/table/apis/table.entity.azure.data-tables.test.ts
@@ -20,6 +20,7 @@ import {
   LargeDataTablesTestEntityFactory,
   LargeTableTestEntity
 } from "../models/LargeDataTablesTestEntityFactory";
+import { AzuriteTelemetryClient } from "../../../src/common/Telemetry";
 // Set true to enable debug log
 configLogger(false);
 // For convenience, we have a switch to control the use
@@ -38,10 +39,13 @@ describe("table Entity APIs test - using Azure/data-tables", () => {
   before(async () => {
     server = createTableServerForTestHttps();
     await server.start();
+    AzuriteTelemetryClient.init("", true, undefined);
+    await AzuriteTelemetryClient.TraceStartEvent("Table Test");
   });
 
   after(async () => {
     await server.close();
+    AzuriteTelemetryClient.TraceStopEvent("Table Test");
   });
 
   it("01. Batch API should return row keys in format understood by @azure/data-tables, @loki", async () => {

diff --git a/tests/table/utils/TableTestServerFactory.ts b/tests/table/utils/TableTestServerFactory.ts
@@ -1,3 +1,4 @@
+import { AzuriteTelemetryClient } from "../../../src/common/Telemetry";
 import TableConfiguration from "../../../src/table/TableConfiguration";
 import TableServer from "../../../src/table/TableServer";
 
@@ -42,6 +43,7 @@ export default class TableTestServerFactory {
       undefined,
       inMemoryPersistence
     );
+    AzuriteTelemetryClient.init("", true, undefined);
     return new TableServer(config);
   }
 }