74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.asio.asia/nixevol/NixMsg/internal/config"
|
|
"github.com/mochi-mqtt/server/v2/packets"
|
|
)
|
|
|
|
func TestServeShutdownWithLiveMQTT(t *testing.T) {
|
|
dataDir := t.TempDir()
|
|
cfgPath := writeTestConfig(t, dataDir)
|
|
initAdminForTest(t, dataDir)
|
|
enableRegistration(t, dataDir, "uplink-code")
|
|
|
|
cfg, err := config.Load(cfgPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if vErr := cfg.Validate(); vErr != nil {
|
|
t.Fatal(vErr)
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
errCh := make(chan error, 1)
|
|
go func() { errCh <- runServe(ctx, cfg) }()
|
|
|
|
addr := waitListenAddr(t, dataDir, 15*time.Second)
|
|
base := "http://" + addr
|
|
registerEP(t, base, "alice", "password12", "Alice")
|
|
registerEP(t, base, "bob", "password12", "Bob")
|
|
|
|
ws := mqttSessionLogin(t, base, "alice", "password12")
|
|
tcp := mqttSessionLoginTCP(t, addr, "bob", "password12")
|
|
|
|
cancel()
|
|
|
|
var gotWG sync.WaitGroup
|
|
var wsReason, tcpReason byte
|
|
var wsOK, tcpOK bool
|
|
gotWG.Add(2)
|
|
go func() {
|
|
defer gotWG.Done()
|
|
wsReason, wsOK = ws.disconnectReason(12 * time.Second)
|
|
}()
|
|
go func() {
|
|
defer gotWG.Done()
|
|
tcpReason, tcpOK = tcp.disconnectReason(12 * time.Second)
|
|
}()
|
|
|
|
select {
|
|
case err := <-errCh:
|
|
if err != nil {
|
|
t.Fatalf("runServe: %v", err)
|
|
}
|
|
case <-time.After(15 * time.Second):
|
|
t.Fatal("runServe did not return within 15s")
|
|
}
|
|
gotWG.Wait()
|
|
|
|
want := packets.ErrServerShuttingDown.Code
|
|
if !wsOK || wsReason != want {
|
|
t.Fatalf("ws disconnect ok=%v reason=%#x want %#x", wsOK, wsReason, want)
|
|
}
|
|
if !tcpOK || tcpReason != want {
|
|
t.Fatalf("tcp disconnect ok=%v reason=%#x want %#x", tcpOK, tcpReason, want)
|
|
}
|
|
ws.Close()
|
|
tcp.Close()
|
|
}
|