Fix replica resend

kylecarbs · kylecarbs · commit c3a77fe2d048 · 2022-10-16T16:21:13.000Z
diff --git a/enterprise/derpmesh/derpmesh_test.go b/enterprise/derpmesh/derpmesh_test.go
@@ -94,11 +94,28 @@ func TestDERPMesh(t *testing.T) {
 		secondClient.TLSConfig = tlsConfig
 		err = secondClient.Connect(context.Background())
 		require.NoError(t, err)
+
+		closed := make(chan struct{})
+		ctx, cancelFunc := context.WithCancel(context.Background())
+		defer cancelFunc()
 		sent := []byte("hello world")
-		err = firstClient.Send(second.Public(), sent)
-		require.NoError(t, err)
+		go func() {
+			defer close(closed)
+			ticker := time.NewTicker(50 * time.Millisecond)
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				case <-ticker.C:
+				}
+				err = firstClient.Send(second.Public(), sent)
+				require.NoError(t, err)
+			}
+		}()
 		got := recvData(t, secondClient)
 		require.Equal(t, sent, got)
+		cancelFunc()
+		<-closed
 	})
 	t.Run("TwentyMeshes", func(t *testing.T) {
 		t.Parallel()
@@ -135,7 +152,7 @@ func TestDERPMesh(t *testing.T) {
 		sent := []byte("hello world")
 		go func() {
 			defer close(closed)
-			ticker := time.NewTicker(time.Second)
+			ticker := time.NewTicker(50 * time.Millisecond)
 			for {
 				select {
 				case <-ctx.Done():
