import { startPluginsServer } from '../../src/main/pluginsServer'
import { LogLevel } from '../../src/types'
import { delay } from '../../src/utils/utils'
import { makePiscina } from '../../src/worker/piscina'
import { pluginConfig39 } from '../helpers/plugins'
import { getErrorForPluginConfig, resetTestDatabase } from '../helpers/sql'

jest.mock('../../src/utils/status')
jest.setTimeout(60000) // 60 sec timeout

const defaultEvent = {
    distinct_id: 'my_id',
    ip: '127.0.0.1',
    site_url: 'http://localhost',
    team_id: 2,
    now: new Date().toISOString(),
    event: 'default event',
    properties: { key: 'value' },
}

describe('teardown', () => {
    test('teardown code runs when stopping', async () => {
        await resetTestDatabase(`
            async function processEvent (event) {
                event.properties.processed = 'hell yes'
                event.properties.upperUuid = event.properties.uuid?.toUpperCase()
                return event
            }
            async function teardownPlugin() {
                throw new Error('This Happened In The Teardown Palace')
            }
        `)
        const { stop } = await startPluginsServer(
            {
                WORKER_CONCURRENCY: 2,
                LOG_LEVEL: LogLevel.Log,
                KAFKA_ENABLED: false,
            },
            makePiscina
        )

        const error1 = await getErrorForPluginConfig(pluginConfig39.id)
        expect(error1).toBe(null)

        await stop()

        // verify the teardownPlugin code runs
        const error2 = await getErrorForPluginConfig(pluginConfig39.id)
        expect(error2.message).toBe('This Happened In The Teardown Palace')
    })

    test('teardown code runs when reloading', async () => {
        await resetTestDatabase(`
            async function processEvent (event, meta) {
                event.properties.storage = await meta.storage.get('storage', 'nope')
                return event
            }
            async function teardownPlugin(meta) {
                await meta.storage.set('storage', 'tore down')
            }
        `)
        const { piscina, stop, hub } = await startPluginsServer(
            {
                WORKER_CONCURRENCY: 2,
                LOG_LEVEL: LogLevel.Log,
                KAFKA_ENABLED: false,
            },
            makePiscina
        )

        const error1 = await getErrorForPluginConfig(pluginConfig39.id)
        expect(error1).toBe(null)

        await delay(100)

        await hub.db.postgresQuery(
            'update posthog_pluginconfig set updated_at = now() where id = $1',
            [pluginConfig39.id],
            'testTag'
        )
        const event1 = await piscina!.runTask({ task: 'processEvent', args: { event: { ...defaultEvent } } })
        expect(event1.properties.storage).toBe('nope')

        await piscina!.broadcastTask({ task: 'reloadPlugins' })
        await delay(2000)

        const event2 = await piscina!.runTask({ task: 'processEvent', args: { event: { ...defaultEvent } } })
        expect(event2.properties.storage).toBe('tore down')

        await stop()
    })
})