From 639c9875b2a573ab74c80d4d331b90886b0770de Mon Sep 17 00:00:00 2001 From: Conner Fromknecht Date: Tue, 19 Jun 2018 13:36:12 +0100 Subject: [PATCH] channeldb/channel_test: test packager source updated --- channeldb/channel_test.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/channeldb/channel_test.go b/channeldb/channel_test.go index f7f331af..efd36abb 100644 --- a/channeldb/channel_test.go +++ b/channeldb/channel_test.go @@ -898,6 +898,16 @@ func TestRefreshShortChanID(t *testing.T) { "updated before refreshing short_chan_id") } + // Now that the receiver's short channel id has been updated, check to + // ensure that the channel packager's source has been updated as well. + // This ensures that the packager will read and write to buckets + // corresponding to the new short chan id, instead of the prior. + if state.Packager.(*ChannelPackager).source != chanOpenLoc { + t.Fatalf("channel packager source was not updated: want %v, "+ + "got %v", chanOpenLoc, + state.Packager.(*ChannelPackager).source) + } + // Now, refresh the short channel ID of the pending channel. err = pendingChannel.RefreshShortChanID() if err != nil { @@ -911,4 +921,14 @@ func TestRefreshShortChanID(t *testing.T) { "refreshed: want %v, got %v", state.ShortChanID(), pendingChannel.ShortChanID()) } + + // Check to ensure that the _other_ OpenChannel channel packager's + // source has also been updated after the refresh. This ensures that the + // other packagers will read and write to buckets corresponding to the + // updated short chan id. + if pendingChannel.Packager.(*ChannelPackager).source != chanOpenLoc { + t.Fatalf("channel packager source was not updated: want %v, "+ + "got %v", chanOpenLoc, + pendingChannel.Packager.(*ChannelPackager).source) + } }