1 files changed, 150 insertions, 0 deletions
diff --git a/src/airmx.test.ts b/src/airmx.test.ts
new file mode 100644
index 0000000..aee5bbe
--- /dev/null
+++ b/src/airmx.test.ts
@@ -0,0 +1,150 @@
+import { MqttClient } from 'mqtt'
+import { Airmx, Topic } from './airmx.js'
+import { EagleStatus } from './eagle.js'
+
+describe('topic', () => {
+  test('parse parses topic from string', () => {
+    const topic = Topic.parse('airmx/01/0/1/1/1/12345')
+    expect(topic.deviceId).toBe(12345)
+    expect(topic.unknown1).toBe(false)
+    expect(topic.unknown2).toBe(true)
+    expect(topic.unknown3).toBe(true)
+    expect(topic.unknown4).toBe(true)
+  })
+
+  test('the topic format is expected to have 7 parts', () => {
+    expect(() => Topic.parse('foo')).toThrow(
+      'The topic format is expected to be airmx/+/+/+/+/+/+.',
+    )
+  })
+
+  test('the 1st part is expected to be airmx', () => {
+    expect(() => Topic.parse('foo/01/0/1/1/1/12345')).toThrow(
+      'The 1st part of the topic must be "airmx".',
+    )
+  })
+
+  test('the 2nd part is expected to be "01"', () => {
+    expect(() => Topic.parse('airmx/00/0/1/1/1/12345')).toThrow(
+      'The 2nd part of the topic must be "01".',
+    )
+  })
+
+  test('the 3rd part is expected to be either "1" or "0"', () => {
+    expect(Topic.parse('airmx/01/0/1/1/1/12345')).toBeInstanceOf(Topic)
+    expect(Topic.parse('airmx/01/1/1/1/1/12345')).toBeInstanceOf(Topic)
+
+    expect(() => Topic.parse('airmx/01/2/1/1/1/12345')).toThrow(
+      'The 3rd part of the topic must be either "1" or "0".',
+    )
+  })
+
+  test('the 4th part is expected to be either "1" or "0"', () => {
+    expect(Topic.parse('airmx/01/0/0/1/1/12345')).toBeInstanceOf(Topic)
+    expect(Topic.parse('airmx/01/0/1/1/1/12345')).toBeInstanceOf(Topic)
+
+    expect(() => Topic.parse('airmx/01/0/2/1/1/12345')).toThrow(
+      'The 4th part of the topic must be either "1" or "0".',
+    )
+  })
+
+  test('the 5th part is expected to be either "1" or "0"', () => {
+    expect(Topic.parse('airmx/01/0/1/0/1/12345')).toBeInstanceOf(Topic)
+    expect(Topic.parse('airmx/01/0/1/1/1/12345')).toBeInstanceOf(Topic)
+
+    expect(() => Topic.parse('airmx/01/0/1/2/1/12345')).toThrow(
+      'The 5th part of the topic must be either "1" or "0".',
+    )
+  })
+
+  test('the 6th part is expected to be either "1" or "0"', () => {
+    expect(Topic.parse('airmx/01/0/1/1/0/12345')).toBeInstanceOf(Topic)
+    expect(Topic.parse('airmx/01/0/1/1/1/12345')).toBeInstanceOf(Topic)
+
+    expect(() => Topic.parse('airmx/01/0/1/1/2/12345')).toThrow(
+      'The 6th part of the topic must be either "1" or "0".',
+    )
+  })
+
+  test('the 7th part is expected to be the device id', () => {
+    expect(() => Topic.parse('airmx/01/0/1/1/1/')).toThrow(
+      'The 7th part of the topic must be a device ID.',
+    )
+
+    expect(() => Topic.parse('airmx/01/0/1/1/1/foo')).toThrow(
+      'The 7th part of the topic must be a device ID.',
+    )
+  })
+})
+
+describe('airmx', () => {
+  let mockMqttClient: jest.Mocked<MqttClient>
+
+  beforeEach(() => {
+    mockMqttClient = {
+      on: jest.fn(),
+      subscribe: jest.fn(),
+    } as unknown as jest.Mocked<MqttClient>
+  })
+
+  it('should subscribe to the topic when the client connects', () => {
+    new Airmx({ mqtt: mockMqttClient, devices: [] })
+    const connectHandler = mockMqttClient.on.mock.calls.find(
+      ([event]) => event === 'connect',
+    )?.[1] as (() => void) | undefined
+    connectHandler?.()
+    expect(mockMqttClient.subscribe).toHaveBeenCalledWith('airmx/01/+/+/1/1/+')
+  })
+
+  it('should record the latest status for eagles', () => {
+    const testDevice = { id: 1, key: 'f0eb21fe346c88e1d1ac73546022cd5d' }
+    const message =
+      '{"cmdId": 210,"name":"eagleStatus","time":1752675701,"from":2,"data":{"version":"10.00.17","power":1,"heatStatus":0,"mode":0,"cadr":47,"prm":1320,"g4Percent": 100,"hepaPercent":100,"carbonId":"031","g4Id":"041","hepaId":"021","carbonPercent":17,"diffPressure1":99999,"diffPressure2":99999,"t0":35,"status":0,"denoise":1},"sig":"b8796682da77e8c929dddf7e6461afec"}'
+
+    const airmx = new Airmx({ mqtt: mockMqttClient, devices: [testDevice] })
+    const messageHandler = mockMqttClient.on.mock.calls.find(
+      ([event]) => event === 'message',
+    )?.[1] as ((topic: string, message: Buffer) => void) | undefined
+    expect(messageHandler).toBeDefined()
+
+    messageHandler!('airmx/01/0/1/1/1/1', Buffer.from(message))
+    const status = airmx.getEagleStatus(1)
+    expect(status).toBeInstanceOf(EagleStatus)
+  })
+
+  describe('message validation', () => {
+    const testDevice = { id: 1, key: 'f0eb21fe346c88e1d1ac73546022cd5d' }
+    let messageHandler: ((topic: string, message: Buffer) => void) | undefined
+
+    beforeEach(() => {
+      new Airmx({ mqtt: mockMqttClient, devices: [testDevice] })
+      messageHandler = mockMqttClient.on.mock.calls.find(
+        ([event]) => event === 'message',
+      )?.[1] as ((topic: string, message: Buffer) => void) | undefined
+      expect(messageHandler).toBeDefined()
+    })
+
+    const validMessage =
+      '{"cmdId": 210,"name":"eagleStatus","time":1752675701,"from":2,"data":{"version":"10.00.17","power":1,"heatStatus":0,"mode":0,"cadr":47,"prm":1320,"g4Percent": 100,"hepaPercent":100,"carbonId":"031","g4Id":"041","hepaId":"021","carbonPercent":17,"diffPressure1":99999,"diffPressure2":99999,"t0":35,"status":0,"denoise":1},"sig":"b8796682da77e8c929dddf7e6461afec"}'
+    const invalidMessage =
+      '{"cmdId": 210,"name":"eagleStatus","time":1752675701,"from":2,"data":{"version":"10.00.17","power":1,"heatStatus":0,"mode":0,"cadr":47,"prm":1320,"g4Percent": 100,"hepaPercent":100,"carbonId":"031","g4Id":"041","hepaId":"021","carbonPercent":17,"diffPressure1":99999,"diffPressure2":99999,"t0":35,"status":0,"denoise":1},"sig":"invalid"}'
+
+    it('should validate message signatures when receiving messages', () => {
+      expect(() => {
+        messageHandler?.('airmx/01/0/1/1/1/1', Buffer.from(validMessage))
+      }).not.toThrow()
+    })
+
+    it('should throw errors when message signatures are invalid', () => {
+      expect(() => {
+        messageHandler?.('airmx/01/0/1/1/1/1', Buffer.from(invalidMessage))
+      }).toThrow('Failed to validate the message.')
+    })
+
+    it('should throw errors when device does not exist', () => {
+      expect(() => {
+        messageHandler?.('airmx/01/0/1/1/1/99999', Buffer.from(validMessage))
+      }).toThrow('Could not find the device with ID 99999.')
+    })
+  })
+})