test：添加网络与认证回归测试脚本

2026-03-10 16:17:55 +08:00
parent d96abbb8b9
commit 64850c2cae
2 changed files with 327 additions and 0 deletions
--- a/tests/unit/test_websocket_close_code.gd
+++ b/tests/unit/test_websocket_close_code.gd
@@ -0,0 +1,142 @@
+extends SceneTree
+
+# WebSocket 关闭码与重连行为测试（无需 GUT）
+#
+# 运行方式:
+#   "D:\\technology\\biancheng\\Godot\\Godot_v4.5.1-stable_win64.exe" --headless --path . --script tests/unit/test_websocket_close_code.gd
+
+var _failures: Array[String] = []
+
+func _init() -> void:
+	call_deferred("_run")
+
+func _run() -> void:
+	var manager := ChatWebSocketManager.new()
+	root.add_child(manager)
+	await process_frame
+
+	_test_close_code_mapping(manager)
+	_test_idle_closed_state_does_not_trigger_reconnect(manager)
+	_test_reconnect_on_unclean_close(manager)
+	_test_no_reconnect_on_clean_close(manager)
+	_test_reconnect_attempt_not_reset_on_reconnect_connect(manager)
+	_test_reconnection_success_emits_and_resets_attempt(manager)
+
+	manager.queue_free()
+	await process_frame
+
+	if _failures.is_empty():
+		print("PASS: test_websocket_close_code")
+		quit(0)
+		return
+
+	for failure in _failures:
+		push_error(failure)
+	print("FAIL: test_websocket_close_code (%d)" % _failures.size())
+	quit(1)
+
+func _test_close_code_mapping(manager: ChatWebSocketManager) -> void:
+	_assert(
+		not manager._is_clean_close_code(-1),
+		"close_code=-1 应判定为非干净关闭（异常断开）"
+	)
+	_assert(
+		manager._is_clean_close_code(1000),
+		"close_code=1000 应判定为干净关闭"
+	)
+	_assert(
+		manager._is_clean_close_code(0),
+		"close_code=0 不应判定为异常断开"
+	)
+
+func _test_idle_closed_state_does_not_trigger_reconnect(manager: ChatWebSocketManager) -> void:
+	manager.enable_auto_reconnect(true, 5, 3.0)
+	manager._set_connection_state(ChatWebSocketManager.ConnectionState.DISCONNECTED)
+	manager._reconnect_attempt = 0
+	manager._closed_state_handled = false
+
+	manager._check_websocket_state()
+
+	_assert(
+		manager.get_connection_state() == ChatWebSocketManager.ConnectionState.DISCONNECTED,
+		"空闲 DISCONNECTED + STATE_CLOSED 不应触发自动重连"
+	)
+	_assert(
+		manager._reconnect_attempt == 0,
+		"空闲关闭态不应消耗重连次数"
+	)
+
+func _test_reconnect_on_unclean_close(manager: ChatWebSocketManager) -> void:
+	var lost_count: int = 0
+	manager.connection_lost.connect(func(): lost_count += 1)
+
+	manager.enable_auto_reconnect(true, 5, 3.0)
+	manager._reconnect_attempt = 0
+	manager._set_connection_state(ChatWebSocketManager.ConnectionState.CONNECTED)
+	manager._on_websocket_closed(false)
+
+	_assert(
+		manager.get_connection_state() == ChatWebSocketManager.ConnectionState.RECONNECTING,
+		"异常断开时应进入 RECONNECTING 状态"
+	)
+	_assert(
+		manager._reconnect_attempt == 1,
+		"异常断开时应增加一次重连尝试计数"
+	)
+	_assert(
+		lost_count == 1,
+		"异常断开时应发射一次 connection_lost 信号"
+	)
+
+	# 清理重连定时器，避免影响后续用例
+	if is_instance_valid(manager._reconnect_timer):
+		manager._reconnect_timer.stop()
+
+func _test_no_reconnect_on_clean_close(manager: ChatWebSocketManager) -> void:
+	var reconnect_attempt_before: int = manager._reconnect_attempt
+	manager._set_connection_state(ChatWebSocketManager.ConnectionState.CONNECTED)
+	manager._on_websocket_closed(true)
+
+	_assert(
+		manager.get_connection_state() == ChatWebSocketManager.ConnectionState.DISCONNECTED,
+		"干净关闭时应回到 DISCONNECTED 状态"
+	)
+	_assert(
+		manager._reconnect_attempt == reconnect_attempt_before,
+		"干净关闭时不应增加重连尝试计数"
+	)
+
+func _test_reconnect_attempt_not_reset_on_reconnect_connect(manager: ChatWebSocketManager) -> void:
+	manager._reconnect_attempt = 3
+	manager._set_connection_state(ChatWebSocketManager.ConnectionState.RECONNECTING)
+	manager.connect_to_game_server(true)
+
+	_assert(
+		manager._reconnect_attempt == 3,
+		"重连触发 connect_to_game_server(true) 时不应重置重连尝试计数"
+	)
+
+func _test_reconnection_success_emits_and_resets_attempt(manager: ChatWebSocketManager) -> void:
+	var success_count: int = 0
+	manager.reconnection_succeeded.connect(func(): success_count += 1)
+
+	manager._reconnect_attempt = 2
+	manager._set_connection_state(ChatWebSocketManager.ConnectionState.RECONNECTING)
+	manager._on_websocket_connected()
+
+	_assert(
+		success_count == 1,
+		"重连成功时应发射 reconnection_succeeded 信号"
+	)
+	_assert(
+		manager._reconnect_attempt == 0,
+		"重连成功时应重置重连尝试计数"
+	)
+	_assert(
+		manager.get_connection_state() == ChatWebSocketManager.ConnectionState.CONNECTED,
+		"重连成功后应进入 CONNECTED 状态"
+	)
+
+func _assert(condition: bool, message: String) -> void:
+	if not condition:
+		_failures.append(message)