|
| 1 | +import * as test from 'vitest'; |
| 2 | +import {faker} from '@faker-js/faker'; |
| 3 | +import Keyv from 'keyv'; |
| 4 | +import KeyvSqlite from '../src/index.js'; |
| 5 | + |
| 6 | +test.it('Async Iterator with Keyv and useKeyPrefix true', async t => { |
| 7 | + const store = new KeyvSqlite({uri: 'sqlite://test/testdb2.sqlite', busyTimeout: 3000}); |
| 8 | + const keyv = new Keyv({store, useKeyPrefix: true}); |
| 9 | + await keyv.clear(); |
| 10 | + // Test with Keyv instance |
| 11 | + const keyvData = { |
| 12 | + key: faker.string.alphanumeric(10), |
| 13 | + value: faker.string.alphanumeric(10), |
| 14 | + }; |
| 15 | + await keyv.set(keyvData.key, keyvData.value); |
| 16 | + const keyvResult = await keyv.get(keyvData.key); |
| 17 | + t.expect(keyvResult).toBe(keyvData.value); |
| 18 | + // Ensure the Keyv instance can still use the iterator |
| 19 | + t.expect(keyv.iterator).toBeDefined(); |
| 20 | + if (typeof keyv.iterator === 'function') { |
| 21 | + const keyvIterator = keyv.iterator({}); |
| 22 | + let keyvDataFound = false; |
| 23 | + for await (const [key, raw] of keyvIterator) { |
| 24 | + t.expect(key).toBe(keyvData.key); |
| 25 | + t.expect(raw).toBe(keyvData.value); |
| 26 | + keyvDataFound = true; |
| 27 | + } |
| 28 | + |
| 29 | + if (!keyvDataFound) { |
| 30 | + t.expect.fail('Keyv iterator did not find the expected data'); |
| 31 | + } |
| 32 | + } else { |
| 33 | + t.expect.fail('Keyv iterator is not a function'); |
| 34 | + } |
| 35 | +}); |
| 36 | + |
| 37 | +test.it('Async Iterator with Keyv and useKeyPrefix false', async t => { |
| 38 | + const store = new KeyvSqlite({uri: 'sqlite://test/testdb2.sqlite', busyTimeout: 3000}); |
| 39 | + const keyv = new Keyv({store}); |
| 40 | + keyv.namespace = undefined; |
| 41 | + keyv.useKeyPrefix = false; |
| 42 | + await keyv.clear(); |
| 43 | + // Test with Keyv instance |
| 44 | + const keyvData = { |
| 45 | + key: faker.string.alphanumeric(10), |
| 46 | + value: faker.string.alphanumeric(10), |
| 47 | + }; |
| 48 | + await keyv.set(keyvData.key, keyvData.value); |
| 49 | + const keyvResult = await keyv.get(keyvData.key); |
| 50 | + t.expect(keyvResult).toBe(keyvData.value); |
| 51 | + // Ensure the Keyv instance can still use the iterator |
| 52 | + t.expect(keyv.iterator).toBeDefined(); |
| 53 | + if (typeof keyv.iterator === 'function') { |
| 54 | + const keyvIterator = keyv.iterator({}); |
| 55 | + let keyvDataFound = false; |
| 56 | + for await (const [key, raw] of keyvIterator) { |
| 57 | + t.expect(key).toBe(keyvData.key); |
| 58 | + t.expect(raw).toBe(keyvData.value); |
| 59 | + keyvDataFound = true; |
| 60 | + } |
| 61 | + |
| 62 | + if (!keyvDataFound) { |
| 63 | + t.expect.fail('Keyv iterator did not find the expected data'); |
| 64 | + } |
| 65 | + } else { |
| 66 | + t.expect.fail('Keyv iterator is not a function'); |
| 67 | + } |
| 68 | +}); |
0 commit comments