Signal End-Of-RIB when registering client on LocRIB (#368)

bio-routing · Aug 30, 2022 · 162a33f · 162a33f
1 parent 3cfb73a
commit 162a33f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/protocols/bgp/server/fsm_address_family_test.go b/protocols/bgp/server/fsm_address_family_test.go
@@ -1,6 +1,7 @@
 package server
 
 import (
+	"bytes"
 	"testing"
 
 	"github.com/bio-routing/bio-rd/protocols/bgp/packet"
@@ -10,6 +11,8 @@ import (
 	"github.com/bio-routing/bio-rd/routingtable/filter"
 	"github.com/bio-routing/bio-rd/routingtable/locRIB"
 	"github.com/stretchr/testify/assert"
+
+	biotesting "github.com/bio-routing/bio-rd/testing"
 )
 
 func TestFSMAFIInitDispose(t *testing.T) {
@@ -25,6 +28,9 @@ func TestFSMAFIInitDispose(t *testing.T) {
 				localASN:        15169,
 				adjRIBInFactory: adjRIBInFactory{},
 			},
+			con: &biotesting.MockConn{
+				Buf: bytes.NewBuffer(nil),
+			},
 		},
 		addPathTX: routingtable.ClientOptions{
 			BestOnly: true,

diff --git a/routingtable/locRIB/loc_rib.go b/routingtable/locRIB/loc_rib.go
@@ -116,6 +116,7 @@ func (a *LocRIB) UpdateNewClient(client routingtable.RouteTableClient) error {
 		}
 	}
 
+	client.EndOfRIB()
 	return nil
 }
-Original file line number
+Diff line change
@@ Expand Up @@
     		}
     	}
+    	client.EndOfRIB()
     	return nil
     }
@@ Expand Down @@