From 3e80f0b384ea2dc14c1b60b5897e90040cab9a24 Mon Sep 17 00:00:00 2001
From: DD <didinele.dev@gmail.com>
Date: Fri, 5 May 2023 23:01:28 +0300
Subject: [PATCH] tests(ws): fix tests (#9520)

---
 .../strategy/WorkerShardingStrategy.test.ts        |  6 +++---
 packages/ws/__tests__/ws/WebSocketManager.test.ts  | 14 +++++++++-----
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/packages/ws/__tests__/strategy/WorkerShardingStrategy.test.ts b/packages/ws/__tests__/strategy/WorkerShardingStrategy.test.ts
index b62fceb59..3277dcb3d 100644
--- a/packages/ws/__tests__/strategy/WorkerShardingStrategy.test.ts
+++ b/packages/ws/__tests__/strategy/WorkerShardingStrategy.test.ts
@@ -146,9 +146,9 @@ beforeEach(() => {
 });
 
 afterEach(() => {
-	mockConstructor.mockRestore();
-	mockSend.mockRestore();
-	mockTerminate.mockRestore();
+	mockConstructor.mockClear();
+	mockSend.mockClear();
+	mockTerminate.mockClear();
 });
 
 test('spawn, connect, send a message, session info, and destroy', async () => {
diff --git a/packages/ws/__tests__/ws/WebSocketManager.test.ts b/packages/ws/__tests__/ws/WebSocketManager.test.ts
index d34a4dd84..2f4e33fa6 100644
--- a/packages/ws/__tests__/ws/WebSocketManager.test.ts
+++ b/packages/ws/__tests__/ws/WebSocketManager.test.ts
@@ -54,13 +54,13 @@ test('fetch gateway information', async () => {
 	expect(initial).toEqual(data);
 	expect(fetch).toHaveBeenCalledOnce();
 
-	fetch.mockRestore();
+	fetch.mockClear();
 
 	const cached = await manager.fetchGatewayInformation();
 	expect(cached).toEqual(data);
 	expect(fetch).not.toHaveBeenCalled();
 
-	fetch.mockRestore();
+	fetch.mockClear();
 	mockPool
 		.intercept({
 			path: '/api/v10/gateway/bot',
@@ -72,7 +72,7 @@ test('fetch gateway information', async () => {
 	expect(forced).toEqual(data);
 	expect(fetch).toHaveBeenCalledOnce();
 
-	fetch.mockRestore();
+	fetch.mockClear();
 	mockPool
 		.intercept({
 			path: '/api/v10/gateway/bot',
@@ -146,7 +146,7 @@ test('update shard count', async () => {
 	expect(await manager.getShardCount()).toBe(2);
 	expect(fetch).not.toHaveBeenCalled();
 
-	fetch.mockRestore();
+	fetch.mockClear();
 	mockPool
 		.intercept({
 			path: '/api/v10/gateway/bot',
@@ -230,7 +230,11 @@ test('strategies', async () => {
 	expect(strategy.destroy).toHaveBeenCalledWith(destroyOptions);
 
 	// eslint-disable-next-line id-length
-	const send: GatewaySendPayload = { op: GatewayOpcodes.RequestGuildMembers, d: { guild_id: '1234', limit: 0 } };
+	const send: GatewaySendPayload = {
+		op: GatewayOpcodes.RequestGuildMembers,
+		// eslint-disable-next-line id-length
+		d: { guild_id: '1234', limit: 0, query: '' },
+	};
 	await manager.send(0, send);
 	expect(strategy.send).toHaveBeenCalledWith(0, send);
 });