From ddffdb389a1053897680da8325642d730d4f33ac Mon Sep 17 00:00:00 2001 From: sjlleo Date: Mon, 20 Jun 2022 22:25:58 +0800 Subject: [PATCH] fix: fast trace test --- fast_trace/fast_trace_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/fast_trace/fast_trace_test.go b/fast_trace/fast_trace_test.go index b8763ae..285c3b8 100644 --- a/fast_trace/fast_trace_test.go +++ b/fast_trace/fast_trace_test.go @@ -1,14 +1,24 @@ package fastTrace import ( + "os" + "os/signal" "testing" "github.com/xgadget-lab/nexttrace/trace" + "github.com/xgadget-lab/nexttrace/wshandle" ) // ICMP Use Too Many Time to Wait So we don't test it. func TestTCPTrace(t *testing.T) { ft := FastTracer{} + // 建立 WebSocket 连接 + w := wshandle.New() + w.Interrupt = make(chan os.Signal, 1) + signal.Notify(w.Interrupt, os.Interrupt) + defer func() { + w.Conn.Close() + }() ft.TracerouteMethod = trace.TCPTrace ft.testCM() ft.testEDU()