Skip to content

Commit a72d04c

Browse files
committed
Misc fixes
1 parent ce3e474 commit a72d04c

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

enterprise/replicasync/replicasync_test.go

+11-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,10 @@ func TestReplica(t *testing.T) {
3434
db, pubsub := dbtestutil.NewDB(t)
3535
closeChan := make(chan struct{}, 1)
3636
cancel, err := pubsub.Subscribe(replicasync.PubsubEvent, func(ctx context.Context, message []byte) {
37-
closeChan <- struct{}{}
37+
select {
38+
case closeChan <- struct{}{}:
39+
default:
40+
}
3841
})
3942
require.NoError(t, err)
4043
defer cancel()
@@ -70,6 +73,8 @@ func TestReplica(t *testing.T) {
7073
RelayAddress: "http://169.254.169.254",
7174
})
7275
require.NoError(t, err)
76+
defer server.Close()
77+
7378
require.Len(t, server.Regional(), 1)
7479
require.Equal(t, peer.ID, server.Regional()[0].ID)
7580
require.Empty(t, server.Self().Error)
@@ -113,6 +118,8 @@ func TestReplica(t *testing.T) {
113118
TLSConfig: tlsConfig,
114119
})
115120
require.NoError(t, err)
121+
defer server.Close()
122+
116123
require.Len(t, server.Regional(), 1)
117124
require.Equal(t, peer.ID, server.Regional()[0].ID)
118125
require.Empty(t, server.Self().Error)
@@ -138,6 +145,8 @@ func TestReplica(t *testing.T) {
138145
RelayAddress: "http://127.0.0.1:1",
139146
})
140147
require.NoError(t, err)
148+
defer server.Close()
149+
141150
require.Len(t, server.Regional(), 1)
142151
require.Equal(t, peer.ID, server.Regional()[0].ID)
143152
require.NotEmpty(t, server.Self().Error)
@@ -152,6 +161,7 @@ func TestReplica(t *testing.T) {
152161
defer cancelCtx()
153162
server, err := replicasync.New(ctx, slogtest.Make(t, nil), db, pubsub, nil)
154163
require.NoError(t, err)
164+
defer server.Close()
155165
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
156166
w.WriteHeader(http.StatusOK)
157167
}))

0 commit comments

Comments
 (0)