test: fix parentNode tests

aws · Feb 27, 2025 · 9543f86 · 9543f86
1 parent bda476a
commit 9543f86
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/packages/core/src/test/awsService/ec2/explorer/ec2ParentNode.test.ts b/packages/core/src/test/awsService/ec2/explorer/ec2ParentNode.test.ts
@@ -34,7 +34,7 @@ export const testParentNode = new Ec2ParentNode('fake-region', 'testPartition',
 describe('ec2ParentNode', function () {
     let testNode: Ec2ParentNode
     let client: Ec2Client
-    let getInstancesStub: sinon.SinonStub<[filters?: Filter[] | undefined], AsyncCollection<PatchedEc2Instance>>
+    let getInstancesStub: sinon.SinonStub<[filters?: Filter[] | undefined], AsyncCollection<PatchedEc2Instance[]>>
     let clock: FakeTimers.InstalledClock
     let refreshStub: sinon.SinonStub<[], Promise<void>>
     let statusUpdateStub: sinon.SinonStub<[status: string], Promise<string>>
@@ -78,7 +78,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: '0', LastSeenStatus: 'running' },
             { Name: 'secondOne', InstanceId: '1', LastSeenStatus: 'stopped' },
         ] satisfies PatchedEc2Instance[]
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         const childNodes = await testNode.getChildren()
 
         assert.strictEqual(childNodes.length, instances.length, 'Unexpected child count')
@@ -101,7 +101,7 @@ describe('ec2ParentNode', function () {
             { Name: 'cd', InstanceId: '5', LastSeenStatus: 'running' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
 
         const childNodes = await testNode.getChildren()
 
@@ -124,7 +124,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: '2', LastSeenStatus: 'running' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
 
         const childNodes = await testNode.getChildren()
         assert.strictEqual(childNodes.length, instances.length, 'Unexpected child count')
@@ -138,7 +138,7 @@ describe('ec2ParentNode', function () {
             { Name: 'thirdOne', InstanceId: '2', LastSeenStatus: 'running' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
         assert.strictEqual(testNode.pollingSet.size, 1)
         getInstancesStub.restore()
@@ -152,7 +152,7 @@ describe('ec2ParentNode', function () {
             { Name: 'thirdOne', InstanceId: '2', LastSeenStatus: 'running' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
 
         await testNode.updateChildren()
         await clock.tickAsync(6000)
@@ -166,7 +166,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: '0', LastSeenStatus: 'pending' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
 
         sinon.assert.notCalled(refreshStub)
@@ -179,7 +179,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: 'node1', LastSeenStatus: 'pending' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
         const node = testNode.getInstanceNode('node1')
         assert.strictEqual(node.InstanceId, instances[0].InstanceId)
@@ -191,7 +191,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: 'node1', LastSeenStatus: 'pending' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
         assert.throws(() => testNode.getInstanceNode('node2'))
         getInstancesStub.restore()
@@ -202,7 +202,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: 'node1', LastSeenStatus: 'pending' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
         testNode.trackPendingNode('node1')
         assert.strictEqual(testNode.pollingSet.size, 1)
@@ -214,7 +214,7 @@ describe('ec2ParentNode', function () {
             { Name: 'firstOne', InstanceId: 'node1', LastSeenStatus: 'pending' },
         ] satisfies PatchedEc2Instance[]
 
-        getInstancesStub.returns(intoCollection(instances))
+        getInstancesStub.returns(intoCollection([instances]))
         await testNode.updateChildren()
         assert.throws(() => testNode.trackPendingNode('node2'))
         getInstancesStub.restore()