telemetry(ec2): remote VSCode connection #3723

Author: Hweinstock

packages/core/src/awsService/ec2/activation.ts

@@ -44,7 +44,10 @@ export async function activate(ctx: ExtContext): Promise<void> {
         }),
 
         Commands.register('aws.ec2.openRemoteConnection', async (node?: Ec2Node) => {
-            await openRemoteConnection(connectionManagers, node)
+            await telemetry.ec2_connectToInstance.run(async (span) => {
+                span.record({ ec2ConnectionType: 'remoteWorkspace' })
+                await openRemoteConnection(connectionManagers, node)
+            })
         }),
 
         Commands.register('aws.ec2.startInstance', async (node?: Ec2Node) => {

packages/core/src/test/awsService/ec2/activation.test.ts

@@ -0,0 +1,48 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as vscode from 'vscode'
+import * as sinon from 'sinon'
+import { assertTelemetry } from '../../testUtil'
+import { Ec2InstanceNode } from '../../../awsService/ec2/explorer/ec2InstanceNode'
+import { Ec2ParentNode } from '../../../awsService/ec2/explorer/ec2ParentNode'
+import { Ec2Client } from '../../../shared/clients/ec2Client'
+import { Ec2Connecter } from '../../../awsService/ec2/model'
+import { PollingSet } from '../../../shared/utilities/pollingSet'
+
+describe('ec2 activation', function () {
+    let testNode: Ec2InstanceNode
+
+    before(function () {
+        const testRegion = 'test-region'
+        const testPartition = 'test-partition'
+        // Don't want to be polling here, that is tested in ../ec2ParentNode.test.ts
+        // disabled here for convenience (avoiding race conditions with timeout)
+        sinon.stub(PollingSet.prototype, 'start')
+        const testClient = new Ec2Client(testRegion)
+        const parentNode = new Ec2ParentNode(testRegion, testPartition, new Ec2Client(testRegion))
+        testNode = new Ec2InstanceNode(parentNode, testClient, testRegion, testPartition, {
+            InstanceId: 'testId',
+            LastSeenStatus: 'status',
+        })
+    })
+
+    after(function () {
+        sinon.restore()
+    })
+
+    it('telemetry', async function () {
+        const terminalStub = sinon.stub(Ec2Connecter.prototype, 'attemptToOpenEc2Terminal')
+        await vscode.commands.executeCommand('aws.ec2.openTerminal', testNode)
+
+        assertTelemetry('ec2_connectToInstance', { ec2ConnectionType: 'ssm' })
+        terminalStub.restore()
+
+        const stopInstanceStub = sinon.stub(Ec2Client.prototype, 'stopInstanceWithCancel')
+        await vscode.commands.executeCommand('aws.ec2.stopInstance', testNode)
+
+        assertTelemetry('ec2_changeState', { ec2InstanceState: 'stop' })
+        stopInstanceStub.restore()
+    })
+})