Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions packages/api/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
"@aws-sdk/client-rds": "^3.1076.0",
"@aws-sdk/client-s3": "^3.1076.0",
"@aws-sdk/client-secrets-manager": "^3.1076.0",
"@aws-sdk/client-sqs": "^3.1090.0",
"dotenv": "^17.4.2",
"hono": "^4.12.27"
},
Expand Down
180 changes: 180 additions & 0 deletions packages/api/src/adapter-aws/AwsQueueAdapter.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,180 @@
import {describe, expect, test} from 'bun:test'
import {AwsQueueAdapter} from './AwsQueueAdapter'

const QUEUE_URL = 'http://localhost:4566/000000000000/orders'

function fakeSqs(overrides: Record<string, (input: unknown) => unknown> = {}) {
return {
async send(command: {constructor: {name: string}; input: unknown}) {
const handler = overrides[command.constructor.name]
if (handler) return handler(command.input)

switch (command.constructor.name) {
case 'ListQueuesCommand':
return {QueueUrls: [QUEUE_URL]}
case 'GetQueueAttributesCommand':
return {
Attributes: {
QueueArn: 'arn:aws:sqs:us-east-1:000000000000:orders',
ApproximateNumberOfMessages: '3',
ApproximateNumberOfMessagesNotVisible: '1',
ApproximateNumberOfMessagesDelayed: '0',
CreatedTimestamp: '1700000000',
FifoQueue: 'false',
},
}
default:
throw new Error(`Unhandled command: ${command.constructor.name}`)
}
},
} as never
}

describe('AwsQueueAdapter', () => {
test('list returns mapped CloudResource array with message counts in metadata', async () => {
const adapter = new AwsQueueAdapter(fakeSqs())
const result = await adapter.list()

expect(result).toHaveLength(1)
expect(result[0].id).toBe(QUEUE_URL)
expect(result[0].name).toBe('orders')
expect(result[0].cloud).toBe('aws')
expect(result[0].service).toBe('queue')
expect(result[0].type).toBe('queue')
expect(result[0].metadata.messagesAvailable).toBe(3)
expect(result[0].metadata.messagesInFlight).toBe(1)
})

test('list filters results by search term', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
ListQueuesCommand: () => ({QueueUrls: [QUEUE_URL, `${QUEUE_URL}-dlq`]}),
}))

const result = await adapter.list({search: 'orders-dlq'})

expect(result).toHaveLength(1)
expect(result[0].name).toBe('orders-dlq')
})

test('create builds a queue and returns the mapped resource', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
CreateQueueCommand: () => ({QueueUrl: QUEUE_URL}),
}))

const resource = await adapter.create({values: {queueName: 'orders'}})

expect(resource.id).toBe(QUEUE_URL)
expect(resource.name).toBe('orders')
})

test('create rejects a missing queueName', async () => {
const adapter = new AwsQueueAdapter(fakeSqs())
await expect(adapter.create({values: {}})).rejects.toThrow('queueName is required')
})

test('sendMessage returns the message id from the SDK response', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
SendMessageCommand: () => ({MessageId: 'msg-1'}),
}))

const message = await adapter.sendMessage(QUEUE_URL, '{"hello":"world"}')

expect(message.id).toBe('msg-1')
expect(message.body).toBe('{"hello":"world"}')
})

test('receiveMessages maps SDK messages into QueueMessage records', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
ReceiveMessageCommand: () => ({
Messages: [
{
MessageId: 'msg-1',
ReceiptHandle: 'receipt-1',
Body: '{"hello":"world"}',
Attributes: {SentTimestamp: '1700000000000', ApproximateReceiveCount: '2'},
MessageAttributes: {source: {StringValue: 'flow-panel', DataType: 'String'}},
},
],
}),
}))

const messages = await adapter.receiveMessages(QUEUE_URL)

expect(messages).toHaveLength(1)
expect(messages[0].id).toBe('msg-1')
expect(messages[0].receiptHandle).toBe('receipt-1')
expect(messages[0].receiveCount).toBe(2)
expect(messages[0].messageAttributes.source).toBe('flow-panel')
})

test('deleteMessage and purgeQueue delegate to the SDK without throwing', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
DeleteMessageCommand: () => ({}),
PurgeQueueCommand: () => ({}),
}))

await expect(adapter.deleteMessage(QUEUE_URL, 'receipt-1')).resolves.toBeUndefined()
await expect(adapter.purgeQueue(QUEUE_URL)).resolves.toBeUndefined()
})

test('list follows NextToken across multiple pages', async () => {
const pageTwoUrl = `${QUEUE_URL}-page2`
let call = 0
const adapter = new AwsQueueAdapter(fakeSqs({
ListQueuesCommand: () => {
call += 1
if (call === 1) return {QueueUrls: [QUEUE_URL], NextToken: 'token-2'}
return {QueueUrls: [pageTwoUrl]}
},
}))

const result = await adapter.list()

expect(call).toBe(2)
expect(result.map((r) => r.id)).toEqual([QUEUE_URL, pageTwoUrl])
})

test('sendMessage rejects a FIFO queue without a messageGroupId', async () => {
const fifoUrl = `${QUEUE_URL}.fifo`
const adapter = new AwsQueueAdapter(fakeSqs())

await expect(adapter.sendMessage(fifoUrl, 'body')).rejects.toThrow(
'messageGroupId is required to send a message to a FIFO queue',
)
})

test('sendMessage passes messageGroupId and messageDeduplicationId for a FIFO queue', async () => {
const fifoUrl = `${QUEUE_URL}.fifo`
let captured: {MessageGroupId?: string; MessageDeduplicationId?: string} | undefined
const adapter = new AwsQueueAdapter(fakeSqs({
SendMessageCommand: (input) => {
captured = input as typeof captured
return {MessageId: 'msg-1'}
},
}))

await adapter.sendMessage(fifoUrl, 'body', {
messageGroupId: 'group-1',
messageDeduplicationId: 'dedup-1',
})

expect(captured?.MessageGroupId).toBe('group-1')
expect(captured?.MessageDeduplicationId).toBe('dedup-1')
})

test('a malformed RedrivePolicy does not break list()', async () => {
const adapter = new AwsQueueAdapter(fakeSqs({
GetQueueAttributesCommand: () => ({
Attributes: {
QueueArn: 'arn:aws:sqs:us-east-1:000000000000:orders',
RedrivePolicy: 'not-valid-json{',
},
}),
}))

const result = await adapter.list()

expect(result).toHaveLength(1)
expect(result[0].metadata.redrivePolicy).toBeNull()
})
})
Loading