From 4d995cf4db9256a323e5ad90147b38e2b84471e9 Mon Sep 17 00:00:00 2001
From: Aaron Abbott <aaronabbott@google.com>
Date: Sat, 27 Feb 2021 00:06:58 +0000
Subject: [PATCH] chore: add unit test for CollectorExporterNodeBase

---
 .../test/CollectorExporterNodeBase.test.ts    | 154 ++++++++++++++++++
 1 file changed, 154 insertions(+)
 create mode 100644 packages/opentelemetry-exporter-collector-grpc/test/CollectorExporterNodeBase.test.ts

diff --git a/packages/opentelemetry-exporter-collector-grpc/test/CollectorExporterNodeBase.test.ts b/packages/opentelemetry-exporter-collector-grpc/test/CollectorExporterNodeBase.test.ts
new file mode 100644
index 00000000000..a0a3312b3d6
--- /dev/null
+++ b/packages/opentelemetry-exporter-collector-grpc/test/CollectorExporterNodeBase.test.ts
@@ -0,0 +1,154 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// import * as protoLoader from "@grpc/proto-loader";
+import { collectorTypes } from '@opentelemetry/exporter-collector';
+import { ReadableSpan } from '@opentelemetry/tracing';
+
+import * as assert from 'assert';
+import { CollectorExporterNodeBase } from '../src/CollectorExporterNodeBase';
+import { CollectorExporterConfigNode, ServiceClientType } from '../src/types';
+import { mockedReadableSpan } from './helper';
+
+class MockCollectorExporter extends CollectorExporterNodeBase<
+  ReadableSpan,
+  ReadableSpan[]
+> {
+  /**
+   * Callbacks passed to _send()
+   */
+  sendCallbacks: {
+    onSuccess: () => void;
+    onError: (error: collectorTypes.CollectorExporterError) => void;
+  }[] = [];
+
+  getDefaultUrl(config: CollectorExporterConfigNode): string {
+    return '';
+  }
+
+  getDefaultServiceName(config: CollectorExporterConfigNode): string {
+    return '';
+  }
+
+  convert(spans: ReadableSpan[]): ReadableSpan[] {
+    return spans;
+  }
+
+  getServiceClientType() {
+    return ServiceClientType.SPANS;
+  }
+
+  getServiceProtoPath(): string {
+    return 'opentelemetry/proto/collector/trace/v1/trace_service.proto';
+  }
+}
+
+// Mocked _send which just saves the callbacks for later
+MockCollectorExporter.prototype['_send'] = function _sendMock(
+  self: MockCollectorExporter,
+  objects: ReadableSpan[],
+  onSuccess: () => void,
+  onError: (error: collectorTypes.CollectorExporterError) => void
+): void {
+  self.sendCallbacks.push({ onSuccess, onError });
+};
+
+describe('CollectorExporterNodeBase', () => {
+  let exporter: MockCollectorExporter;
+  const concurrencyLimit = 5;
+
+  beforeEach(done => {
+    exporter = new MockCollectorExporter({ concurrencyLimit });
+    done();
+  });
+
+  describe('export', () => {
+    it('should export requests concurrently', async () => {
+      const spans = [Object.assign({}, mockedReadableSpan)];
+      const numToExport = concurrencyLimit;
+
+      for (let i = 0; i < numToExport; ++i) {
+        exporter.export(spans, () => {});
+      }
+
+      assert.strictEqual(exporter['_sendingPromises'].length, numToExport);
+      const promisesAllDone = Promise.all(exporter['_sendingPromises']);
+      // Mock that all requests finish sending
+      exporter.sendCallbacks.forEach(({ onSuccess }) => onSuccess());
+
+      // All finished promises should be dropped off
+      await promisesAllDone;
+      assert.strictEqual(exporter['_sendingPromises'].length, 0);
+    });
+
+    it('should drop export requests when already sending concurrencyLimit', async () => {
+      const spans = [Object.assign({}, mockedReadableSpan)];
+      const numToExport = concurrencyLimit + 5;
+
+      for (let i = 0; i < numToExport; ++i) {
+        exporter.export(spans, () => {});
+      }
+
+      assert.strictEqual(exporter['_sendingPromises'].length, concurrencyLimit);
+      const promisesAllDone = Promise.all(exporter['_sendingPromises']);
+      // Mock that all requests finish sending
+      exporter.sendCallbacks.forEach(({ onSuccess }) => onSuccess());
+
+      // All finished promises should be dropped off
+      await promisesAllDone;
+      assert.strictEqual(exporter['_sendingPromises'].length, 0);
+    });
+
+    it('should drop export requests even if they failed', async () => {
+      const spans = [Object.assign({}, mockedReadableSpan)];
+
+      exporter.export(spans, () => {});
+      assert.strictEqual(exporter['_sendingPromises'].length, 1);
+      const promisesAllDone = Promise.all(exporter['_sendingPromises']);
+      // Mock that all requests fail sending
+      exporter.sendCallbacks.forEach(({ onError }) =>
+        onError(new Error('Failed to send!!'))
+      );
+
+      // All finished promises should be dropped off
+      await promisesAllDone;
+      assert.strictEqual(exporter['_sendingPromises'].length, 0);
+    });
+
+    it('should drop export requests even if success callback throws error', async () => {
+      const spans = [Object.assign({}, mockedReadableSpan)];
+
+      exporter['_sendPromise'](
+        spans,
+        () => {
+          throw new Error('Oops');
+        },
+        () => {}
+      );
+
+      assert.strictEqual(exporter['_sendingPromises'].length, 1);
+      const promisesAllDone = Promise.all(exporter['_sendingPromises']);
+      // Mock that the request finishes sending
+      exporter.sendCallbacks.forEach(({ onSuccess }) => {
+        onSuccess();
+      });
+
+      // All finished promises should be dropped off
+      await promisesAllDone;
+      assert.strictEqual(exporter['_sendingPromises'].length, 0);
+    });
+  });
+});