tg-shuttle-score/test/signup.test.js

import { test } from 'node:test'
import assert from 'node:assert/strict'
import { createSignup, renderPoll, keyboard } from '../src/signup.js'
import { commandRegex, dateLabel, todayKey } from '../src/util.js'

// ---- 假 DB：用 SQL 開頭判斷要做什麼，維護 attendance / tg_members / tg_poll 三張表 ----
function fakeDb() {
	const attendance = [] // { poll_date, user_id, nickname }
	const members = new Map() // user_id -> { nickname, tg_name, username }
	const polls = new Map() // poll_date -> { chat_id, message_id }
	const sqlLog = []

	async function query(sql, params) {
		const s = sql.replace(/\s+/g, ' ').trim()
		sqlLog.push(s.slice(0, 40))
		if (s.startsWith('SELECT user_id, nickname FROM attendance')) {
			return attendance.filter((r) => r.poll_date === params[0]).map((r) => ({ user_id: r.user_id, nickname: r.nickname }))
		}
		if (s.startsWith('SELECT m.nickname, m.user_id AS member')) {
			const [uid, date] = params
			const m = members.get(uid)
			const a = attendance.find((r) => r.user_id === uid && r.poll_date === date)
			return [{ nickname: m?.nickname ?? null, member: m ? uid : null, joined: a ? uid : null }]
		}
		if (s.startsWith('INSERT INTO attendance')) {
			const [poll_date, user_id, nickname] = params
			attendance.push({ poll_date, user_id, nickname })
			return { affectedRows: 1 }
		}
		if (s.startsWith('DELETE FROM attendance')) {
			const [date, uid] = params
			const i = attendance.findIndex((r) => r.poll_date === date && r.user_id === uid)
			if (i >= 0) attendance.splice(i, 1)
			return { affectedRows: i >= 0 ? 1 : 0 }
		}
		if (s.startsWith('INSERT INTO tg_members (user_id, nickname')) {
			const [uid, nickname, tg_name, username] = params
			members.set(uid, { ...(members.get(uid) ?? {}), nickname, tg_name, username })
			return {}
		}
		if (s.startsWith('INSERT INTO tg_members (user_id, tg_name')) {
			const [uid, tg_name, username] = params
			members.set(uid, { ...(members.get(uid) ?? {}), tg_name, username })
			return {}
		}
		if (s.startsWith('UPDATE attendance SET nickname')) {
			const [nickname, uid, from] = params
			attendance.forEach((r) => {
				if (r.user_id === uid && r.poll_date >= from) r.nickname = nickname
			})
			return {}
		}
		if (s.startsWith('INSERT INTO tg_poll')) {
			const [date, chat_id, message_id] = params
			polls.set(date, { chat_id, message_id })
			return {}
		}
		if (s.startsWith('SELECT chat_id, message_id FROM tg_poll')) {
			const p = polls.get(params[0])
			return p ? [p] : []
		}
		throw new Error(`fakeDb 不認得的 SQL：${s}`)
	}
	return { query, attendance, members, polls, sqlLog }
}

function fakeBot() {
	const calls = []
	let nextId = 100
	return {
		calls,
		async sendMessage(chatId, text, opts) {
			calls.push({ fn: 'sendMessage', chatId, text, opts })
			return { message_id: nextId++, chat: { id: chatId } }
		},
		async editMessageText(text, opts) {
			calls.push({ fn: 'editMessageText', text, opts })
			return true
		},
		async answerCallbackQuery(id, opts) {
			calls.push({ fn: 'answerCallbackQuery', id, opts })
			return true
		},
	}
}

const silentLog = { info() {}, warn() {}, error() {} }
const DATE = '20260817'
const cq = (userId, choice, name = `U${userId}`) => ({
	id: `cq${userId}${choice}`,
	data: `bd|${choice}||${DATE}`,
	from: { id: userId, first_name: name, username: null },
	message: { chat: { id: -100 }, message_id: 4 },
})

test('renderPoll / keyboard 與 Node-RED 版格式一致', () => {
	assert.equal(dateLabel('20260817'), '08/17(一)')
	const text = renderPoll('20260817', [{ nickname: '建喵' }, { nickname: '<G>' }])
	assert.match(text, /^🏸 <b>今日羽球 08\/17\(一\)<\/b>\n按下方按鈕報名，可隨時反悔\n\n✅ 參加 \(2\)：建喵、&lt;G&gt;\n\n🕑 更新 \d\d:\d\d$/)
	assert.match(renderPoll('20260817', []), /✅ 參加 \(0\)：—/)
	assert.deepEqual(keyboard('20260817').inline_keyboard[0].map((b) => b.callback_data), ['bd|yes||20260817', 'bd|no||20260817'])
})

test('rsvp：參加 → 寫入、名單更新；重按 → 提示不動 DB；不參加 → 刪除', async () => {
	const db = fakeDb()
	const bot = fakeBot()
	const signup = createSignup(bot, { log: silentLog, query: db.query })

	await signup.rsvp(cq(1, 'yes', '建喵'))
	assert.deepEqual(db.attendance.map((r) => r.nickname), ['建喵'])
	assert.ok(db.members.has(1), '順手寫入 tg_members')
	let edit = bot.calls.find((c) => c.fn === 'editMessageText')
	assert.match(edit.text, /✅ 參加 \(1\)：建喵/)
	assert.equal(bot.calls.at(-1).opts.text, '已報名 ✅（建喵）')

	bot.calls.length = 0
	await signup.rsvp(cq(1, 'yes', '建喵'))
	assert.equal(db.attendance.length, 1)
	assert.equal(bot.calls.filter((c) => c.fn === 'editMessageText').length, 0, '重按不重畫')
	assert.equal(bot.calls.at(-1).opts.text, '你已經報名了 👌')

	bot.calls.length = 0
	await signup.rsvp(cq(2, 'no', '路人'))
	assert.equal(bot.calls.at(-1).opts.text, '歡迎加入羽球團 🏸', '沒報名的新面孔按不參加')
	await signup.rsvp(cq(2, 'no', '路人'))
	assert.equal(bot.calls.at(-1).opts.text, '已取消報名 ❌', '已是成員但沒報名')

	bot.calls.length = 0
	await signup.rsvp(cq(1, 'no', '建喵'))
	assert.equal(db.attendance.length, 0)
	edit = bot.calls.find((c) => c.fn === 'editMessageText')
	assert.match(edit.text, /✅ 參加 \(0\)：—/)
	assert.equal(bot.calls.at(-1).opts.text, '已取消報名 ❌')
})

test('rsvp 使用 tg_members 裡設定的暱稱', async () => {
	const db = fakeDb()
	const bot = fakeBot()
	const signup = createSignup(bot, { log: silentLog, query: db.query })
	db.members.set(9, { nickname: 'RURU' })
	await signup.rsvp(cq(9, 'yes', 'Ru Chen'))
	assert.equal(db.attendance[0].nickname, 'RURU')
	assert.equal(bot.calls.at(-1).opts.text, '已報名 ✅（RURU）')
})

test('post 會發訊息並記錄 tg_poll；setNickname 會更新今日名單與訊息', async () => {
	const db = fakeDb()
	const bot = fakeBot()
	const signup = createSignup(bot, { log: silentLog, query: db.query })
	const today = todayKey()

	await signup.post(-100)
	assert.deepEqual(db.polls.get(today), { chat_id: '-100', message_id: 100 })
	assert.deepEqual(bot.calls[0].opts.reply_markup, keyboard(today))

	await signup.rsvp({ ...cq(5, 'yes', 'Old'), data: `bd|yes||${today}` })
	bot.calls.length = 0
	await signup.setNickname(-100, 77, { id: 5, name: 'Old', username: null }, '新名')
	assert.equal(db.members.get(5).nickname, '新名')
	assert.equal(db.attendance[0].nickname, '新名')
	assert.match(bot.calls[0].text, /暱稱已設為「新名」/)
	const edit = bot.calls.find((c) => c.fn === 'editMessageText')
	assert.equal(edit.opts.message_id, 100)
	assert.match(edit.text, /新名/)
})

test('commandRegex：支援 @BotName 後綴與中英文別名', () => {
	const r = commandRegex('羽球|badminton', 'JianMiauBadmintonBot', '(?:\\s+(\\d{8}))?')
	assert.ok(r.test('/羽球'))
	assert.ok(r.test('/badminton@JianMiauBadmintonBot 20260817'))
	assert.equal('/羽球 20260817'.match(r)[1], '20260817')
	assert.ok(!r.test('/羽球名單'), '不能誤吃 /羽球名單')
	assert.ok(!r.test('/羽球x'))
	const s = commandRegex('記分|score|new', 'JianMiauBadmintonBot', '(?:\\s+([\\s\\S]+))?')
	assert.equal('/score 小明 vs 阿強'.match(s)[1], '小明 vs 阿強')
	assert.equal('/記分'.match(s)[1], undefined)
})