@@ -5,7 +5,7 @@ import { diag, SpanContext, SpanKind } from '@opentelemetry/api';
55import { ExportResultCode } from '@opentelemetry/core' ;
66import { Resource } from '@opentelemetry/resources' ;
77import { ProtobufTraceSerializer } from '@opentelemetry/otlp-transformer' ;
8- import { OTLPUdpSpanExporter , UdpExporter } from '../src/otlp- udp-exporter' ;
8+ import { AwsXrayUdpSpanExporter , UdpExporter } from '../src/aws-xray- udp-span -exporter' ;
99import { ReadableSpan } from '@opentelemetry/sdk-trace-base' ;
1010import * as sinon from 'sinon' ;
1111import expect from 'expect' ;
@@ -74,8 +74,8 @@ describe('UdpExporterTest', () => {
7474 } ) ;
7575} ) ;
7676
77- describe ( 'OTLPUdpSpanExporterTest ' , ( ) => {
78- let otlpUdpSpanExporter : OTLPUdpSpanExporter ;
77+ describe ( 'AwsXrayUdpSpanExporterTest ' , ( ) => {
78+ let AwsXrayUdpSpanExporter : AwsXrayUdpSpanExporter ;
7979 let udpExporterMock : { sendData : any ; shutdown : any } ;
8080 let diagErrorSpy : sinon . SinonSpy < [ message : string , ...args : unknown [ ] ] , void > ;
8181 const endpoint = '127.0.0.1:3000' ;
@@ -123,8 +123,8 @@ describe('OTLPUdpSpanExporterTest', () => {
123123 // Stub the diag.error method
124124 diagErrorSpy = sinon . spy ( diag , 'error' ) ;
125125
126- // Create an instance of OTLPUdpSpanExporter
127- otlpUdpSpanExporter = new OTLPUdpSpanExporter ( endpoint , prefix ) ;
126+ // Create an instance of AwsXrayUdpSpanExporter
127+ AwsXrayUdpSpanExporter = new AwsXrayUdpSpanExporter ( endpoint , prefix ) ;
128128 } ) ;
129129
130130 afterEach ( ( ) => {
@@ -137,7 +137,7 @@ describe('OTLPUdpSpanExporterTest', () => {
137137 // Stub ProtobufTraceSerializer.serializeRequest
138138 sinon . stub ( ProtobufTraceSerializer , 'serializeRequest' ) . returns ( serializedData ) ;
139139
140- otlpUdpSpanExporter . export ( spans , callback ) ;
140+ AwsXrayUdpSpanExporter . export ( spans , callback ) ;
141141
142142 expect ( udpExporterMock . sendData . calledOnceWith ( serializedData , 'T1' ) ) . toBe ( true ) ;
143143 expect ( callback . calledOnceWith ( { code : ExportResultCode . SUCCESS } ) ) . toBe ( true ) ;
@@ -149,7 +149,7 @@ describe('OTLPUdpSpanExporterTest', () => {
149149 sinon . stub ( ProtobufTraceSerializer , 'serializeRequest' ) . returns ( undefined ) ;
150150 const callback = sinon . stub ( ) ;
151151
152- otlpUdpSpanExporter . export ( spans , callback ) ;
152+ AwsXrayUdpSpanExporter . export ( spans , callback ) ;
153153
154154 expect ( callback . notCalled ) . toBe ( true ) ;
155155 expect ( udpExporterMock . sendData . notCalled ) . toBe ( true ) ;
@@ -162,26 +162,26 @@ describe('OTLPUdpSpanExporterTest', () => {
162162
163163 const callback = sinon . stub ( ) ;
164164
165- otlpUdpSpanExporter . export ( spans , callback ) ;
165+ AwsXrayUdpSpanExporter . export ( spans , callback ) ;
166166
167167 expect ( diagErrorSpy . calledOnceWith ( 'Error exporting spans: %s' , sinon . match . instanceOf ( Error ) ) ) . toBe ( true ) ;
168168 expect ( callback . calledOnceWith ( { code : ExportResultCode . FAILED } ) ) . toBe ( true ) ;
169169 } ) ;
170170
171171 it ( 'should forceFlush without throwing' , async ( ) => {
172- expect ( otlpUdpSpanExporter . forceFlush ( ) ) . resolves . not . toThrow ( ) ;
172+ expect ( AwsXrayUdpSpanExporter . forceFlush ( ) ) . resolves . not . toThrow ( ) ;
173173 } ) ;
174174
175175 it ( 'should shutdown the UDP exporter successfully' , async ( ) => {
176- await otlpUdpSpanExporter . shutdown ( ) ;
176+ await AwsXrayUdpSpanExporter . shutdown ( ) ;
177177 expect ( udpExporterMock . shutdown . calledOnce ) . toBe ( true ) ;
178178 } ) ;
179179
180180 it ( 'should use expected Environment Variables to configure endpoint' , ( ) => {
181181 process . env . AWS_LAMBDA_FUNCTION_NAME = 'testFunctionName' ;
182182 process . env . AWS_XRAY_DAEMON_ADDRESS = 'someaddress:1234' ;
183183
184- const exporter = new OTLPUdpSpanExporter ( undefined ) ;
184+ const exporter = new AwsXrayUdpSpanExporter ( undefined ) ;
185185 expect ( exporter [ '_endpoint' ] ) . toBe ( 'someaddress:1234' ) ;
186186 expect ( exporter [ '_udpExporter' ] [ '_host' ] ) . toBe ( 'someaddress' ) ;
187187 expect ( exporter [ '_udpExporter' ] [ '_port' ] ) . toBe ( 1234 ) ;
0 commit comments