haste-server/test/document-stores/redis.test.ts

import RedisDocumentStore from 'src/lib/document-stores/redis'
import { StoreNames } from 'src/types/store-names'

describe('Redis document store', () => {
  let store: RedisDocumentStore
  /* reconnect to redis on each test */
  afterEach(() => {
    if (store) {
      store.client?.quit()
    }
  })

  describe('set', () => {
    it('should be able to set a key and have an expiration set', async () => {
      store = new RedisDocumentStore({
        expire: 10,
        type: StoreNames.Redis
      })
      return store.set('hello1', 'world', async () => {
        const res = await store.client?.ttl('hello1')
        expect(res).toBeGreaterThan(1)
      })
    })

    it('should not set an expiration when told not to', async () => {
      store = new RedisDocumentStore({
        expire: 10,
        type: StoreNames.Redis
      })

      store.set(
        'hello2',
        'world',
        async () => {
          const res = await store.client?.ttl('hello2')
          expect(res).toEqual(-1)
        },
        true
      )
    })

    it('should not set an expiration when expiration is off', async () => {
      store = new RedisDocumentStore({
        type: StoreNames.Redis
      })

      store.set('hello3', 'world', async () => {
        const res = await store.client?.ttl('hello3')
        expect(res).toEqual(-1)
      })
    })
  })
})