From 910dfc2bb45e6a9c4fdee383374e9e8dacd0efd6 Mon Sep 17 00:00:00 2001
From: Yichuan Shen <shen.yichuan@gmail.com>
Date: Wed, 22 Mar 2023 03:16:51 +0100
Subject: [PATCH] Test generics

---
 test/enum.test.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/test/enum.test.ts b/test/enum.test.ts
index af64d45..9bce8a3 100644
--- a/test/enum.test.ts
+++ b/test/enum.test.ts
@@ -8,14 +8,17 @@ import {
 } from "../dev_deps.ts";
 import { TypeExtends } from "./utils.ts";
 
-class MessageVariants {
+class MessageVariants<T> {
   Quit = null;
-  Plaintext = Variant<string>();
+  Plaintext = Variant<T>();
   Encrypted = Variant<number[]>();
 }
 
-type Message = Enum<MessageVariants>;
-const Message = () => Enum.factory(MessageVariants);
+type Message = Enum<MessageVariants<string>>;
+const Message = () => Enum.factory(MessageVariants<string>);
+
+type GenericMessage<T> = Enum<MessageVariants<T>>;
+const GenericMessage = <T>() => Enum.factory(MessageVariants<T>);
 
 Deno.test({
   name: "Enums can contain one and only one variant",
@@ -49,13 +52,16 @@ Deno.test({
 
     msg = Message().Encrypted([1, 2, 3]);
     assertEquals(msg, { Encrypted: [1, 2, 3] });
+
+    let genMsg = GenericMessage<boolean>().Plaintext(true);
+    assertEquals(genMsg, { Plaintext: true });
   },
 });
 
 Deno.test({
   name: "Enum.match() has to be exhaustive",
   fn() {
-    type M = Matcher<MessageVariants>;
+    type M = Matcher<MessageVariants<string>>;
 
     expectType<M>({
       Quit: () => -1,