From 839b6271e52f0a88752d3c907cc8f39f1011ac2b Mon Sep 17 00:00:00 2001 From: yyforyongyu Date: Fri, 13 Oct 2023 00:00:46 +0800 Subject: [PATCH] peer: fix unit test flake --- peer/brontide_test.go | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) diff --git a/peer/brontide_test.go b/peer/brontide_test.go index 428ea934e..10ef1bded 100644 --- a/peer/brontide_test.go +++ b/peer/brontide_test.go @@ -1267,18 +1267,6 @@ func TestHandleRemovePendingChannel(t *testing.T) { chanIDNotExist := lnwire.ChannelID{1} chanIDPending := lnwire.ChannelID{2} - // Create a test brontide. - dummyConfig := Config{} - peer := NewBrontide(dummyConfig) - - // Create the test state. - peer.activeChannels.Store(chanIDActive, &lnwallet.LightningChannel{}) - peer.activeChannels.Store(chanIDPending, nil) - - // Assert test state, we should have two channels store, one active and - // one pending. - require.Equal(t, 2, peer.activeChannels.Len()) - testCases := []struct { name string chanID lnwire.ChannelID @@ -1314,8 +1302,23 @@ func TestHandleRemovePendingChannel(t *testing.T) { err: errChan, } + // Create a test brontide. + dummyConfig := Config{} + peer := NewBrontide(dummyConfig) + + // Create the test state. + peer.activeChannels.Store( + chanIDActive, &lnwallet.LightningChannel{}, + ) + peer.activeChannels.Store(chanIDPending, nil) + + // Assert test state, we should have two channels store, one + // active and one pending. + require.Equal(t, 2, peer.activeChannels.Len()) + t.Run(tc.name, func(t *testing.T) { t.Parallel() + require := require.New(t) // Get the number of channels before mutating the