[SEBA-873] add reboot olt support
Change-Id: I1570d05313661a6d66e1596b9f9a1a1cc17d1a73
diff --git a/internal/bbsim/devices/nni_test.go b/internal/bbsim/devices/nni_test.go
index 60856c6..348d40f 100644
--- a/internal/bbsim/devices/nni_test.go
+++ b/internal/bbsim/devices/nni_test.go
@@ -19,9 +19,10 @@
import (
"errors"
+ "testing"
+
"github.com/opencord/bbsim/internal/bbsim/types"
"gotest.tools/assert"
- "testing"
)
func TestSetVethUpSuccess(t *testing.T) {
@@ -49,7 +50,7 @@
startDHCPServerCalled := false
_startDHCPServer := startDHCPServer
defer func() { startDHCPServer = _startDHCPServer }()
- startDHCPServer = func() error {
+ startDHCPServer = func(upstreamVeth string, dhcpServerIp string) error {
startDHCPServerCalled = true
return nil
}
@@ -67,8 +68,10 @@
}
olt := OltDevice{}
+ nni := NniPort{}
- err := createNNIPair(spy, &olt)
+ err := createNNIPair(spy, &olt, &nni)
+ olt.nniPktInChannel, _ = nni.NewVethChan()
assert.Equal(t, spy.CommandCallCount, 3)
assert.Equal(t, startDHCPServerCalled, true)