From 599f23c79458b8cf08a6e63b17cd18691dec6818 Mon Sep 17 00:00:00 2001
From: mshanemc <shane.mclaughlin@salesforce.com>
Date: Thu, 2 Nov 2023 11:52:25 -0500
Subject: [PATCH] test: mock throws when no user

---
 src/testSetup.ts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/testSetup.ts b/src/testSetup.ts
index aa8d0c4007..a7beed37a2 100644
--- a/src/testSetup.ts
+++ b/src/testSetup.ts
@@ -340,10 +340,11 @@ export class TestContext {
       })
     );
 
-    this.SANDBOX.stub(User.prototype, 'retrieve').callsFake(
-      // @ts-expect-error the real method guarantees a user, but find might not return one
-      (username): Promise<UserFields | undefined> => Promise.resolve(mockUsers.find((org) => org.username === username))
-    );
+    this.SANDBOX.stub(User.prototype, 'retrieve').callsFake((username): Promise<UserFields> => {
+      const user = mockUsers.find((org) => org.username === username);
+      if (!user) throw new SfError('User not found', 'UserNotFoundError');
+      return Promise.resolve(user);
+    });
 
     const retrieveContents = async function (this: { path: string }): Promise<{ usernames?: string[] }> {
       const orgId = basename(this.path.replace('.json', ''));