test: add type tests for stores with actions

vuejs · Apr 1, 2024 · 50aa53b · 50aa53b
1 parent 1d54589
commit 50aa53b
Showing 1 changed file with 57 additions and 0 deletions.
diff --git a/packages/pinia/test-dts/customizations.test-d.ts b/packages/pinia/test-dts/customizations.test-d.ts
@@ -215,6 +215,33 @@ expectType<{
   )
 )
 
+expectType<{
+  n: Ref<number>
+  customN: Ref<number> & { plusOne: () => void }
+  double: ComputedRef<number>
+  myState: Ref<number>
+  stateOnly: Ref<number>
+}>(
+  storeToRefs(
+    defineStore('a', () => {
+      const n = ref(1)
+      const customN = ref(1) as Ref<number> & { plusOne: () => void }
+      const double = computed(() => n.value * 2)
+
+      function plusOne() {
+        customN.value++
+      }
+
+      return {
+        n,
+        customN,
+        double,
+        plusOne,
+      }
+    })()
+  )
+)
+
 expectType<{
   n: Ref<number>
   customN: Ref<number> & { plusOne: () => void }
@@ -231,6 +258,36 @@ expectType<{
       getters: {
         double: (state) => state.n * 2,
       },
+      actions: {
+        plusOne() {
+          this.n++
+        },
+      },
+    })()
+  )
+)
+
+expectType<{
+  n: Ref<number>
+  customN: Ref<number> & { plusOne: () => void }
+  double: ComputedRef<number>
+  myState: Ref<number>
+  stateOnly: Ref<number>
+}>(
+  storeToRefs(
+    defineStore('a', {
+      state: () => ({
+        n: 1,
+        customN: ref(1) as Ref<number> & { plusOne: () => void },
+      }),
+      getters: {
+        double: (state) => state.n * 2,
+      },
+      actions: {
+        plusOne() {
+          this.n++
+        },
+      },
     })()
   )
 )