first commit

2026-09-22 16:13:21 -04:00
commit 7e49be27f1
297 changed files with 63162 additions and 0 deletions
@@ -0,0 +1,144 @@
+import {describe, expect, test} from 'bun:test'
+import {AwsComputeAdapter} from './AwsComputeAdapter'
+import type {Ec2Instance, RunInstanceInput} from '../services/ec2'
+
+const baseInstance: Ec2Instance = {
+    instanceId: 'i-0abc123def456',
+    name: 'web-server-1',
+    state: 'running',
+    instanceType: 't3.micro',
+    availabilityZone: 'us-east-1a',
+    publicIpAddress: '54.1.2.3',
+    privateIpAddress: '10.0.1.5',
+    vpcId: 'vpc-abc',
+    subnetId: 'subnet-abc',
+    imageId: 'ami-0abcdef',
+    keyName: 'my-key',
+    launchTime: '2025-01-01T00:00:00.000Z',
+    architecture: 'x86_64',
+    platform: undefined,
+    securityGroups: [{id: 'sg-abc', name: 'default'}],
+    tags: [{key: 'Name', value: 'web-server-1'}],
+}
+
+function fakeService(overrides: Partial<{
+    listInstances: () => Promise<Ec2Instance[]>
+    describeInstance: (id: string) => Promise<Ec2Instance>
+    terminateInstance: (id: string) => Promise<void>
+    listAmis: () => Promise<[]>
+    runInstance: (input: RunInstanceInput) => Promise<Ec2Instance>
+}> = {}) {
+    return {
+        listInstances: async () => [baseInstance],
+        describeInstance: async (_id: string) => baseInstance,
+        terminateInstance: async (_id: string) => {},
+        listAmis: async () => [] as [],
+        runInstance: async (_input: RunInstanceInput) => baseInstance,
+        ...overrides,
+    }
+}
+
+describe('AwsComputeAdapter', () => {
+    test('list returns mapped CloudResource array', async () => {
+        const adapter = new AwsComputeAdapter(fakeService())
+        const result = await adapter.list()
+
+        expect(result).toHaveLength(1)
+        expect(result[0].id).toBe('i-0abc123def456')
+        expect(result[0].name).toBe('web-server-1')
+        expect(result[0].cloud).toBe('aws')
+        expect(result[0].service).toBe('compute')
+        expect(result[0].type).toBe('instance')
+        expect(result[0].status).toBe('running')
+        expect(result[0].region).toBe('us-east-1a')
+    })
+
+    test('list filters results by search term', async () => {
+        const adapter = new AwsComputeAdapter(fakeService({
+            listInstances: async () => [
+                baseInstance,
+                {...baseInstance, instanceId: 'i-other', name: 'db-server-1'},
+            ],
+        }))
+
+        const result = await adapter.list({search: 'web'})
+
+        expect(result).toHaveLength(1)
+        expect(result[0].name).toBe('web-server-1')
+    })
+
+    test('list returns all instances when search is empty', async () => {
+        const adapter = new AwsComputeAdapter(fakeService({
+            listInstances: async () => [
+                baseInstance,
+                {...baseInstance, instanceId: 'i-other', name: 'db-server-1'},
+            ],
+        }))
+
+        const result = await adapter.list({})
+
+        expect(result).toHaveLength(2)
+    })
+
+    test('get returns null when instance has 404 http status', async () => {
+        const err = Object.assign(new Error('Not found'), {$metadata: {httpStatusCode: 404}})
+        const adapter = new AwsComputeAdapter(fakeService({
+            describeInstance: async () => { throw err },
+        }))
+
+        const result = await adapter.get('i-missing')
+
+        expect(result).toBeNull()
+    })
+
+    test('get rethrows non-404 errors', async () => {
+        const err = Object.assign(new Error('InternalError'), {$metadata: {httpStatusCode: 500}})
+        const adapter = new AwsComputeAdapter(fakeService({
+            describeInstance: async () => { throw err },
+        }))
+
+        await expect(adapter.get('i-bad')).rejects.toThrow('InternalError')
+    })
+
+    test('create calls runInstance and returns mapped CloudResource', async () => {
+        const launched: RunInstanceInput[] = []
+        const adapter = new AwsComputeAdapter(fakeService({
+            runInstance: async (input) => { launched.push(input); return baseInstance },
+        }))
+        const result = await adapter.create({values: {
+            name: 'test-box',
+            imageId: 'ami-0abcdef123',
+            instanceType: 't3.micro',
+            keyName: 'my-key',
+            subnetId: 'subnet-abc',
+            securityGroupIds: 'sg-111, sg-222',
+        }})
+        expect(launched).toHaveLength(1)
+        expect(launched[0].name).toBe('test-box')
+        expect(launched[0].imageId).toBe('ami-0abcdef123')
+        expect(launched[0].instanceType).toBe('t3.micro')
+        expect(launched[0].securityGroupIds).toEqual(['sg-111', 'sg-222'])
+        expect(result.type).toBe('instance')
+        expect(result.cloud).toBe('aws')
+    })
+
+    test('delete calls terminateInstance with the id', async () => {
+        const terminated: string[] = []
+        const adapter = new AwsComputeAdapter(fakeService({
+            terminateInstance: async (id) => { terminated.push(id) },
+        }))
+        await adapter.delete('i-0abc123def456')
+        expect(terminated).toEqual(['i-0abc123def456'])
+    })
+
+    test('schema returns aws compute schema', () => {
+        const adapter = new AwsComputeAdapter(fakeService())
+        const schema = adapter.schema()
+
+        expect(schema.cloud).toBe('aws')
+        expect(schema.service).toBe('compute')
+        expect(schema.displayName).toBe('Compute')
+        expect(schema.actions).toContain('list')
+        expect(schema.actions).toContain('inspect')
+    })
+})