implement core logic with basic tests

Author: Hweinstock

src/ec2/explorer/ec2InstanceNode.ts

@@ -23,7 +23,7 @@ export class Ec2InstanceNode extends AWSTreeNodeBase implements AWSResourceNode
         public readonly client: Ec2Client,
         public override readonly regionCode: string,
         private readonly partitionId: string,
-        protected instance: Ec2Instance
+        public instance: Ec2Instance
     ) {
         super('')
         this.updateInstance(instance)
@@ -65,6 +65,10 @@ export class Ec2InstanceNode extends AWSTreeNodeBase implements AWSResourceNode
         }
     }
 
+    public get status(): string {
+        return this.instance.status!
+    }
+
     public get name(): string {
         return getNameOfInstance(this.instance) ?? `(no name)`
     }

src/ec2/explorer/ec2ParentNode.ts

@@ -10,14 +10,20 @@ import { Ec2InstanceNode } from './ec2InstanceNode'
 import { Ec2Client } from '../../shared/clients/ec2Client'
 import { updateInPlace } from '../../shared/utilities/collectionUtils'
 import { Commands } from '../../shared/vscode/commands'
+import globals from '../../shared/extensionGlobals'
+import { ToolkitError } from '../../shared/errors'
 
 export const parentContextValue = 'awsEc2ParentNode'
 export type Ec2Node = Ec2InstanceNode | Ec2ParentNode
 
+const pollingInterval = 3000
+
 export class Ec2ParentNode extends AWSTreeNodeBase {
     protected readonly placeHolderMessage = '[No EC2 Instances Found]'
     protected ec2InstanceNodes: Map<string, Ec2InstanceNode>
     public override readonly contextValue: string = parentContextValue
+    protected pollingNodes: Set<string> = new Set<string>()
+    private pollTimer?: NodeJS.Timeout
 
     public constructor(
         public override readonly regionCode: string,
@@ -50,6 +56,38 @@ export class Ec2ParentNode extends AWSTreeNodeBase {
         )
     }
 
+    public isPolling(): boolean {
+        return this.pollingNodes.size !== 0
+    }
+
+    public startPolling(instanceId: string) {
+        this.pollingNodes.add(instanceId)
+        this.pollTimer =
+            this.pollTimer ?? globals.clock.setInterval(this.updatePollingNodes.bind(this), pollingInterval)
+    }
+
+    public updatePollingNodes() {
+        console.log('here')
+        this.pollingNodes.forEach(async value => {
+            const instanceNode = this.getChild(value)
+            await instanceNode.updateStatus()
+
+            if (instanceNode.status != 'pending') {
+                this.pollingNodes.delete(value)
+            }
+        })
+        this.refreshNode()
+    }
+
+    public getChild(instanceId: string): Ec2InstanceNode {
+        if (!this.ec2InstanceNodes.has(instanceId)) {
+            throw new ToolkitError(`Unable to retrieve child instance requested with id: ${instanceId}`, {
+                code: 'MissingChild',
+            })
+        }
+        return this.ec2InstanceNodes.get(instanceId)!
+    }
+
     public async clearChildren() {
         this.ec2InstanceNodes = new Map<string, Ec2InstanceNode>()
     }

src/test/ec2/explorer/ec2ParentNode.test.ts

@@ -4,6 +4,9 @@
  */
 
 import * as assert from 'assert'
+import * as FakeTimers from '@sinonjs/fake-timers'
+import * as sinon from 'sinon'
+import { verify, anything, instance, mock, when } from 'ts-mockito'
 import { Ec2ParentNode } from '../../../ec2/explorer/ec2ParentNode'
 import { stub } from '../../utilities/stubber'
 import { Ec2Client, Ec2Instance } from '../../../shared/clients/ec2Client'
@@ -13,10 +16,15 @@ import {
     assertNodeListOnlyHasPlaceholderNode,
 } from '../../utilities/explorerNodeAssertions'
 import { Ec2InstanceNode } from '../../../ec2/explorer/ec2InstanceNode'
+import { installFakeClock } from '../../testUtil'
+import { AWSTreeNodeBase } from '../../../shared/treeview/nodes/awsTreeNodeBase'
 
 describe('ec2ParentNode', function () {
     let testNode: Ec2ParentNode
     let instances: Ec2Instance[]
+    let clock: FakeTimers.InstalledClock
+    let refreshStub: sinon.SinonStub<[], Promise<void>>
+
     const testRegion = 'testRegion'
     const testPartition = 'testPartition'
 
@@ -34,9 +42,14 @@ describe('ec2ParentNode', function () {
         return client
     }
 
+    before(function () {
+        clock = installFakeClock()
+        refreshStub = sinon.stub(Ec2ParentNode.prototype, 'refreshNode')
+    })
+
     beforeEach(function () {
         instances = [
-            { name: 'firstOne', InstanceId: '0', status: 'running' },
+            { name: 'firstOne', InstanceId: '0', status: 'pending' },
             { name: 'secondOne', InstanceId: '1', status: 'stopped' },
         ]
 
@@ -96,4 +109,8 @@ describe('ec2ParentNode', function () {
         const childNodes = await testNode.getChildren()
         assert.strictEqual(childNodes.length, instances.length, 'Unexpected child count')
     })
+
+    it('is not polling on initialization', async function () {
+        assert.strictEqual(testNode.isPolling(), false)
+    })
 })