add enum of strings allowed type

Author: vsnig

package.json

@@ -1,6 +1,6 @@
 {
   "name": "zod-cdk-context",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "description": "Validate AWS CDK context with Zod",
   "keywords": [
     "zod",

src/index.ts

@@ -6,7 +6,11 @@ export { z } from 'zod'
 
 export const getValidatedContext = <
   T extends {
-    [key: string]: z.ZodString | z.ZodNumber | z.ZodOptional<z.ZodString | z.ZodNumber>
+    [key: string]:
+      | z.ZodString
+      | z.ZodNumber
+      | z.ZodOptional<z.ZodString | z.ZodNumber>
+      | z.ZodEnum<[string, ...string[]]>
   },
 >(
   node: Node,

test/index.test.ts

@@ -27,4 +27,13 @@ describe('validated context', () => {
     const app = new cdk.App({ context: {} })
     expect(() => getValidatedContext(app.node, { buzz: z.string() })).toThrow()
   })
+
+  test('enum of strings', () => {
+    const app = new cdk.App({ context: { stage: 'prd' } })
+
+    expect(getValidatedContext(app.node, { stage: z.enum(['prd', 'dev']) })).toEqual({
+      stage: 'prd',
+    })
+    expect(() => getValidatedContext(app.node, { stage: z.enum(['foo']) })).toThrow()
+  })
 })