[VOL-3736]: Bug - Restarting multicast iperf server (at RG) will not restore the data stream at the RG.
- Use the right flow id when clearing mcast flow
- Fix UT
Change-Id: Ib392f0ec8fb62b7124774108fb507438ad00f14d
diff --git a/pkg/mocks/mockTechprofile.go b/pkg/mocks/mockTechprofile.go
index 1bdbbe4..a016b2c 100644
--- a/pkg/mocks/mockTechprofile.go
+++ b/pkg/mocks/mockTechprofile.go
@@ -43,9 +43,54 @@
// GetTPInstanceFromKVStore to mock techprofile GetTPInstanceFromKVStore method
func (m MockTechProfile) GetTPInstanceFromKVStore(ctx context.Context, techProfiletblID uint32, path string) (interface{}, error) {
- logger.Debug(ctx, "Warning Warning Warning: GetTPInstanceFromKVStore")
- return nil, nil
-
+ logger.Debug(ctx, "GetTPInstanceFromKVStore")
+ if techProfiletblID == 64 {
+ return &tp.TechProfile{
+ Name: "mock-tech-profile",
+ SubscriberIdentifier: "257",
+ ProfileType: "mock",
+ Version: 0,
+ NumGemPorts: 1,
+ UsScheduler: tp.IScheduler{
+ AllocID: 1,
+ Direction: "upstream",
+ AdditionalBw: "None",
+ Priority: 0,
+ Weight: 0,
+ QSchedPolicy: "",
+ },
+ DsScheduler: tp.IScheduler{
+ AllocID: 1,
+ Direction: "downstream",
+ AdditionalBw: "None",
+ Priority: 0,
+ Weight: 0,
+ QSchedPolicy: "",
+ },
+ UpstreamGemPortAttributeList: []tp.IGemPortAttribute{{
+ GemportID: 1,
+ PbitMap: "0b11111111",
+ },
+ },
+ DownstreamGemPortAttributeList: []tp.IGemPortAttribute{{
+ GemportID: 1,
+ PbitMap: "0b11111111",
+ },
+ },
+ }, nil
+ } else if techProfiletblID == 65 {
+ return &tp.EponProfile{
+ Name: "mock-epon-profile",
+ SubscriberIdentifier: "257",
+ ProfileType: "mock",
+ Version: 0,
+ NumGemPorts: 2,
+ UpstreamQueueAttributeList: nil,
+ DownstreamQueueAttributeList: nil,
+ }, nil
+ } else {
+ return nil, nil
+ }
}
// CreateTechProfInstance to mock techprofile CreateTechProfInstance method