From 7a2958f356ab3cbb920291f52e1dd7d8f64e82c8 Mon Sep 17 00:00:00 2001
From: dakkar <dakkar@thenautilus.net>
Date: Tue, 13 May 2025 14:36:01 +0100
Subject: [PATCH] fix `test/e2e/antennas.ts`

---
 packages/backend/test/e2e/antennas.ts  |  8 +++++---
 packages/backend/test/e2e/timelines.ts | 17 +----------------
 packages/backend/test/utils.ts         | 16 ++++++++++++++++
 3 files changed, 22 insertions(+), 19 deletions(-)

diff --git a/packages/backend/test/e2e/antennas.ts b/packages/backend/test/e2e/antennas.ts
index 1bbacd065b..b3b91ebcea 100644
--- a/packages/backend/test/e2e/antennas.ts
+++ b/packages/backend/test/e2e/antennas.ts
@@ -16,6 +16,7 @@ import {
 	testPaginationConsistency,
 	uploadFile,
 	userList,
+  withNotesCount,
 } from '../utils.js';
 import type * as misskey from 'misskey-js';
 import { DEFAULT_POLICIES } from '@/core/RoleService.js';
@@ -114,6 +115,7 @@ describe('アンテナ', () => {
 		userBlockedByAlice = await signup({ username: 'userBlockedByAlice' });
 		await post(userBlockedByAlice, { text: 'test' });
 		await api('blocking/create', { userId: userBlockedByAlice.id }, alice);
+		await api('mute/delete', { userId: userBlockedByAlice.id }, alice); // blocking implies muting, in Sharkey, but we want to test un-muted block
 		userMutingAlice = await signup({ username: 'userMutingAlice' });
 		await post(userMutingAlice, { text: 'test' });
 		await api('mute/create', { userId: alice.id }, userMutingAlice);
@@ -347,7 +349,7 @@ describe('アンテナ', () => {
 				parameters: { antennaId: antenna.id },
 				user: alice,
 			});
-			const expected = [note];
+			const expected = withNotesCount([note], 2);
 			assert.deepStrictEqual(response, expected);
 		});
 
@@ -666,10 +668,10 @@ describe('アンテナ', () => {
 				user: alice,
 			});
 			// 最後に投稿したものが先頭に来る。
-			const expected = [
+			const expected = withNotesCount([
 				noteInNonSensitiveChannel,
 				noteInLocal,
-			];
+			], 64);
 			assert.deepStrictEqual(response, expected);
 		});
 
diff --git a/packages/backend/test/e2e/timelines.ts b/packages/backend/test/e2e/timelines.ts
index 9685776a71..feb9825bfc 100644
--- a/packages/backend/test/e2e/timelines.ts
+++ b/packages/backend/test/e2e/timelines.ts
@@ -9,7 +9,7 @@
 import * as assert from 'assert';
 import { setTimeout } from 'node:timers/promises';
 import { Redis } from 'ioredis';
-import { api, post, randomString, sendEnvUpdateRequest, signup, uploadUrl } from '../utils.js';
+import { api, post, randomString, sendEnvUpdateRequest, signup, uploadUrl, withNotesCount } from '../utils.js';
 import { loadConfig } from '@/config.js';
 
 function genHost() {
@@ -20,21 +20,6 @@ function waitForPushToTl() {
 	return setTimeout(500);
 }
 
-// the packed user inside each note returned by `users/notes` has the
-// latest `notesCount`, not the count at the time the note was
-// created, so we override it
-function withNotesCount(notes, count) {
-	return notes.map( note => {
-		return {
-			...note,
-			user: {
-				...note.user,
-				notesCount: count,
-			},
-		};
-	});
-}
-
 let redisForTimelines: Redis;
 
 describe('Timelines', () => {
diff --git a/packages/backend/test/utils.ts b/packages/backend/test/utils.ts
index 7f2768488f..90a00e2096 100644
--- a/packages/backend/test/utils.ts
+++ b/packages/backend/test/utils.ts
@@ -691,3 +691,19 @@ export async function captureWebhook<T = SystemWebhookPayload>(postAction: () =>
 
 	return JSON.parse(result) as T;
 }
+
+
+// the packed user inside each note returned by `users/notes` has the
+// latest `notesCount`, not the count at the time the note was
+// created, so we override it
+export function withNotesCount(notes, count) {
+	return notes.map( note => {
+		return {
+			...note,
+			user: {
+				...note.user,
+				notesCount: count,
+			},
+		};
+	});
+}