@mtcute/test 0.6.0 → 0.7.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (73) hide show
  1. package/cjs/client.d.ts +3 -4
  2. package/cjs/client.js +5 -8
  3. package/cjs/client.js.map +1 -1
  4. package/cjs/crypto.js +1 -1
  5. package/cjs/crypto.js.map +1 -1
  6. package/cjs/index.d.ts +1 -1
  7. package/cjs/index.js +1 -1
  8. package/cjs/index.js.map +1 -1
  9. package/cjs/storage/auth-keys.d.ts +3 -0
  10. package/cjs/storage/auth-keys.js +95 -0
  11. package/cjs/storage/auth-keys.js.map +1 -0
  12. package/cjs/storage/index.d.ts +4 -0
  13. package/cjs/storage/index.js +21 -0
  14. package/cjs/storage/index.js.map +1 -0
  15. package/cjs/storage/key-value.d.ts +3 -0
  16. package/cjs/storage/key-value.js +41 -0
  17. package/cjs/storage/key-value.js.map +1 -0
  18. package/cjs/storage/peers.d.ts +3 -0
  19. package/cjs/storage/peers.js +75 -0
  20. package/cjs/storage/peers.js.map +1 -0
  21. package/cjs/storage/ref-messages.d.ts +3 -0
  22. package/cjs/storage/ref-messages.js +69 -0
  23. package/cjs/storage/ref-messages.js.map +1 -0
  24. package/cjs/storage.d.ts +2 -16
  25. package/cjs/storage.js +18 -29
  26. package/cjs/storage.js.map +1 -1
  27. package/cjs/transport.test.js +1 -2
  28. package/cjs/transport.test.js.map +1 -1
  29. package/cjs/utils.js +1 -2
  30. package/cjs/utils.js.map +1 -1
  31. package/esm/client.d.ts +3 -4
  32. package/esm/client.js +6 -9
  33. package/esm/client.js.map +1 -1
  34. package/esm/crypto.js +1 -1
  35. package/esm/crypto.js.map +1 -1
  36. package/esm/index.d.ts +1 -1
  37. package/esm/index.js +1 -1
  38. package/esm/index.js.map +1 -1
  39. package/esm/storage/auth-keys.d.ts +3 -0
  40. package/esm/storage/auth-keys.js +90 -0
  41. package/esm/storage/auth-keys.js.map +1 -0
  42. package/esm/storage/index.d.ts +4 -0
  43. package/esm/storage/index.js +5 -0
  44. package/esm/storage/index.js.map +1 -0
  45. package/esm/storage/key-value.d.ts +3 -0
  46. package/esm/storage/key-value.js +36 -0
  47. package/esm/storage/key-value.js.map +1 -0
  48. package/esm/storage/peers.d.ts +3 -0
  49. package/esm/storage/peers.js +70 -0
  50. package/esm/storage/peers.js.map +1 -0
  51. package/esm/storage/ref-messages.d.ts +3 -0
  52. package/esm/storage/ref-messages.js +64 -0
  53. package/esm/storage/ref-messages.js.map +1 -0
  54. package/esm/storage.d.ts +2 -16
  55. package/esm/storage.js +18 -29
  56. package/esm/storage.js.map +1 -1
  57. package/esm/transport.test.js +1 -2
  58. package/esm/transport.test.js.map +1 -1
  59. package/esm/utils.js +2 -3
  60. package/esm/utils.js.map +1 -1
  61. package/package.json +2 -2
  62. package/cjs/storage-test.d.ts +0 -24
  63. package/cjs/storage-test.js +0 -365
  64. package/cjs/storage-test.js.map +0 -1
  65. package/cjs/storage.test.d.ts +0 -1
  66. package/cjs/storage.test.js +0 -43
  67. package/cjs/storage.test.js.map +0 -1
  68. package/esm/storage-test.d.ts +0 -24
  69. package/esm/storage-test.js +0 -357
  70. package/esm/storage-test.js.map +0 -1
  71. package/esm/storage.test.d.ts +0 -1
  72. package/esm/storage.test.js +0 -41
  73. package/esm/storage.test.js.map +0 -1
@@ -1,365 +0,0 @@
1
- "use strict";
2
- var __importDefault = (this && this.__importDefault) || function (mod) {
3
- return (mod && mod.__esModule) ? mod : { "default": mod };
4
- };
5
- Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.testStateStorage = exports.testStorage = exports.stubPeerUser = void 0;
7
- const long_1 = __importDefault(require("long"));
8
- const vitest_1 = require("vitest");
9
- const utils_js_1 = require("@mtcute/core/utils.js");
10
- const reader_js_1 = require("@mtcute/tl/binary/reader.js");
11
- const writer_js_1 = require("@mtcute/tl/binary/writer.js");
12
- const stub_js_1 = require("./stub.js");
13
- exports.stubPeerUser = {
14
- id: 123123,
15
- accessHash: long_1.default.fromBits(123, 456),
16
- type: 'user',
17
- username: 'some_user',
18
- phone: '78005553535',
19
- full: (0, stub_js_1.createStub)('user', { id: 123123 }),
20
- };
21
- const peerUserInput = {
22
- _: 'inputPeerUser',
23
- userId: 123123,
24
- accessHash: long_1.default.fromBits(123, 456),
25
- };
26
- const peerChannel = {
27
- id: -1001183945448,
28
- accessHash: long_1.default.fromBits(666, 555),
29
- type: 'channel',
30
- username: 'some_channel',
31
- full: (0, stub_js_1.createStub)('channel', { id: 123123 }),
32
- };
33
- const peerChannelInput = {
34
- _: 'inputPeerChannel',
35
- channelId: 1183945448,
36
- accessHash: long_1.default.fromBits(666, 555),
37
- };
38
- function maybeHexEncode(x) {
39
- if (x == null)
40
- return null;
41
- return (0, utils_js_1.hexEncode)(x);
42
- }
43
- function testStorage(s, params) {
44
- (0, vitest_1.beforeAll)(async () => {
45
- const logger = new utils_js_1.LogManager();
46
- logger.level = 0;
47
- s.setup?.(logger, reader_js_1.__tlReaderMap, writer_js_1.__tlWriterMap);
48
- await s.load?.();
49
- });
50
- (0, vitest_1.afterAll)(() => s.destroy?.());
51
- (0, vitest_1.beforeEach)(() => s.reset(true));
52
- (0, vitest_1.describe)('default dc', () => {
53
- (0, vitest_1.it)('should store', async () => {
54
- await s.setDefaultDcs(utils_js_1.defaultProductionDc);
55
- (0, vitest_1.expect)(await s.getDefaultDcs()).toEqual(utils_js_1.defaultProductionDc);
56
- });
57
- (0, vitest_1.it)('should remove', async () => {
58
- await s.setDefaultDcs(null);
59
- (0, vitest_1.expect)(await s.getDefaultDcs()).toBeNull();
60
- });
61
- });
62
- (0, vitest_1.describe)('auth keys', () => {
63
- (0, vitest_1.beforeAll)(() => void vitest_1.vi.useFakeTimers());
64
- (0, vitest_1.afterAll)(() => void vitest_1.vi.useRealTimers());
65
- const key2 = new Uint8Array(256).fill(0x42);
66
- const key3 = new Uint8Array(256).fill(0x43);
67
- const key2i0 = new Uint8Array(256).fill(0x44);
68
- const key2i1 = new Uint8Array(256).fill(0x45);
69
- const key3i0 = new Uint8Array(256).fill(0x46);
70
- const key3i1 = new Uint8Array(256).fill(0x47);
71
- (0, vitest_1.it)('should store perm auth key', async () => {
72
- await s.setAuthKeyFor(2, key2);
73
- await s.setAuthKeyFor(3, key3);
74
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(2))).toEqual((0, utils_js_1.hexEncode)(key2));
75
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual((0, utils_js_1.hexEncode)(key3));
76
- });
77
- (0, vitest_1.it)('should store temp auth keys', async () => {
78
- const expire = Date.now() + 1000;
79
- await s.setTempAuthKeyFor(2, 0, key2i0, expire);
80
- await s.setTempAuthKeyFor(2, 1, key2i1, expire);
81
- await s.setTempAuthKeyFor(3, 0, key3i0, expire);
82
- await s.setTempAuthKeyFor(3, 1, key3i1, expire);
83
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(2, 0))).toEqual((0, utils_js_1.hexEncode)(key2i0));
84
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(2, 1))).toEqual((0, utils_js_1.hexEncode)(key2i1));
85
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3, 0))).toEqual((0, utils_js_1.hexEncode)(key3i0));
86
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3, 1))).toEqual((0, utils_js_1.hexEncode)(key3i1));
87
- });
88
- (0, vitest_1.it)('should expire temp auth keys', async () => {
89
- const expire = Date.now() + 1000;
90
- await s.setTempAuthKeyFor(2, 0, key2i0, expire);
91
- await s.setTempAuthKeyFor(2, 1, key2i1, expire);
92
- await s.setTempAuthKeyFor(3, 0, key3i0, expire);
93
- await s.setTempAuthKeyFor(3, 1, key3i1, expire);
94
- vitest_1.vi.advanceTimersByTime(10000);
95
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 0)).toBeNull();
96
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 1)).toBeNull();
97
- (0, vitest_1.expect)(await s.getAuthKeyFor(3, 0)).toBeNull();
98
- (0, vitest_1.expect)(await s.getAuthKeyFor(3, 1)).toBeNull();
99
- });
100
- (0, vitest_1.it)('should remove auth keys', async () => {
101
- const expire = Date.now() + 1000;
102
- await s.setTempAuthKeyFor(2, 0, key2i0, expire);
103
- await s.setTempAuthKeyFor(2, 1, key2i1, expire);
104
- await s.setAuthKeyFor(2, key2);
105
- await s.setAuthKeyFor(3, key3);
106
- await s.setAuthKeyFor(2, null);
107
- await s.setTempAuthKeyFor(2, 0, null, 0);
108
- await s.setTempAuthKeyFor(2, 1, null, 0);
109
- (0, vitest_1.expect)(await s.getAuthKeyFor(2)).toBeNull();
110
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 0)).toBeNull();
111
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 1)).toBeNull();
112
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual((0, utils_js_1.hexEncode)(key3)); // should not be removed
113
- });
114
- (0, vitest_1.it)('should remove all auth keys with dropAuthKeysFor', async () => {
115
- const expire = Date.now() + 1000;
116
- await s.setTempAuthKeyFor(2, 0, key2i0, expire);
117
- await s.setTempAuthKeyFor(2, 1, key2i1, expire);
118
- await s.setAuthKeyFor(2, key2);
119
- await s.setAuthKeyFor(3, key3);
120
- await s.dropAuthKeysFor(2);
121
- (0, vitest_1.expect)(await s.getAuthKeyFor(2)).toBeNull();
122
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 0)).toBeNull();
123
- (0, vitest_1.expect)(await s.getAuthKeyFor(2, 1)).toBeNull();
124
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual((0, utils_js_1.hexEncode)(key3)); // should not be removed
125
- });
126
- (0, vitest_1.it)('should not reset auth keys on reset()', async () => {
127
- await s.setAuthKeyFor(2, key2);
128
- await s.setAuthKeyFor(3, key3);
129
- await s.reset();
130
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(2))).toEqual((0, utils_js_1.hexEncode)(key2));
131
- (0, vitest_1.expect)(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual((0, utils_js_1.hexEncode)(key3));
132
- });
133
- (0, vitest_1.it)('should reset auth keys on reset(true)', async () => {
134
- await s.setAuthKeyFor(2, key2);
135
- await s.setAuthKeyFor(3, key3);
136
- await s.reset(true);
137
- (0, vitest_1.expect)(await s.getAuthKeyFor(2)).toBeNull();
138
- (0, vitest_1.expect)(await s.getAuthKeyFor(3)).toBeNull();
139
- });
140
- });
141
- (0, vitest_1.describe)('future salts', () => {
142
- const someFutureSalt1 = long_1.default.fromBits(123, 456);
143
- const someFutureSalt2 = long_1.default.fromBits(789, 101112);
144
- const someFutureSalt3 = long_1.default.fromBits(131415, 161718);
145
- const someFutureSalt4 = long_1.default.fromBits(192021, 222324);
146
- const salts1 = [
147
- { _: 'mt_future_salt', validSince: 123, validUntil: 456, salt: someFutureSalt1 },
148
- { _: 'mt_future_salt', validSince: 789, validUntil: 101112, salt: someFutureSalt2 },
149
- ];
150
- const salts2 = [
151
- { _: 'mt_future_salt', validSince: 123, validUntil: 456, salt: someFutureSalt3 },
152
- { _: 'mt_future_salt', validSince: 789, validUntil: 101112, salt: someFutureSalt4 },
153
- ];
154
- (0, vitest_1.it)('should store and retrieve future salts', async () => {
155
- await s.setFutureSalts(1, salts1);
156
- await s.setFutureSalts(2, salts2);
157
- (0, vitest_1.expect)(await s.getFutureSalts(1)).toEqual(salts1);
158
- (0, vitest_1.expect)(await s.getFutureSalts(2)).toEqual(salts2);
159
- });
160
- });
161
- (0, vitest_1.describe)('peers', () => {
162
- (0, vitest_1.it)('should cache and return peers', async () => {
163
- await s.updatePeers([exports.stubPeerUser, peerChannel]);
164
- await s.save?.(); // update-related methods are batched, so we need to save
165
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual(peerUserInput);
166
- (0, vitest_1.expect)(await s.getPeerById(peerChannel.id)).toEqual(peerChannelInput);
167
- });
168
- (0, vitest_1.it)('should cache and return peers by username', async () => {
169
- await s.updatePeers([exports.stubPeerUser, peerChannel]);
170
- await s.save?.(); // update-related methods are batched, so we need to save
171
- (0, vitest_1.expect)(await s.getPeerByUsername(exports.stubPeerUser.username)).toEqual(peerUserInput);
172
- (0, vitest_1.expect)(await s.getPeerByUsername(peerChannel.username)).toEqual(peerChannelInput);
173
- });
174
- (0, vitest_1.it)('should cache and return peers by phone', async () => {
175
- await s.updatePeers([exports.stubPeerUser]);
176
- await s.save?.(); // update-related methods are batched, so we need to save
177
- (0, vitest_1.expect)(await s.getPeerByPhone(exports.stubPeerUser.phone)).toEqual(peerUserInput);
178
- });
179
- if (!params?.skipEntityOverwrite) {
180
- (0, vitest_1.it)('should overwrite existing cached peers', async () => {
181
- await s.updatePeers([exports.stubPeerUser]);
182
- await s.updatePeers([{ ...exports.stubPeerUser, username: 'whatever' }]);
183
- await s.save?.(); // update-related methods are batched, so we need to save
184
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual(peerUserInput);
185
- (0, vitest_1.expect)(await s.getPeerByUsername(exports.stubPeerUser.username)).toBeNull();
186
- (0, vitest_1.expect)(await s.getPeerByUsername('whatever')).toEqual(peerUserInput);
187
- });
188
- }
189
- (0, vitest_1.it)('should cache full peer info', async () => {
190
- await s.updatePeers([exports.stubPeerUser, peerChannel]);
191
- await s.save?.(); // update-related methods are batched, so we need to save
192
- (0, vitest_1.expect)({
193
- ...(await s.getFullPeerById(exports.stubPeerUser.id)),
194
- usernames: [],
195
- restrictionReason: [],
196
- }).toEqual(exports.stubPeerUser.full);
197
- (0, vitest_1.expect)({
198
- ...(await s.getFullPeerById(peerChannel.id)),
199
- usernames: [],
200
- restrictionReason: [],
201
- }).toEqual(peerChannel.full);
202
- });
203
- (0, vitest_1.describe)('min peers', () => {
204
- (0, vitest_1.it)('should generate *FromMessage constructors from reference messages', async () => {
205
- await s.updatePeers([peerChannel]);
206
- await s.saveReferenceMessage(exports.stubPeerUser.id, peerChannel.id, 456);
207
- await s.save?.(); // update-related methods are batched, so we need to save
208
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual({
209
- _: 'inputPeerUserFromMessage',
210
- peer: peerChannelInput,
211
- msgId: 456,
212
- userId: exports.stubPeerUser.id,
213
- });
214
- });
215
- (0, vitest_1.it)('should handle cases when referenced chat is not available', async () => {
216
- // this shouldn't really happen, but the storage should be able to handle it
217
- await s.saveReferenceMessage(exports.stubPeerUser.id, peerChannel.id, 456);
218
- await s.save?.(); // update-related methods are batched, so we need to save
219
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual(null);
220
- });
221
- (0, vitest_1.it)('should return full peer if it gets available', async () => {
222
- await s.updatePeers([peerChannel]);
223
- await s.saveReferenceMessage(exports.stubPeerUser.id, peerChannel.id, 456);
224
- await s.save?.(); // update-related methods are batched, so we need to save
225
- await s.updatePeers([exports.stubPeerUser]);
226
- await s.save?.();
227
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual(peerUserInput);
228
- });
229
- (0, vitest_1.it)('should handle cases when referenced message is deleted', async () => {
230
- await s.updatePeers([peerChannel]);
231
- await s.saveReferenceMessage(exports.stubPeerUser.id, peerChannel.id, 456);
232
- await s.save?.(); // update-related methods are batched, so we need to save
233
- await s.deleteReferenceMessages(peerChannel.id, [456]);
234
- await s.save?.();
235
- (0, vitest_1.expect)(await s.getPeerById(exports.stubPeerUser.id)).toEqual(null);
236
- });
237
- });
238
- });
239
- (0, vitest_1.describe)('current user', () => {
240
- const self = {
241
- userId: 123123,
242
- isBot: false,
243
- };
244
- (0, vitest_1.it)('should store and return self info', async () => {
245
- await s.setSelf(self);
246
- (0, vitest_1.expect)(await s.getSelf()).toEqual(self);
247
- });
248
- (0, vitest_1.it)('should remove self info', async () => {
249
- await s.setSelf(self);
250
- await s.setSelf(null);
251
- (0, vitest_1.expect)(await s.getSelf()).toBeNull();
252
- });
253
- });
254
- (0, vitest_1.describe)('updates state', () => {
255
- (0, vitest_1.it)('should store and return updates state', async () => {
256
- await s.setUpdatesPts(1);
257
- await s.setUpdatesQts(2);
258
- await s.setUpdatesDate(3);
259
- await s.setUpdatesSeq(4);
260
- await s.save?.(); // update-related methods are batched, so we need to save
261
- (0, vitest_1.expect)(await s.getUpdatesState()).toEqual([1, 2, 3, 4]);
262
- });
263
- (0, vitest_1.it)('should store and return channel pts', async () => {
264
- await s.setManyChannelPts(new Map([
265
- [1, 2],
266
- [3, 4],
267
- ]));
268
- await s.save?.(); // update-related methods are batched, so we need to save
269
- (0, vitest_1.expect)(await s.getChannelPts(1)).toEqual(2);
270
- (0, vitest_1.expect)(await s.getChannelPts(3)).toEqual(4);
271
- (0, vitest_1.expect)(await s.getChannelPts(2)).toBeNull();
272
- });
273
- (0, vitest_1.it)('should be null after reset', async () => {
274
- (0, vitest_1.expect)(await s.getUpdatesState()).toBeNull();
275
- });
276
- });
277
- params?.customTests?.(s);
278
- }
279
- exports.testStorage = testStorage;
280
- function testStateStorage(s) {
281
- (0, vitest_1.beforeAll)(async () => {
282
- const logger = new utils_js_1.LogManager();
283
- logger.level = 0;
284
- s.setup?.(logger, reader_js_1.__tlReaderMap, writer_js_1.__tlWriterMap);
285
- await s.load?.();
286
- });
287
- (0, vitest_1.afterAll)(() => s.destroy?.());
288
- (0, vitest_1.beforeEach)(() => s.reset());
289
- (0, vitest_1.describe)('key-value state', () => {
290
- (0, vitest_1.beforeAll)(() => void vitest_1.vi.useFakeTimers());
291
- (0, vitest_1.afterAll)(() => void vitest_1.vi.useRealTimers());
292
- (0, vitest_1.it)('should store and return state', async () => {
293
- await s.setState('a', 'b');
294
- await s.setState('c', 'd');
295
- await s.setState('e', 'f');
296
- (0, vitest_1.expect)(await s.getState('a')).toEqual('b');
297
- (0, vitest_1.expect)(await s.getState('c')).toEqual('d');
298
- (0, vitest_1.expect)(await s.getState('e')).toEqual('f');
299
- });
300
- (0, vitest_1.it)('should remove state', async () => {
301
- await s.setState('a', 'b');
302
- await s.setState('c', 'd');
303
- await s.setState('e', 'f');
304
- await s.deleteState('a');
305
- await s.deleteState('c');
306
- await s.deleteState('e');
307
- (0, vitest_1.expect)(await s.getState('a')).toBeNull();
308
- (0, vitest_1.expect)(await s.getState('c')).toBeNull();
309
- (0, vitest_1.expect)(await s.getState('e')).toBeNull();
310
- });
311
- (0, vitest_1.it)('should expire state', async () => {
312
- await s.setState('a', 'b', 1);
313
- await s.setState('c', 'd', 1);
314
- await s.setState('e', 'f', 1);
315
- vitest_1.vi.advanceTimersByTime(10000);
316
- (0, vitest_1.expect)(await s.getState('a')).toBeNull();
317
- (0, vitest_1.expect)(await s.getState('c')).toBeNull();
318
- (0, vitest_1.expect)(await s.getState('e')).toBeNull();
319
- });
320
- });
321
- (0, vitest_1.describe)('scenes', () => {
322
- (0, vitest_1.it)('should store and return scenes', async () => {
323
- await s.setCurrentScene('a', 'b');
324
- await s.setCurrentScene('c', 'd');
325
- await s.setCurrentScene('e', 'f');
326
- (0, vitest_1.expect)(await s.getCurrentScene('a')).toEqual('b');
327
- (0, vitest_1.expect)(await s.getCurrentScene('c')).toEqual('d');
328
- (0, vitest_1.expect)(await s.getCurrentScene('e')).toEqual('f');
329
- });
330
- (0, vitest_1.it)('should remove scenes', async () => {
331
- await s.setCurrentScene('a', 'b');
332
- await s.setCurrentScene('c', 'd');
333
- await s.setCurrentScene('e', 'f');
334
- await s.deleteCurrentScene('a');
335
- await s.deleteCurrentScene('c');
336
- await s.deleteCurrentScene('e');
337
- (0, vitest_1.expect)(await s.getCurrentScene('a')).toBeNull();
338
- (0, vitest_1.expect)(await s.getCurrentScene('c')).toBeNull();
339
- (0, vitest_1.expect)(await s.getCurrentScene('e')).toBeNull();
340
- });
341
- });
342
- (0, vitest_1.describe)('rate limit', () => {
343
- (0, vitest_1.beforeAll)(() => void vitest_1.vi.useFakeTimers());
344
- (0, vitest_1.afterAll)(() => void vitest_1.vi.useRealTimers());
345
- const check = () => s.getRateLimit('test', 3, 1);
346
- (0, vitest_1.it)('should implement basic rate limiting', async () => {
347
- vitest_1.vi.setSystemTime(0);
348
- (0, vitest_1.expect)(await check()).toEqual([3, 1000]);
349
- (0, vitest_1.expect)(await check()).toEqual([2, 1000]);
350
- (0, vitest_1.expect)(await check()).toEqual([1, 1000]);
351
- (0, vitest_1.expect)(await check()).toEqual([0, 1000]);
352
- vitest_1.vi.setSystemTime(1001);
353
- (0, vitest_1.expect)(await check()).toEqual([3, 2001]);
354
- });
355
- (0, vitest_1.it)('should allow resetting rate limit', async () => {
356
- vitest_1.vi.setSystemTime(0);
357
- await check();
358
- await check();
359
- await s.resetRateLimit('test');
360
- (0, vitest_1.expect)(await check()).toEqual([3, 1000]);
361
- });
362
- });
363
- }
364
- exports.testStateStorage = testStateStorage;
365
- //# sourceMappingURL=storage-test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"storage-test.js","sourceRoot":"","sources":["../../src/storage-test.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAuB;AACvB,mCAAkF;AAGlF,oDAAoH;AAEpH,2DAA2D;AAC3D,2DAA2D;AAE3D,uCAAsC;AAEzB,QAAA,YAAY,GAA8B;IACnD,EAAE,EAAE,MAAM;IACV,UAAU,EAAE,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;IACnC,IAAI,EAAE,MAAM;IACZ,QAAQ,EAAE,WAAW;IACrB,KAAK,EAAE,aAAa;IACpB,IAAI,EAAE,IAAA,oBAAU,EAAC,MAAM,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;CAC3C,CAAA;AACD,MAAM,aAAa,GAAqB;IACpC,CAAC,EAAE,eAAe;IAClB,MAAM,EAAE,MAAM;IACd,UAAU,EAAE,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;CACtC,CAAA;AAED,MAAM,WAAW,GAA8B;IAC3C,EAAE,EAAE,CAAC,aAAa;IAClB,UAAU,EAAE,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;IACnC,IAAI,EAAE,SAAS;IACf,QAAQ,EAAE,cAAc;IACxB,IAAI,EAAE,IAAA,oBAAU,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;CAC9C,CAAA;AAED,MAAM,gBAAgB,GAAqB;IACvC,CAAC,EAAE,kBAAkB;IACrB,SAAS,EAAE,UAAU;IACrB,UAAU,EAAE,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;CACtC,CAAA;AAED,SAAS,cAAc,CAAC,CAAoB;IACxC,IAAI,CAAC,IAAI,IAAI;QAAE,OAAO,IAAI,CAAA;IAE1B,OAAO,IAAA,oBAAS,EAAC,CAAC,CAAC,CAAA;AACvB,CAAC;AAED,SAAgB,WAAW,CACvB,CAAI,EACJ,MAGC;IAED,IAAA,kBAAS,EAAC,KAAK,IAAI,EAAE;QACjB,MAAM,MAAM,GAAG,IAAI,qBAAU,EAAE,CAAA;QAC/B,MAAM,CAAC,KAAK,GAAG,CAAC,CAAA;QAChB,CAAC,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,yBAAa,EAAE,yBAAa,CAAC,CAAA;QAE/C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAA;IAC7B,IAAA,mBAAU,EAAC,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;IAE/B,IAAA,iBAAQ,EAAC,YAAY,EAAE,GAAG,EAAE;QACxB,IAAA,WAAE,EAAC,cAAc,EAAE,KAAK,IAAI,EAAE;YAC1B,MAAM,CAAC,CAAC,aAAa,CAAC,8BAAmB,CAAC,CAAA;YAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,8BAAmB,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,eAAe,EAAE,KAAK,IAAI,EAAE;YAC3B,MAAM,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;YAC3B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAA;QAC9C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,WAAW,EAAE,GAAG,EAAE;QACvB,IAAA,kBAAS,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QACxC,IAAA,iBAAQ,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QAEvC,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3C,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE3C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC7C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC7C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC7C,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE7C,IAAA,WAAE,EAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAE9B,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA;YACzE,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA;QAC7E,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;YAEhC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAE/C,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,MAAM,CAAC,CAAC,CAAA;YAC9E,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,MAAM,CAAC,CAAC,CAAA;YAC9E,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,MAAM,CAAC,CAAC,CAAA;YAC9E,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,MAAM,CAAC,CAAC,CAAA;QAClF,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;YAEhC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAE/C,WAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAA;YAE7B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;YAEhC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAE9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;YACxC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;YAExC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA,CAAC,wBAAwB;QACtG,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAA;YAEhC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;YAC/C,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAE9B,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YAE1B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC9C,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA,CAAC,wBAAwB;QACtG,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,KAAK,EAAE,CAAA;YAEf,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA;YACzE,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAA,oBAAS,EAAC,IAAI,CAAC,CAAC,CAAA;QAC7E,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAEnB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QAC/C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,cAAc,EAAE,GAAG,EAAE;QAC1B,MAAM,eAAe,GAAG,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;QAC/C,MAAM,eAAe,GAAG,cAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;QAClD,MAAM,eAAe,GAAG,cAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QACrD,MAAM,eAAe,GAAG,cAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAErD,MAAM,MAAM,GAA4B;YACpC,EAAE,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,IAAI,EAAE,eAAe,EAAE;YAChF,EAAE,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE;SACtF,CAAA;QACD,MAAM,MAAM,GAA4B;YACpC,EAAE,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,IAAI,EAAE,eAAe,EAAE;YAChF,EAAE,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE;SACtF,CAAA;QAED,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;YACjC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;YAEjC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;YACjD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACrD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,OAAO,EAAE,GAAG,EAAE;QACnB,IAAA,WAAE,EAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,EAAE,WAAW,CAAC,CAAC,CAAA;YAChD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;YACnE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;QACzE,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,EAAE,WAAW,CAAC,CAAC,CAAA;YAChD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,oBAAY,CAAC,QAAS,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;YAChF,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,WAAW,CAAC,QAAS,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;QACtF,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,CAAC,CAAC,CAAA;YACnC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,cAAc,CAAC,oBAAY,CAAC,KAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;QAC9E,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE;YAC9B,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;gBACpD,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,CAAC,CAAC,CAAA;gBACnC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,EAAE,GAAG,oBAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC,CAAA;gBAChE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;gBAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;gBACnE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,oBAAY,CAAC,QAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;gBACpE,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;YACxE,CAAC,CAAC,CAAA;SACL;QAED,IAAA,WAAE,EAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,EAAE,WAAW,CAAC,CAAC,CAAA;YAChD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC;gBACH,GAAG,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC;gBAC7C,SAAS,EAAE,EAAE;gBACb,iBAAiB,EAAE,EAAE;aACxB,CAAC,CAAC,OAAO,CAAC,oBAAY,CAAC,IAAI,CAAC,CAAA;YAC7B,IAAA,eAAM,EAAC;gBACH,GAAG,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;gBAC5C,SAAS,EAAE,EAAE;gBACb,iBAAiB,EAAE,EAAE;aACxB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,IAAA,iBAAQ,EAAC,WAAW,EAAE,GAAG,EAAE;YACvB,IAAA,WAAE,EAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;gBAC/E,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC,CAAC,CAAA;gBAClC,MAAM,CAAC,CAAC,oBAAoB,CAAC,oBAAY,CAAC,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;gBAClE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;gBAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;oBACjD,CAAC,EAAE,0BAA0B;oBAC7B,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE,GAAG;oBACV,MAAM,EAAE,oBAAY,CAAC,EAAE;iBAC1B,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YAEF,IAAA,WAAE,EAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;gBACvE,4EAA4E;gBAC5E,MAAM,CAAC,CAAC,oBAAoB,CAAC,oBAAY,CAAC,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;gBAClE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;gBAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YAC9D,CAAC,CAAC,CAAA;YAEF,IAAA,WAAE,EAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;gBAC1D,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC,CAAC,CAAA;gBAClC,MAAM,CAAC,CAAC,oBAAoB,CAAC,oBAAY,CAAC,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;gBAClE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;gBAE1E,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,oBAAY,CAAC,CAAC,CAAA;gBACnC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA;gBAEhB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;YACvE,CAAC,CAAC,CAAA;YAEF,IAAA,WAAE,EAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;gBACpE,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,WAAW,CAAC,CAAC,CAAA;gBAClC,MAAM,CAAC,CAAC,oBAAoB,CAAC,oBAAY,CAAC,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;gBAClE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;gBAE1E,MAAM,CAAC,CAAC,uBAAuB,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAA;gBACtD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA;gBAEhB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,CAAC,oBAAY,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YAC9D,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,cAAc,EAAE,GAAG,EAAE;QAC1B,MAAM,IAAI,GAA8B;YACpC,MAAM,EAAE,MAAM;YACd,KAAK,EAAE,KAAK;SACf,CAAA;QAED,IAAA,WAAE,EAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACrB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACrB,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACrB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAA;QACxC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,eAAe,EAAE,GAAG,EAAE;QAC3B,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;YACxB,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;YACxB,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAA;YACzB,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;YACxB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,CAAC,CAAC,iBAAiB,CACrB,IAAI,GAAG,CAAC;gBACJ,CAAC,CAAC,EAAE,CAAC,CAAC;gBACN,CAAC,CAAC,EAAE,CAAC,CAAC;aACT,CAAC,CACL,CAAA;YACD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA,CAAC,yDAAyD;YAE1E,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;YAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;YAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QAC/C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YACxC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAA;QAChD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAA;AAC5B,CAAC;AAhTD,kCAgTC;AA8BD,SAAgB,gBAAgB,CAAC,CAAgB;IAC7C,IAAA,kBAAS,EAAC,KAAK,IAAI,EAAE;QACjB,MAAM,MAAM,GAAG,IAAI,qBAAU,EAAE,CAAA;QAC/B,MAAM,CAAC,KAAK,GAAG,CAAC,CAAA;QAChB,CAAC,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,yBAAa,EAAE,yBAAa,CAAC,CAAA;QAE/C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAA;IAC7B,IAAA,mBAAU,EAAC,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAE3B,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;QAC7B,IAAA,kBAAS,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QACxC,IAAA,iBAAQ,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QAEvC,IAAA,WAAE,EAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAC1B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAC1B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAE1B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAC1B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAC1B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAE1B,MAAM,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAA;YACxB,MAAM,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAA;YACxB,MAAM,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAA;YAExB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;YAC7B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;YAC7B,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;YAE7B,WAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAA;YAE7B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QAC5C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,QAAQ,EAAE,GAAG,EAAE;QACpB,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YACjC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YACjC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAEjC,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YACjD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;YACjD,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;QACrD,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YACjC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YACjC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;YAEjC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;YAC/B,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;YAC/B,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;YAE/B,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;YAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;QACnD,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,YAAY,EAAE,GAAG,EAAE;QACxB,IAAA,kBAAS,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QACxC,IAAA,iBAAQ,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;QAEvC,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAEhD,IAAA,WAAE,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YAClD,WAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;YAEnB,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;YACxC,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;YAExC,WAAE,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;YAEtB,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YAC/C,WAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;YAEnB,MAAM,KAAK,EAAE,CAAA;YACb,MAAM,KAAK,EAAE,CAAA;YAEb,MAAM,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;YAC9B,IAAA,eAAM,EAAC,MAAM,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC;AA5GD,4CA4GC","sourcesContent":["import Long from 'long'\nimport { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest'\n\nimport { ITelegramStorage, MaybeAsync } from '@mtcute/core'\nimport { defaultProductionDc, hexEncode, Logger, LogManager, TlReaderMap, TlWriterMap } from '@mtcute/core/utils.js'\nimport { mtp, tl } from '@mtcute/tl'\nimport { __tlReaderMap } from '@mtcute/tl/binary/reader.js'\nimport { __tlWriterMap } from '@mtcute/tl/binary/writer.js'\n\nimport { createStub } from './stub.js'\n\nexport const stubPeerUser: ITelegramStorage.PeerInfo = {\n id: 123123,\n accessHash: Long.fromBits(123, 456),\n type: 'user',\n username: 'some_user',\n phone: '78005553535',\n full: createStub('user', { id: 123123 }),\n}\nconst peerUserInput: tl.TypeInputPeer = {\n _: 'inputPeerUser',\n userId: 123123,\n accessHash: Long.fromBits(123, 456),\n}\n\nconst peerChannel: ITelegramStorage.PeerInfo = {\n id: -1001183945448,\n accessHash: Long.fromBits(666, 555),\n type: 'channel',\n username: 'some_channel',\n full: createStub('channel', { id: 123123 }),\n}\n\nconst peerChannelInput: tl.TypeInputPeer = {\n _: 'inputPeerChannel',\n channelId: 1183945448,\n accessHash: Long.fromBits(666, 555),\n}\n\nfunction maybeHexEncode(x: Uint8Array | null): string | null {\n if (x == null) return null\n\n return hexEncode(x)\n}\n\nexport function testStorage<T extends ITelegramStorage>(\n s: T,\n params?: {\n skipEntityOverwrite?: boolean\n customTests?: (s: T) => void\n },\n): void {\n beforeAll(async () => {\n const logger = new LogManager()\n logger.level = 0\n s.setup?.(logger, __tlReaderMap, __tlWriterMap)\n\n await s.load?.()\n })\n\n afterAll(() => s.destroy?.())\n beforeEach(() => s.reset(true))\n\n describe('default dc', () => {\n it('should store', async () => {\n await s.setDefaultDcs(defaultProductionDc)\n expect(await s.getDefaultDcs()).toEqual(defaultProductionDc)\n })\n\n it('should remove', async () => {\n await s.setDefaultDcs(null)\n expect(await s.getDefaultDcs()).toBeNull()\n })\n })\n\n describe('auth keys', () => {\n beforeAll(() => void vi.useFakeTimers())\n afterAll(() => void vi.useRealTimers())\n\n const key2 = new Uint8Array(256).fill(0x42)\n const key3 = new Uint8Array(256).fill(0x43)\n\n const key2i0 = new Uint8Array(256).fill(0x44)\n const key2i1 = new Uint8Array(256).fill(0x45)\n const key3i0 = new Uint8Array(256).fill(0x46)\n const key3i1 = new Uint8Array(256).fill(0x47)\n\n it('should store perm auth key', async () => {\n await s.setAuthKeyFor(2, key2)\n await s.setAuthKeyFor(3, key3)\n\n expect(maybeHexEncode(await s.getAuthKeyFor(2))).toEqual(hexEncode(key2))\n expect(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual(hexEncode(key3))\n })\n\n it('should store temp auth keys', async () => {\n const expire = Date.now() + 1000\n\n await s.setTempAuthKeyFor(2, 0, key2i0, expire)\n await s.setTempAuthKeyFor(2, 1, key2i1, expire)\n await s.setTempAuthKeyFor(3, 0, key3i0, expire)\n await s.setTempAuthKeyFor(3, 1, key3i1, expire)\n\n expect(maybeHexEncode(await s.getAuthKeyFor(2, 0))).toEqual(hexEncode(key2i0))\n expect(maybeHexEncode(await s.getAuthKeyFor(2, 1))).toEqual(hexEncode(key2i1))\n expect(maybeHexEncode(await s.getAuthKeyFor(3, 0))).toEqual(hexEncode(key3i0))\n expect(maybeHexEncode(await s.getAuthKeyFor(3, 1))).toEqual(hexEncode(key3i1))\n })\n\n it('should expire temp auth keys', async () => {\n const expire = Date.now() + 1000\n\n await s.setTempAuthKeyFor(2, 0, key2i0, expire)\n await s.setTempAuthKeyFor(2, 1, key2i1, expire)\n await s.setTempAuthKeyFor(3, 0, key3i0, expire)\n await s.setTempAuthKeyFor(3, 1, key3i1, expire)\n\n vi.advanceTimersByTime(10000)\n\n expect(await s.getAuthKeyFor(2, 0)).toBeNull()\n expect(await s.getAuthKeyFor(2, 1)).toBeNull()\n expect(await s.getAuthKeyFor(3, 0)).toBeNull()\n expect(await s.getAuthKeyFor(3, 1)).toBeNull()\n })\n\n it('should remove auth keys', async () => {\n const expire = Date.now() + 1000\n\n await s.setTempAuthKeyFor(2, 0, key2i0, expire)\n await s.setTempAuthKeyFor(2, 1, key2i1, expire)\n await s.setAuthKeyFor(2, key2)\n await s.setAuthKeyFor(3, key3)\n\n await s.setAuthKeyFor(2, null)\n await s.setTempAuthKeyFor(2, 0, null, 0)\n await s.setTempAuthKeyFor(2, 1, null, 0)\n\n expect(await s.getAuthKeyFor(2)).toBeNull()\n expect(await s.getAuthKeyFor(2, 0)).toBeNull()\n expect(await s.getAuthKeyFor(2, 1)).toBeNull()\n expect(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual(hexEncode(key3)) // should not be removed\n })\n\n it('should remove all auth keys with dropAuthKeysFor', async () => {\n const expire = Date.now() + 1000\n\n await s.setTempAuthKeyFor(2, 0, key2i0, expire)\n await s.setTempAuthKeyFor(2, 1, key2i1, expire)\n await s.setAuthKeyFor(2, key2)\n await s.setAuthKeyFor(3, key3)\n\n await s.dropAuthKeysFor(2)\n\n expect(await s.getAuthKeyFor(2)).toBeNull()\n expect(await s.getAuthKeyFor(2, 0)).toBeNull()\n expect(await s.getAuthKeyFor(2, 1)).toBeNull()\n expect(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual(hexEncode(key3)) // should not be removed\n })\n\n it('should not reset auth keys on reset()', async () => {\n await s.setAuthKeyFor(2, key2)\n await s.setAuthKeyFor(3, key3)\n await s.reset()\n\n expect(maybeHexEncode(await s.getAuthKeyFor(2))).toEqual(hexEncode(key2))\n expect(maybeHexEncode(await s.getAuthKeyFor(3))).toEqual(hexEncode(key3))\n })\n\n it('should reset auth keys on reset(true)', async () => {\n await s.setAuthKeyFor(2, key2)\n await s.setAuthKeyFor(3, key3)\n await s.reset(true)\n\n expect(await s.getAuthKeyFor(2)).toBeNull()\n expect(await s.getAuthKeyFor(3)).toBeNull()\n })\n })\n\n describe('future salts', () => {\n const someFutureSalt1 = Long.fromBits(123, 456)\n const someFutureSalt2 = Long.fromBits(789, 101112)\n const someFutureSalt3 = Long.fromBits(131415, 161718)\n const someFutureSalt4 = Long.fromBits(192021, 222324)\n\n const salts1: mtp.RawMt_future_salt[] = [\n { _: 'mt_future_salt', validSince: 123, validUntil: 456, salt: someFutureSalt1 },\n { _: 'mt_future_salt', validSince: 789, validUntil: 101112, salt: someFutureSalt2 },\n ]\n const salts2: mtp.RawMt_future_salt[] = [\n { _: 'mt_future_salt', validSince: 123, validUntil: 456, salt: someFutureSalt3 },\n { _: 'mt_future_salt', validSince: 789, validUntil: 101112, salt: someFutureSalt4 },\n ]\n\n it('should store and retrieve future salts', async () => {\n await s.setFutureSalts(1, salts1)\n await s.setFutureSalts(2, salts2)\n\n expect(await s.getFutureSalts(1)).toEqual(salts1)\n expect(await s.getFutureSalts(2)).toEqual(salts2)\n })\n })\n\n describe('peers', () => {\n it('should cache and return peers', async () => {\n await s.updatePeers([stubPeerUser, peerChannel])\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual(peerUserInput)\n expect(await s.getPeerById(peerChannel.id)).toEqual(peerChannelInput)\n })\n\n it('should cache and return peers by username', async () => {\n await s.updatePeers([stubPeerUser, peerChannel])\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerByUsername(stubPeerUser.username!)).toEqual(peerUserInput)\n expect(await s.getPeerByUsername(peerChannel.username!)).toEqual(peerChannelInput)\n })\n\n it('should cache and return peers by phone', async () => {\n await s.updatePeers([stubPeerUser])\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerByPhone(stubPeerUser.phone!)).toEqual(peerUserInput)\n })\n\n if (!params?.skipEntityOverwrite) {\n it('should overwrite existing cached peers', async () => {\n await s.updatePeers([stubPeerUser])\n await s.updatePeers([{ ...stubPeerUser, username: 'whatever' }])\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual(peerUserInput)\n expect(await s.getPeerByUsername(stubPeerUser.username!)).toBeNull()\n expect(await s.getPeerByUsername('whatever')).toEqual(peerUserInput)\n })\n }\n\n it('should cache full peer info', async () => {\n await s.updatePeers([stubPeerUser, peerChannel])\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect({\n ...(await s.getFullPeerById(stubPeerUser.id)),\n usernames: [],\n restrictionReason: [],\n }).toEqual(stubPeerUser.full)\n expect({\n ...(await s.getFullPeerById(peerChannel.id)),\n usernames: [],\n restrictionReason: [],\n }).toEqual(peerChannel.full)\n })\n\n describe('min peers', () => {\n it('should generate *FromMessage constructors from reference messages', async () => {\n await s.updatePeers([peerChannel])\n await s.saveReferenceMessage(stubPeerUser.id, peerChannel.id, 456)\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual({\n _: 'inputPeerUserFromMessage',\n peer: peerChannelInput,\n msgId: 456,\n userId: stubPeerUser.id,\n })\n })\n\n it('should handle cases when referenced chat is not available', async () => {\n // this shouldn't really happen, but the storage should be able to handle it\n await s.saveReferenceMessage(stubPeerUser.id, peerChannel.id, 456)\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual(null)\n })\n\n it('should return full peer if it gets available', async () => {\n await s.updatePeers([peerChannel])\n await s.saveReferenceMessage(stubPeerUser.id, peerChannel.id, 456)\n await s.save?.() // update-related methods are batched, so we need to save\n\n await s.updatePeers([stubPeerUser])\n await s.save?.()\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual(peerUserInput)\n })\n\n it('should handle cases when referenced message is deleted', async () => {\n await s.updatePeers([peerChannel])\n await s.saveReferenceMessage(stubPeerUser.id, peerChannel.id, 456)\n await s.save?.() // update-related methods are batched, so we need to save\n\n await s.deleteReferenceMessages(peerChannel.id, [456])\n await s.save?.()\n\n expect(await s.getPeerById(stubPeerUser.id)).toEqual(null)\n })\n })\n })\n\n describe('current user', () => {\n const self: ITelegramStorage.SelfInfo = {\n userId: 123123,\n isBot: false,\n }\n\n it('should store and return self info', async () => {\n await s.setSelf(self)\n expect(await s.getSelf()).toEqual(self)\n })\n\n it('should remove self info', async () => {\n await s.setSelf(self)\n await s.setSelf(null)\n expect(await s.getSelf()).toBeNull()\n })\n })\n\n describe('updates state', () => {\n it('should store and return updates state', async () => {\n await s.setUpdatesPts(1)\n await s.setUpdatesQts(2)\n await s.setUpdatesDate(3)\n await s.setUpdatesSeq(4)\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getUpdatesState()).toEqual([1, 2, 3, 4])\n })\n\n it('should store and return channel pts', async () => {\n await s.setManyChannelPts(\n new Map([\n [1, 2],\n [3, 4],\n ]),\n )\n await s.save?.() // update-related methods are batched, so we need to save\n\n expect(await s.getChannelPts(1)).toEqual(2)\n expect(await s.getChannelPts(3)).toEqual(4)\n expect(await s.getChannelPts(2)).toBeNull()\n })\n\n it('should be null after reset', async () => {\n expect(await s.getUpdatesState()).toBeNull()\n })\n })\n\n params?.customTests?.(s)\n}\n\ninterface IStateStorage {\n setup?(log: Logger, readerMap: TlReaderMap, writerMap: TlWriterMap): void\n\n load?(): MaybeAsync<void>\n\n save?(): MaybeAsync<void>\n\n destroy?(): MaybeAsync<void>\n\n reset(): MaybeAsync<void>\n\n getState(key: string): MaybeAsync<unknown>\n\n setState(key: string, state: unknown, ttl?: number): MaybeAsync<void>\n\n deleteState(key: string): MaybeAsync<void>\n\n getCurrentScene(key: string): MaybeAsync<string | null>\n\n setCurrentScene(key: string, scene: string, ttl?: number): MaybeAsync<void>\n\n deleteCurrentScene(key: string): MaybeAsync<void>\n\n getRateLimit(key: string, limit: number, window: number): MaybeAsync<[number, number]>\n\n resetRateLimit(key: string): MaybeAsync<void>\n}\n\nexport function testStateStorage(s: IStateStorage) {\n beforeAll(async () => {\n const logger = new LogManager()\n logger.level = 0\n s.setup?.(logger, __tlReaderMap, __tlWriterMap)\n\n await s.load?.()\n })\n\n afterAll(() => s.destroy?.())\n beforeEach(() => s.reset())\n\n describe('key-value state', () => {\n beforeAll(() => void vi.useFakeTimers())\n afterAll(() => void vi.useRealTimers())\n\n it('should store and return state', async () => {\n await s.setState('a', 'b')\n await s.setState('c', 'd')\n await s.setState('e', 'f')\n\n expect(await s.getState('a')).toEqual('b')\n expect(await s.getState('c')).toEqual('d')\n expect(await s.getState('e')).toEqual('f')\n })\n\n it('should remove state', async () => {\n await s.setState('a', 'b')\n await s.setState('c', 'd')\n await s.setState('e', 'f')\n\n await s.deleteState('a')\n await s.deleteState('c')\n await s.deleteState('e')\n\n expect(await s.getState('a')).toBeNull()\n expect(await s.getState('c')).toBeNull()\n expect(await s.getState('e')).toBeNull()\n })\n\n it('should expire state', async () => {\n await s.setState('a', 'b', 1)\n await s.setState('c', 'd', 1)\n await s.setState('e', 'f', 1)\n\n vi.advanceTimersByTime(10000)\n\n expect(await s.getState('a')).toBeNull()\n expect(await s.getState('c')).toBeNull()\n expect(await s.getState('e')).toBeNull()\n })\n })\n\n describe('scenes', () => {\n it('should store and return scenes', async () => {\n await s.setCurrentScene('a', 'b')\n await s.setCurrentScene('c', 'd')\n await s.setCurrentScene('e', 'f')\n\n expect(await s.getCurrentScene('a')).toEqual('b')\n expect(await s.getCurrentScene('c')).toEqual('d')\n expect(await s.getCurrentScene('e')).toEqual('f')\n })\n\n it('should remove scenes', async () => {\n await s.setCurrentScene('a', 'b')\n await s.setCurrentScene('c', 'd')\n await s.setCurrentScene('e', 'f')\n\n await s.deleteCurrentScene('a')\n await s.deleteCurrentScene('c')\n await s.deleteCurrentScene('e')\n\n expect(await s.getCurrentScene('a')).toBeNull()\n expect(await s.getCurrentScene('c')).toBeNull()\n expect(await s.getCurrentScene('e')).toBeNull()\n })\n })\n\n describe('rate limit', () => {\n beforeAll(() => void vi.useFakeTimers())\n afterAll(() => void vi.useRealTimers())\n\n const check = () => s.getRateLimit('test', 3, 1)\n\n it('should implement basic rate limiting', async () => {\n vi.setSystemTime(0)\n\n expect(await check()).toEqual([3, 1000])\n expect(await check()).toEqual([2, 1000])\n expect(await check()).toEqual([1, 1000])\n expect(await check()).toEqual([0, 1000])\n\n vi.setSystemTime(1001)\n\n expect(await check()).toEqual([3, 2001])\n })\n\n it('should allow resetting rate limit', async () => {\n vi.setSystemTime(0)\n\n await check()\n await check()\n\n await s.resetRateLimit('test')\n expect(await check()).toEqual([3, 1000])\n })\n })\n}\n"]}
@@ -1 +0,0 @@
1
- export {};
@@ -1,43 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- const vitest_1 = require("vitest");
4
- const core_1 = require("@mtcute/core");
5
- const storage_js_1 = require("./storage.js");
6
- const stub_js_1 = require("./stub.js");
7
- const transport_js_1 = require("./transport.js");
8
- (0, vitest_1.describe)('storage stub', () => {
9
- (0, vitest_1.it)('should correctly intercept calls', async () => {
10
- const log = [];
11
- const client = new core_1.BaseTelegramClient({
12
- apiId: 0,
13
- apiHash: '',
14
- logLevel: 0,
15
- defaultDcs: {
16
- main: (0, stub_js_1.createStub)('dcOption', { ipAddress: '1.2.3.4', port: 1234 }),
17
- media: (0, stub_js_1.createStub)('dcOption', { ipAddress: '1.2.3.4', port: 5678 }),
18
- },
19
- transport: () => new transport_js_1.StubTelegramTransport({
20
- onMessage: (msg) => {
21
- if (msg.slice(0, 8).reduce((a, b) => a + b, 0) === 0) {
22
- // should not happen, since we're providing stub keys
23
- log.push('unauthed_message');
24
- }
25
- setTimeout(() => {
26
- client.close().catch(() => { });
27
- }, 10);
28
- },
29
- }),
30
- storage: new storage_js_1.StubMemoryTelegramStorage({
31
- hasKeys: true,
32
- onLoad: () => log.push('load'),
33
- onSave: () => log.push('save'),
34
- onDestroy: () => log.push('destroy'),
35
- onReset: () => log.push('reset'),
36
- }),
37
- });
38
- await client.connect();
39
- await client.call({ _: 'help.getConfig' }).catch(() => { }); // ignore "client closed" error
40
- (0, vitest_1.expect)(log).toEqual(['load', 'save', 'destroy']);
41
- });
42
- });
43
- //# sourceMappingURL=storage.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"storage.test.js","sourceRoot":"","sources":["../../src/storage.test.ts"],"names":[],"mappings":";;AAAA,mCAA6C;AAE7C,uCAAiD;AAEjD,6CAAwD;AACxD,uCAAsC;AACtC,iDAAsD;AAEtD,IAAA,iBAAQ,EAAC,cAAc,EAAE,GAAG,EAAE;IAC1B,IAAA,WAAE,EAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,GAAG,GAAa,EAAE,CAAA;QAExB,MAAM,MAAM,GAAG,IAAI,yBAAkB,CAAC;YAClC,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,EAAE;YACX,QAAQ,EAAE,CAAC;YACX,UAAU,EAAE;gBACR,IAAI,EAAE,IAAA,oBAAU,EAAC,UAAU,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAClE,KAAK,EAAE,IAAA,oBAAU,EAAC,UAAU,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aACtE;YACD,SAAS,EAAE,GAAG,EAAE,CACZ,IAAI,oCAAqB,CAAC;gBACtB,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;oBACf,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE;wBAClD,qDAAqD;wBACrD,GAAG,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;qBAC/B;oBACD,UAAU,CAAC,GAAG,EAAE;wBACZ,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;oBAClC,CAAC,EAAE,EAAE,CAAC,CAAA;gBACV,CAAC;aACJ,CAAC;YACN,OAAO,EAAE,IAAI,sCAAyB,CAAC;gBACnC,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC9B,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC9B,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;gBACpC,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC;aACnC,CAAC;SACL,CAAC,CAAA;QAEF,MAAM,MAAM,CAAC,OAAO,EAAE,CAAA;QACtB,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA,CAAC,+BAA+B;QAE1F,IAAA,eAAM,EAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { describe, expect, it } from 'vitest'\n\nimport { BaseTelegramClient } from '@mtcute/core'\n\nimport { StubMemoryTelegramStorage } from './storage.js'\nimport { createStub } from './stub.js'\nimport { StubTelegramTransport } from './transport.js'\n\ndescribe('storage stub', () => {\n it('should correctly intercept calls', async () => {\n const log: string[] = []\n\n const client = new BaseTelegramClient({\n apiId: 0,\n apiHash: '',\n logLevel: 0,\n defaultDcs: {\n main: createStub('dcOption', { ipAddress: '1.2.3.4', port: 1234 }),\n media: createStub('dcOption', { ipAddress: '1.2.3.4', port: 5678 }),\n },\n transport: () =>\n new StubTelegramTransport({\n onMessage: (msg) => {\n if (msg.slice(0, 8).reduce((a, b) => a + b, 0) === 0) {\n // should not happen, since we're providing stub keys\n log.push('unauthed_message')\n }\n setTimeout(() => {\n client.close().catch(() => {})\n }, 10)\n },\n }),\n storage: new StubMemoryTelegramStorage({\n hasKeys: true,\n onLoad: () => log.push('load'),\n onSave: () => log.push('save'),\n onDestroy: () => log.push('destroy'),\n onReset: () => log.push('reset'),\n }),\n })\n\n await client.connect()\n await client.call({ _: 'help.getConfig' }).catch(() => {}) // ignore \"client closed\" error\n\n expect(log).toEqual(['load', 'save', 'destroy'])\n })\n})\n"]}
@@ -1,24 +0,0 @@
1
- import { ITelegramStorage, MaybeAsync } from '@mtcute/core';
2
- import { Logger, TlReaderMap, TlWriterMap } from '@mtcute/core/utils.js';
3
- export declare const stubPeerUser: ITelegramStorage.PeerInfo;
4
- export declare function testStorage<T extends ITelegramStorage>(s: T, params?: {
5
- skipEntityOverwrite?: boolean;
6
- customTests?: (s: T) => void;
7
- }): void;
8
- interface IStateStorage {
9
- setup?(log: Logger, readerMap: TlReaderMap, writerMap: TlWriterMap): void;
10
- load?(): MaybeAsync<void>;
11
- save?(): MaybeAsync<void>;
12
- destroy?(): MaybeAsync<void>;
13
- reset(): MaybeAsync<void>;
14
- getState(key: string): MaybeAsync<unknown>;
15
- setState(key: string, state: unknown, ttl?: number): MaybeAsync<void>;
16
- deleteState(key: string): MaybeAsync<void>;
17
- getCurrentScene(key: string): MaybeAsync<string | null>;
18
- setCurrentScene(key: string, scene: string, ttl?: number): MaybeAsync<void>;
19
- deleteCurrentScene(key: string): MaybeAsync<void>;
20
- getRateLimit(key: string, limit: number, window: number): MaybeAsync<[number, number]>;
21
- resetRateLimit(key: string): MaybeAsync<void>;
22
- }
23
- export declare function testStateStorage(s: IStateStorage): void;
24
- export {};