Просмотр исходного кода

Fix: don't call t.Fatalf on shutdown after success

Rod Hynes 7 лет назад
Родитель
Сommit
5cdf590014
1 измененных файлов с 10 добавлено и 0 удалено
  1. 10 0
      psiphon/server/meek_test.go

+ 10 - 0
psiphon/server/meek_test.go

@@ -289,6 +289,11 @@ func TestMeekResiliency(t *testing.T) {
 	go func() {
 		defer serverWaitGroup.Done()
 		err := server.Run()
+		select {
+		case <-stopBroadcast:
+			return
+		default:
+		}
 		if err != nil {
 			t.Fatalf("MeekServer.Run failed: %s", err)
 		}
@@ -444,6 +449,11 @@ func TestMeekRateLimiter(t *testing.T) {
 	go func() {
 		defer serverWaitGroup.Done()
 		err := server.Run()
+		select {
+		case <-stopBroadcast:
+			return
+		default:
+		}
 		if err != nil {
 			t.Fatalf("MeekServer.Run failed: %s", err)
 		}