fix coverage of UDP Exporter

Author: jj22ee

aws-distro-opentelemetry-node-autoinstrumentation/test/otlp-udp-exporter.test.ts

@@ -69,6 +69,16 @@ describe('UdpExporterTest', () => {
     expect(diagErrorSpy.notCalled).toBe(true);
   });
 
+  it('should handle synchronous errors when sending UDP data', async () => {
+    const errorMessage = 'Synchronous UDP send error';
+    socketSend.throws(new Error(errorMessage)); // Simulate synchronous error
+
+    const data = new Uint8Array([1, 2, 3]);
+    await expect(udpExporter.sendData(data, 'T1')).rejects.toThrow(errorMessage);
+    expect(diagErrorSpy.calledOnce).toBe(true);
+    expect(diagErrorSpy.calledWith('Error sending UDP data: %s', sinon.match.instanceOf(Error))).toBe(true);
+  });
+
   it('should close the socket on shutdown', () => {
     udpExporter.shutdown();
     expect(socketClose.calledOnce).toBe(true);
@@ -183,4 +193,12 @@ describe('OTLPUdpSpanExporterTest', () => {
     await otlpUdpSpanExporter.shutdown();
     expect(udpExporterMock.shutdown.calledOnce).toBe(true);
   });
+
+  it('should handle shutdown errors', async () => {
+    const shutdownError = new Error('Shutdown error');
+    udpExporterMock.shutdown.reset();
+    udpExporterMock.shutdown.throws(shutdownError);
+
+    await expect(otlpUdpSpanExporter.shutdown()).rejects.toThrow('Shutdown error');
+  });
 });

exporters/aws-distro-opentelemetry-exporter-xray-udp/test/aws-xray-udp-span-exporter.test.ts

@@ -70,6 +70,16 @@ describe('UdpExporterTest', () => {
     expect(diagErrorSpy.notCalled).toBe(true);
   });
 
+  it('should handle synchronous errors when sending UDP data', async () => {
+    const errorMessage = 'Synchronous UDP send error';
+    socketSend.throws(new Error(errorMessage)); // Simulate synchronous error
+
+    const data = new Uint8Array([1, 2, 3]);
+    await expect(udpExporter.sendData(data, 'T1')).rejects.toThrow(errorMessage);
+    expect(diagErrorSpy.calledOnce).toBe(true);
+    expect(diagErrorSpy.calledWith('Error sending UDP data: %s', sinon.match.instanceOf(Error))).toBe(true);
+  });
+
   it('should close the socket on shutdown', () => {
     udpExporter.shutdown();
     expect(socketClose.calledOnce).toBe(true);
@@ -195,6 +205,14 @@ describe('AwsXrayUdpSpanExporterTest', () => {
     expect(udpExporterMock.shutdown.calledOnce).toBe(true);
   });
 
+  it('should handle shutdown errors', async () => {
+    const shutdownError = new Error('Shutdown error');
+    udpExporterMock.shutdown.reset();
+    udpExporterMock.shutdown.throws(shutdownError);
+
+    await expect(awsXrayUdpSpanExporter.shutdown()).rejects.toThrow('Shutdown error');
+  });
+
   it('should use expected Environment Variables to configure endpoint', () => {
     process.env.AWS_LAMBDA_FUNCTION_NAME = 'testFunctionName';
     process.env.AWS_XRAY_DAEMON_ADDRESS = 'someaddress:1234';