From 148f7fa316f6fc3128496d12131c9d7dd4699af6 Mon Sep 17 00:00:00 2001
From: Joseph Garrone <joseph.garrone.gj@gmail.com>
Date: Thu, 6 Jun 2024 02:29:09 +0200
Subject: [PATCH] Rollback unarrowing of the getKcContextMock return type

---
 src/account/KcContext/getKcContextMock.ts |  4 +---
 src/login/KcContext/getKcContextMock.ts   |  4 +---
 test/login/KcContext.typelevel-spec.ts    | 10 +++++-----
 3 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/src/account/KcContext/getKcContextMock.ts b/src/account/KcContext/getKcContextMock.ts
index 3adf0688..0735bb81 100644
--- a/src/account/KcContext/getKcContextMock.ts
+++ b/src/account/KcContext/getKcContextMock.ts
@@ -47,9 +47,7 @@ export function createGetKcContextMock<
     >(params: {
         pageId: PageId;
         overrides?: DeepPartial<Extract<KcContext, { pageId: PageId }>>;
-        // NOTE: We choose to have a return type less precise than Extract<KcContext, { pageId: PageId }> {
-        // because we want to be able to use the mock just as the real KcContext.
-    }): KcContext {
+    }): Extract<KcContext, { pageId: PageId }> {
         const { pageId, overrides } = params;
 
         const kcContextMock = structuredCloneButFunctions(
diff --git a/src/login/KcContext/getKcContextMock.ts b/src/login/KcContext/getKcContextMock.ts
index b1e8322a..23b32d03 100644
--- a/src/login/KcContext/getKcContextMock.ts
+++ b/src/login/KcContext/getKcContextMock.ts
@@ -47,9 +47,7 @@ export function createGetKcContextMock<
     >(params: {
         pageId: PageId;
         overrides?: DeepPartial<Extract<KcContext, { pageId: PageId }>>;
-        // NOTE: We choose to have a return type less precise than Extract<KcContext, { pageId: PageId }> {
-        // because we want to be able to use the mock just as the real KcContext.
-    }): KcContext {
+    }): Extract<KcContext, { pageId: PageId }> {
         const { pageId, overrides } = params;
 
         const kcContextMock = structuredCloneButFunctions(
diff --git a/test/login/KcContext.typelevel-spec.ts b/test/login/KcContext.typelevel-spec.ts
index 9404085f..d418a127 100644
--- a/test/login/KcContext.typelevel-spec.ts
+++ b/test/login/KcContext.typelevel-spec.ts
@@ -64,7 +64,7 @@ import { Reflect } from "tsafe/Reflect";
             pageId: "login.ftl"
         });
 
-        type Expected = KcContext;
+        type Expected = Extract<KcContext, { pageId: "login.ftl" }>;
 
         assert<Equals<typeof got, Expected>>();
     }
@@ -74,7 +74,7 @@ import { Reflect } from "tsafe/Reflect";
             pageId: "register.ftl"
         });
 
-        type Expected = KcContext;
+        type Expected = Extract<KcContext, { pageId: "register.ftl" }>;
 
         assert<Equals<typeof got, Expected>>();
     }
@@ -84,7 +84,7 @@ import { Reflect } from "tsafe/Reflect";
             pageId: "my-custom-page.ftl"
         });
 
-        type Expected = KcContext;
+        type Expected = Extract<KcContext, { pageId: "my-custom-page.ftl" }>;
 
         assert<Equals<typeof got, Expected>>();
     }
@@ -181,7 +181,7 @@ import { Reflect } from "tsafe/Reflect";
             pageId: "login.ftl"
         });
 
-        type Expected = KcContext;
+        type Expected = Extract<KcContext, { pageId: "login.ftl" }>;
 
         assert<Equals<typeof got, Expected>>();
     }
@@ -191,7 +191,7 @@ import { Reflect } from "tsafe/Reflect";
             pageId: "register.ftl"
         });
 
-        type Expected = KcContext;
+        type Expected = Extract<KcContext, { pageId: "register.ftl" }>;
 
         assert<Equals<typeof got, Expected>>();
     }