posthog/plugin-server/tests/cdp/utils.test.ts

import { DateTime } from 'luxon'

import { HogFunctionInvocationResult } from '../../src/cdp/types'
import { gzipObject, prepareLogEntriesForClickhouse, unGzipObject } from '../../src/cdp/utils'
import { createHogFunction, createInvocation, insertHogFunction as _insertHogFunction } from './fixtures'

describe('Utils', () => {
    describe('gzip compressions', () => {
        it("should compress and decompress a string using gzip's sync functions", async () => {
            const input = { foo: 'bar', foo2: 'bar' }
            const compressed = await gzipObject(input)
            expect(compressed).toHaveLength(52)
            const decompressed = await unGzipObject(compressed)
            expect(decompressed).toEqual(input)
        })
    })

    describe('prepareLogEntriesForClickhouse', () => {
        const startTime = DateTime.fromMillis(1620000000000)
        const example: HogFunctionInvocationResult = {
            invocation: {
                ...createInvocation(createHogFunction({ id: 'hog-1' })),
                id: 'inv-1',
            },
            finished: false,
            logs: [
                {
                    level: 'info',
                    timestamp: startTime.plus(2),
                    message: 'Third log message',
                },
                {
                    level: 'info',
                    timestamp: startTime,
                    message: 'First log message',
                },
                {
                    level: 'info',
                    timestamp: startTime.plus(1),
                    message: 'Second log message',
                },
                {
                    level: 'info',
                    timestamp: startTime.plus(2),
                    message: 'Duplicate log message',
                },
            ],
        }

        it('should add the relevant info to the logs', () => {
            const prepared = prepareLogEntriesForClickhouse(example)

            expect(prepared).toMatchInlineSnapshot(`
                Array [
                  Object {
                    "instance_id": "inv-1",
                    "level": "info",
                    "log_source": "hog_function",
                    "log_source_id": "hog-1",
                    "message": "First log message",
                    "team_id": 1,
                    "timestamp": "2021-05-03 00:00:00.000",
                  },
                  Object {
                    "instance_id": "inv-1",
                    "level": "info",
                    "log_source": "hog_function",
                    "log_source_id": "hog-1",
                    "message": "Second log message",
                    "team_id": 1,
                    "timestamp": "2021-05-03 00:00:00.001",
                  },
                  Object {
                    "instance_id": "inv-1",
                    "level": "info",
                    "log_source": "hog_function",
                    "log_source_id": "hog-1",
                    "message": "Third log message",
                    "team_id": 1,
                    "timestamp": "2021-05-03 00:00:00.002",
                  },
                  Object {
                    "instance_id": "inv-1",
                    "level": "info",
                    "log_source": "hog_function",
                    "log_source_id": "hog-1",
                    "message": "Duplicate log message",
                    "team_id": 1,
                    "timestamp": "2021-05-03 00:00:00.003",
                  },
                ]
            `)
        })
    })
})