diff --git a/sessions/sessions_test.go b/sessions/sessions_test.go index 14721a172..719b77786 100644 --- a/sessions/sessions_test.go +++ b/sessions/sessions_test.go @@ -1164,6 +1164,9 @@ func TestSessionSV1AuthorizeReplyAsNavigableMap(t *testing.T) { expected.Set([]string{utils.CapMaxUsage}, 5*time.Minute, false, false) v1AuthRpl.SetMaxUsageNeeded(true) + if !v1AuthRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1AuthRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1184,6 +1187,9 @@ func TestSessionSV1AuthorizeReplyAsNavigableMap(t *testing.T) { utils.CapStatQueues: *statIDs, }) v1AuthRpl.SetMaxUsageNeeded(true) + if !v1AuthRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1AuthRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1208,6 +1214,9 @@ func TestSessionSV1InitSessionReplyAsNavigableMap(t *testing.T) { expected.Set([]string{utils.CapMaxUsage}, 5*time.Minute, false, false) v1InitRpl.SetMaxUsageNeeded(true) + if !v1InitRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1InitRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1226,6 +1235,9 @@ func TestSessionSV1InitSessionReplyAsNavigableMap(t *testing.T) { utils.CapStatQueues: *statIDs, }) v1InitRpl.SetMaxUsageNeeded(true) + if !v1InitRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1InitRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1248,6 +1260,9 @@ func TestSessionSV1UpdateSessionReplyAsNavigableMap(t *testing.T) { expected.Set([]string{utils.CapMaxUsage}, 5*time.Minute, false, false) v1UpdtRpl.SetMaxUsageNeeded(true) + if !v1UpdtRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1UpdtRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1269,6 +1284,9 @@ func TestSessionSV1ProcessMessageReplyAsNavigableMap(t *testing.T) { v1PrcEvRpl.MaxUsage = 5 * time.Minute expected.Set([]string{utils.CapMaxUsage}, 5*time.Minute, false, false) v1PrcEvRpl.SetMaxUsageNeeded(true) + if !v1PrcEvRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1PrcEvRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1294,6 +1312,9 @@ func TestSessionSV1ProcessMessageReplyAsNavigableMap(t *testing.T) { expected.Set([]string{utils.CapThresholds}, tmpTresholdIDs, false, false) expected.Set([]string{utils.CapStatQueues}, tmpStatQueueIDs, false, false) v1PrcEvRpl.SetMaxUsageNeeded(true) + if !v1PrcEvRpl.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1PrcEvRpl.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) } @@ -1310,6 +1331,9 @@ func TestV1ProcessEventReplyAsNavigableMap(t *testing.T) { v1per.MaxUsage = 5 * time.Minute expected.Set([]string{utils.CapMaxUsage}, 5*time.Minute, false, false) v1per.SetMaxUsageNeeded(true) + if !v1per.getMaxUsage { + t.Errorf("Expected getMaxUsage to be true") + } if rply, _ := v1per.AsNavigableMap(nil); !reflect.DeepEqual(expected, rply) { t.Errorf("Expecting \n%+v\n, received: \n%+v", expected, rply) }