add testcase

Author: zhiyuanliang-ms

test/requestTracing.test.ts

@@ -5,7 +5,8 @@ import * as chai from "chai";
 import * as chaiAsPromised from "chai-as-promised";
 chai.use(chaiAsPromised);
 const expect = chai.expect;
-import { MAX_TIME_OUT, createMockedConnectionString, createMockedKeyValue, createMockedTokenCredential, mockAppConfigurationClientListConfigurationSettings, restoreMocks, sleepInMs } from "./utils/testHelper.js";
+import { MAX_TIME_OUT, createMockedConnectionString, createMockedKeyValue, createMockedTokenCredential, mockAppConfigurationClientListConfigurationSettings, restoreMocks, sinon, sleepInMs } from "./utils/testHelper.js";
+import { ConfigurationClientManager } from "../src/ConfigurationClientManager.js";
 import { load } from "./exportedApi.js";
 
 class HttpRequestHeadersPolicy {
@@ -75,6 +76,19 @@ describe("request tracing", function () {
         expect(correlationContext.includes("UsesKeyVault")).eq(true);
     });
 
+    it("should have replica count in correlation-context header", async () => {
+        const replicaCount = 2;
+        sinon.stub(ConfigurationClientManager.prototype, "getReplicaCount").returns(replicaCount);
+        try {
+            await load(createMockedConnectionString(fakeEndpoint), { clientOptions });
+        } catch (e) { /* empty */ }
+        expect(headerPolicy.headers).not.undefined;
+        const correlationContext = headerPolicy.headers.get("Correlation-Context");
+        expect(correlationContext).not.undefined;
+        expect(correlationContext.includes(`ReplicaCount=${replicaCount}`)).eq(true);
+        sinon.restore();
+    });
+
     it("should detect env in correlation-context header", async () => {
         process.env.NODE_ENV = "development";
         try {

test/utils/testHelper.ts

@@ -10,8 +10,8 @@ import { RestError } from "@azure/core-rest-pipeline";
 import { promisify } from "util";
 const sleepInMs = promisify(setTimeout);
 import * as crypto from "crypto";
-import { ConfigurationClientManager } from "../../src/ConfigurationClientManager";
-import { ConfigurationClientWrapper } from "../../src/ConfigurationClientWrapper";
+import { ConfigurationClientManager } from "../../src/ConfigurationClientManager.js";
+import { ConfigurationClientWrapper } from "../../src/ConfigurationClientWrapper.js";
 
 const MAX_TIME_OUT = 20000;