highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-09-13 18:10:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

graph/db: update nextUpdateTime in tests

Browse Source 
Let it return time.Time instead. Also, make use of the helper in a few
more places.
This commit is contained in:
Elle Mouton
2025-06-30 12:22:13 +02:00
parent c7aa17971b
commit 64fc92abe7
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
graph/db/graph_test.go
View File

@@ -77,7 +77,7 @@ func createLightningNode(priv *btcec.PrivateKey) *models.LightningNode {
n := &models.LightningNode{ n := &models.LightningNode{
HaveNodeAnnouncement: true, HaveNodeAnnouncement: true,
AuthSigBytes: testSig.Serialize(), AuthSigBytes: testSig.Serialize(),
LastUpdate: time.Unix(nextUpdateTime(), 0), LastUpdate: nextUpdateTime(),
Color: color.RGBA{1, 2, 3, 0}, Color: color.RGBA{1, 2, 3, 0},
Alias: "kek" + hex.EncodeToString(pub), Alias: "kek" + hex.EncodeToString(pub),
Features: testFeatures, Features: testFeatures,
@@ -784,7 +784,7 @@ func createChannelEdge(node1, node2 *models.LightningNode,
edge1 := &models.ChannelEdgePolicy{ edge1 := &models.ChannelEdgePolicy{
SigBytes: testSig.Serialize(), SigBytes: testSig.Serialize(),
ChannelID: chanID, ChannelID: chanID,
LastUpdate: time.Unix(433453, 0), LastUpdate: nextUpdateTime(),
MessageFlags: 1, MessageFlags: 1,
ChannelFlags: 0, ChannelFlags: 0,
TimeLockDelta: 99, TimeLockDelta: 99,
@@ -798,7 +798,7 @@ func createChannelEdge(node1, node2 *models.LightningNode,
edge2 := &models.ChannelEdgePolicy{ edge2 := &models.ChannelEdgePolicy{
SigBytes: testSig.Serialize(), SigBytes: testSig.Serialize(),
ChannelID: chanID, ChannelID: chanID,
LastUpdate: time.Unix(124234, 0), LastUpdate: nextUpdateTime(),
MessageFlags: 1, MessageFlags: 1,
ChannelFlags: 1, ChannelFlags: 1,
TimeLockDelta: 99, TimeLockDelta: 99,
@@ -3455,13 +3455,13 @@ var (
updateTimeMu sync.Mutex updateTimeMu sync.Mutex
) )
func nextUpdateTime() int64 { func nextUpdateTime() time.Time {
updateTimeMu.Lock() updateTimeMu.Lock()
defer updateTimeMu.Unlock() defer updateTimeMu.Unlock()
updateTime++ updateTime++
return updateTime return time.Unix(updateTime, 0)
} }
// TestNodeIsPublic ensures that we properly detect nodes that are seen as // TestNodeIsPublic ensures that we properly detect nodes that are seen as
@@ -3506,7 +3506,7 @@ func TestNodeIsPublic(t *testing.T) {
graphs := []*ChannelGraph{aliceGraph, bobGraph, carolGraph} graphs := []*ChannelGraph{aliceGraph, bobGraph, carolGraph}
for _, graph := range graphs { for _, graph := range graphs {
for _, node := range nodes { for _, node := range nodes {
node.LastUpdate = time.Unix(nextUpdateTime(), 0) node.LastUpdate = nextUpdateTime()
err := graph.AddLightningNode(ctx, node) err := graph.AddLightningNode(ctx, node)
require.NoError(t, err) require.NoError(t, err)
} }
@@ -3621,7 +3621,7 @@ func TestDisabledChannelIDs(t *testing.T) {
// Adding a new channel edge to the graph. // Adding a new channel edge to the graph.
edgeInfo, edge1, edge2 := createChannelEdge(node1, node2) edgeInfo, edge1, edge2 := createChannelEdge(node1, node2)
node2.LastUpdate = time.Unix(nextUpdateTime(), 0) node2.LastUpdate = nextUpdateTime()
if err := graph.AddLightningNode(ctx, node2); err != nil { if err := graph.AddLightningNode(ctx, node2); err != nil {
t.Fatalf("unable to add node: %v", err) t.Fatalf("unable to add node: %v", err)
} }