[VOL-3311] Fix for ONU Delete and Re-Discovery
Change-Id: Ia5ae24a68889c9b160b55ad4af28667390cfa11c
diff --git a/internal/bbsim/devices/onu_flow_test.go b/internal/bbsim/devices/onu_flow_test.go
index b01859e..e1c801f 100644
--- a/internal/bbsim/devices/onu_flow_test.go
+++ b/internal/bbsim/devices/onu_flow_test.go
@@ -510,6 +510,15 @@
// checks that when the last flow is removed we reset the stored flags in the ONU
func Test_HandleFlowRemoveFlowId_LastFlow(t *testing.T) {
onu := createMockOnu(1, 1, 900, 900, true, false)
+
+ onu.InternalState = fsm.NewFSM(
+ "enabled",
+ fsm.Events{
+ {Name: "disable", Src: []string{"enabled"}, Dst: "disabled"},
+ },
+ fsm.Callbacks{},
+ )
+
onu.GemPortAdded = true
onu.DhcpFlowReceived = true
onu.EapolFlowReceived = true
@@ -530,4 +539,4 @@
assert.Equal(t, onu.GemPortAdded, false)
assert.Equal(t, onu.DhcpFlowReceived, false)
assert.Equal(t, onu.EapolFlowReceived, false)
-}
\ No newline at end of file
+}