fixup! fix tests

Azure · Jun 22, 2021 · 71500cd · 71500cd
1 parent 34ca21c
commit 71500cd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sdk/core/core-rest-pipeline/test/tracingPolicy.spec.ts b/sdk/core/core-rest-pipeline/test/tracingPolicy.spec.ts
@@ -23,15 +23,16 @@ import {
   SpanStatus,
   SpanStatusCode,
   SpanAttributes,
-  Tracer
+  Tracer,
+  SpanAttributeValue
 } from "@azure/core-tracing";
 
 class MockSpan implements Span {
   private _endCalled = false;
   private _status: SpanStatus = {
     code: SpanStatusCode.UNSET
   };
-  private _attributes: { [s: string]: unknown } = {};
+  private _attributes: SpanAttributes = {};
 
   constructor(
     private traceId: string,
@@ -78,7 +79,7 @@ class MockSpan implements Span {
     return this;
   }
 
-  setAttribute(key: string, value: unknown) {
+  setAttribute(key: string, value: SpanAttributeValue) {
     this._attributes[key] = value;
     return this;
   }