test(store): demonstrate native claim composition (#799)

* feat(store): native tryClaim fast path for redis and upstash via setNx

Store.tryClaim falls back to a read-modify-write over `update` when an adapter
has no native primitive. On redis/upstash that fallback needs WATCH/MULTI (and
Upstash REST has no WATCH at all), while a replay claim is really just one
atomic set-if-absent with an expiry.

Add an optional `setNx(key, value, expires)` to the redis and upstash
Parameters. When present, the adapter exposes a native tryClaim that maps to
`SET ... PXAT <expires> NX` (or upstash `set(key, value, { nx: true, pxat })`).
`expires` is an absolute Unix-ms timestamp, so it maps to PXAT/pxat, never a
relative PX. Existing callers passing `{ get, set, del, update? }` are
unaffected; the key-prefix wrapper already forwards a native tryClaim.

* test(store): cover native claim composition

* chore: add empty changeset

* chore: format empty changeset

---------

Co-authored-by: EfeDurmaz16 <efebarandurmaz05@gmail.com>

Author: brendanjryan

.changeset/clean-stores-compose.md

@@ -0,0 +1,2 @@
+---
+---

src/Store.test-d.ts

@@ -35,6 +35,22 @@ test('from with keyPrefix preserves AtomicStore', () => {
   expectTypeOf(store).toEqualTypeOf<Store.AtomicStore>()
 })
 
+test('from preserves a composed tryClaim implementation', () => {
+  const adapter = Store.redis({
+    get: async () => null,
+    set: async () => null,
+    del: async () => null,
+    update: async (_key, fn) => fn(null).result,
+  })
+  const store = Store.from({
+    ...adapter,
+    tryClaim: async (_key: string, _expires: number) => true,
+  })
+
+  expectTypeOf(store.tryClaim).toEqualTypeOf<(key: string, expires: number) => Promise<boolean>>()
+  expectTypeOf(Store.tryClaim(store, 'k', 1)).toEqualTypeOf<boolean | Promise<boolean>>()
+})
+
 test('from with keyPrefix preserves typed Store', () => {
   type ItemMap = { [key: `mppx:charge:${string}`]: number }
   const store = {} as Store.Store<ItemMap>

src/Store.test.ts

@@ -60,6 +60,16 @@ function fakeUnknownKv() {
   }
 }
 
+function fakeTryClaim(seen?: string[]) {
+  const claimed = new Set<string>()
+  return async (key: string, _expires: number) => {
+    seen?.push(key)
+    if (claimed.has(key)) return false
+    claimed.add(key)
+    return true
+  }
+}
+
 describe.each([
   { label: 'memory', create: () => Store.memory() },
   { label: 'cloudflare', create: () => Store.cloudflare(fakeStringKv()) },
@@ -237,6 +247,59 @@ describe('tryClaim', () => {
 
     expect(await Store.tryClaim(store, 'k', Date.now() + 60_000)).toBe(false)
   })
+
+  test('composes a Redis-native claim with the adapter', async () => {
+    const kv = fakeStringKv()
+    let updateCalls = 0
+    const store = Store.from({
+      ...Store.redis({
+        get: kv.get,
+        set: kv.put,
+        del: kv.delete,
+        update: (key, fn) => {
+          updateCalls++
+          return kv.update(key, fn)
+        },
+      }),
+      tryClaim: fakeTryClaim(),
+    })
+    const expires = Date.now() + 60_000
+
+    expect(await Store.tryClaim(store, 'k', expires)).toBe(true)
+    expect(await Store.tryClaim(store, 'k', expires)).toBe(false)
+    expect(updateCalls).toBe(0)
+  })
+
+  test('composes an Upstash-native claim with the adapter', async () => {
+    const store = Store.from({
+      ...Store.upstash(fakeUnknownKv()),
+      tryClaim: fakeTryClaim(),
+    })
+    const expires = Date.now() + 60_000
+
+    expect(await Store.tryClaim(store, 'k', expires)).toBe(true)
+    expect(await Store.tryClaim(store, 'k', expires)).toBe(false)
+  })
+
+  test('prefixes a composed native claim', async () => {
+    const kv = fakeStringKv()
+    const seen: string[] = []
+    const store = Store.from(
+      {
+        ...Store.redis({
+          get: kv.get,
+          set: kv.put,
+          del: kv.delete,
+          update: kv.update,
+        }),
+        tryClaim: fakeTryClaim(seen),
+      },
+      { keyPrefix: 'p:' },
+    )
+
+    expect(await Store.tryClaim(store, 'k', Date.now() + 60_000)).toBe(true)
+    expect(seen).toEqual(['p:k'])
+  })
 })
 
 describe('json roundtrip behavior', () => {