From 9ca217fd8a64978aa4b2eabb851e944e3f798fa1 Mon Sep 17 00:00:00 2001
From: Richard Marmorstein <richardm@stripe.com>
Date: Wed, 13 Dec 2023 09:59:10 -0800
Subject: [PATCH] More extensive test

---
 test/telemetry.spec.ts | 61 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/test/telemetry.spec.ts b/test/telemetry.spec.ts
index c91ca11a92..f532a34db6 100644
--- a/test/telemetry.spec.ts
+++ b/test/telemetry.spec.ts
@@ -3,6 +3,7 @@
 import http = require('http');
 import {expect} from 'chai';
 import {FAKE_API_KEY} from './testUtils.js';
+import {StripeResource} from '../src/StripeResource.js';
 let testServer = null;
 
 function createTestServer(handlerFunc, cb) {
@@ -124,6 +125,66 @@ describe('Client Telemetry', () => {
       }
     );
   });
+  it('Sends usage telemetry when registered', (done) => {
+    let numRequests = 0;
+
+    createTestServer(
+      (req, res) => {
+        numRequests += 1;
+
+        const telemetry = req.headers['x-stripe-client-telemetry'];
+
+        switch (numRequests) {
+          case 1:
+            expect(telemetry).to.not.exist;
+            break;
+          case 2:
+            expect(telemetry).to.exist;
+            expect(
+              JSON.parse(telemetry).last_request_metrics.request_id
+            ).to.equal('req_1');
+            expect(
+              JSON.parse(telemetry).last_request_metrics.usage
+            ).to.deep.equal(['llama', 'bufo' ]);
+            break;
+          default:
+            expect.fail(`Should not have reached request ${numRequests}`);
+        }
+
+        res.setHeader('Request-Id', `req_${numRequests}`);
+        res.writeHead(200, {'Content-Type': 'application/json'});
+        res.end('{}');
+      },
+      (host, port) => {
+        const stripe = require('../src/stripe.cjs.node.js')(
+          'sk_test_FEiILxKZwnmmocJDUjUNO6pa',
+          {
+            telemetry: true,
+            host,
+            port,
+            protocol: 'http',
+          }
+        );
+
+        const resource = new (StripeResource.extend({
+          boop: StripeResource.method({
+            method: 'POST',
+            fullPath: '/v1/widgets/{widget}/boop',
+            usage: ['llama', 'bufo' ],
+          })}))(stripe)
+
+        return Promise.resolve()
+          .then(() => resource.boop('w_123'))
+          .then(() => resource.boop('w_123'))
+          .then(() => {
+            expect(numRequests).to.equal(2);
+          })
+          .catch(done)
+          .then(() => done())
+
+      }
+    );
+  });
 
   it('Buffers metrics on concurrent requests', (done) => {
     let numRequests = 0;